/mobilenetv2

TensorFlow implementation for MobileNetV2

Primary LanguagePythonMIT LicenseMIT

MobileNetV2

TensorFlow implementation for MobileNetV2

There is now an implementation in TensorFlow models repository: https://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet

According to the paper: Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation

Testing

Trained with TensorFlow-Slim image classification model library using 2 GTX 1080 Ti.

Dataset Model GPUs Sync gradients Training time Recall 5 Accuracy
Cifar10 MobileNet V2 2 False 3:50 hours 0.9967 0.9071
Cifar10 MobileNet V2 2 False 15 hours 0.9976 0.9315

Added the trained checkpoint under 'checkpoints' folder.