在Tensorflow中使用转移学习训练Mobilnet V2

Question

民间，

如何使用mobilenet v2进行传输/域学习？我使用mobilnet v1培训代码来训练mobilenet v1，但是在模型库或tensorflow存储库中是否有用于训练v2的编写代码？

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

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

Answer 1

This具有mobilenet v2 tfslim模块，以及用于恢复已由tensorflow人员训练的权重的检查点文件。您可以使用this示例重新训练模型，也可以使用this教程采用更简单的方法。如果您正在寻找可移植性，可能需要查看TFLite，这将允许您使用任何张量流模型并使用tensorflow head https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools上的构建工具轻松地针对移动设备进行优化。

祝你好运

Answer 2

@Asher Mancinelli，实际上，此代码完全符合我的需要： https://raw.githubusercontent.com/tensorflow/hub/r0.1/examples/image_retraining/retrain.py