各位,
如何使用 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
请您参考如下方法:
This具有 mobilenet v2 tfslim 模块,以及用于恢复已由 tensorflow 人员训练的权重的检查点文件。您可以使用 this重新训练模型的示例,或者您可以使用 this 采取更简单的方法教程。如果您正在寻找可移植性,可能需要查看 TFLite,它允许您采用任何 tensorflow 模型并使用 tensorflow head 上的构建工具轻松针对移动设备进行优化:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools .
祝你好运
