2

伙计们,

如何使用 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

4

2 回答 2

2

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

祝你好运

于 2018-06-18T23:30:33.460 回答
0

@Asher Mancinelli,实际上,这段代码正是我需要的: https ://raw.githubusercontent.com/tensorflow/hub/r0.1/examples/image_retraining/retrain.py

于 2018-06-22T22:10:33.013 回答