Home

Awesome

Implementation of SSDLite in PyTorch 1.2+

This is the object detection code for our MobileNeXt paper. This repository implements SSDLite, which is presented in MobileNetV2. The implementation is heavily influenced by the SSD project.

Highlights

Experiment Setup

Please refer to INSTALL.md for all the information about installation.

For training on COCO, the batch size is set to 256 and the initial learning rate is set to 0.01. We use 8 GPUs with 12 GB memory to run the experiments with synchronized batch normalization (important). You can also use less GPUs as long as the GPU memory is enough but make sure that the batch size is 256. For example, 4 V100 GPUs are also enough for running.

For training on Pascal VOC, the batch size is set to 24 and the initial learning rate is set to 0.001. We use 4 GPUs with 12 GB memory to run the experiments with the standard batch normalization.

For more implementation details, please refer to the configs in this project.

MODEL ZOO

Pretrained Models:

NetworksLinks
MobilenetV2model
MobileNeXtmodel

COCO:

BackboneInput SizeBox APModel SizeDownload
MobileNetV2 (SSDLite)32022.334Mmodel
MobileNeXt (SSDLite)32023.336Mmodel

PASCAL VOC:

BackboneInput SizemAPModel SizeDownload
VGG16 (SSD)30077.7201MBmodel
VGG16 (SSDLite)51280.7207MBmodel
MobilenetV2 (SSDLite)32071.826MBmodel
MobileNeXt (SSDLite)32072.627MBmodel

Citations

If you use this project in your research, please cite this project.

@inproceedings{daquan2020rethinking,
  title={Rethinking Bottleneck Structure for Efficient Mobile Network Design},
  author={Daquan, Zhou and Hou, Qibin and Chen, Yunpeng and Feng, Jiashi and Yan, Shuicheng},
  booktitle={European conference on computer vision},
  year={2020}
}
@misc{hou2020ssdite-pytorch,
    author = {Qibin Hou},
    title = {{Fast Implementation of SSDLite in PyTorch}},
    year = {2020},
    howpublished = {\url{https://github.com/Andrew-Qibin/ssdlite-pytorch}}
}