convert tensorflow mobilenet-v2 checkpoint to caffemodel
- pycaffe
- tensorflow >= 1.5
Clone https://github.com/tensorflow/models, and put 'models' folder in this repository
sh download.sh mobilenet_v2_1.0_224
See: https://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet
python generate_prototxt.py --factor 1.0 --image_scale 224
revised prototxt will be generated in prototxt_mobilenet_v2/ directory.
python converter_v2.py --factor 1.0 --image_scale 224
caffemodel will be generated in caffemodel_fromckpt/ directory.
- Fix padding problem by using pad=2 and slicing layers.
- Set epsilon to 1e-3 in batch_norm layer which is equal to tensorflow.
- ReLU6 not solved, but result is OK.