Skip to content

MobileNet* models

Luigi edited this page Oct 1, 2018 · 2 revisions

MobileNet-v1

A. G. Howard, M. Zhu, B. Chen, D. Kalenichenko, W. Wang, T. Weyand, M. Andreetto, and H. Adam, “Mobilenets: Efficient convolutional neural networks for mobile vision applications,” arXiv preprint arXiv:1704.04861, 2017.

@article{howard2017mobilenets,
  title={Mobilenets: Efficient convolutional neural networks for mobile vision applications},
  author={Howard, Andrew G and Zhu, Menglong and Chen, Bo and Kalenichenko, Dmitry and Wang, Weijun and Weyand, Tobias and Andreetto, Marco and Adam, Hartwig},
  journal={arXiv preprint arXiv:1704.04861},
  year={2017}
}

Source: MobileNet-v1 repo

  • mobilenet(num_classes=1000, pretrained='imagenet')

MobileNet-v2

M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L.-C. Chen, “Mobilenetv2: Inverted residuals and linear bottlenecks,” in Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2018, pp. 4510–4520.

@inproceedings{sandler2018mobilenetv2,
  title={MobileNetV2: Inverted Residuals and Linear Bottlenecks},
  author={Sandler, Mark and Howard, Andrew and Zhu, Menglong and Zhmoginov, Andrey and Chen, Liang-Chieh},
  booktitle={Conference on Computer Vision and Pattern Recognition (CVPR)},
  pages={4510--4520},
  year={2018},
  organization={IEEE}
}

Source: MobileNet-v2 repo

  • mobilenetv2(num_classes=1000, pretrained='imagenet')
Clone this wiki locally