【问题标题】:Train Mobilnet V2 with transfer learning in Tensorflow在 Tensorflow 中使用迁移学习训练 Mobilnet V2
【发布时间】:2018-11-27 20:01:05
【问题描述】:

伙计们,

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

【问题讨论】:

    标签: python tensorflow models training-data transfer-learning


    【解决方案1】:

    This 具有 mobilenet v2 tfslim 模块,以及用于恢复 tensorflow 人员已经训练的权重的检查点文件。您可以使用this 示例重新训练模型,或者您可以使用this 教程采取更简单的方法。如果您正在寻找可移植性,可能需要查看 TFLite,它可以让您采用任何 tensorflow 模型,并使用 tensorflow 头部的构建工具轻松优化它以适应移动设备:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools

    祝你好运

    【讨论】:

    【解决方案2】:

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

    【讨论】:

      猜你喜欢
      • 2018-04-12
      • 2022-09-27
      • 2021-04-17
      • 2020-05-21
      • 2020-10-31
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多