ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
571 stars 166 forks source link

Where to download the pre-trained MobileNetv2 model? #3

Closed JesseYang closed 6 years ago

JesseYang commented 6 years ago

I want to train ssd_lite with MobileNetv2 as the backbone. The pre-trained MobileNetv2 model should b loaded. Where can I download the pre-trained model? Thanks

ShuangXieIrene commented 6 years ago

Please download from the hyper-link in readme Performance section :)

JesseYang commented 6 years ago

Thanks. But those links refer to the SSD models, which can be used to do inference or evaluate the performance. When training the model by myself, MobileNetv2 model pre-trained on ImageNet should be loaded.

foreverYoungGitHub commented 6 years ago

You can try this one: https://github.com/tonylins/pytorch-mobilenet-v2