Train Mobilnet V2 with transfer learning in Tensorflow

Question

Folks,

How can I do transfer/domain learning with mobilenet v2? I use the mobilnet v1 training code to train mobilenet v1, but is there written code to train v2 either in the models repository, or in the tensorflow repository?

https://github.com/tensorflow/models/tree/master/research

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/image_retraining

Answer 1

@Asher Mancinelli, actually, this code does exactly what I need: https://raw.githubusercontent.com/tensorflow/hub/r0.1/examples/image_retraining/retrain.py