Skip to main content
 首页 » 编程设计

python之在 Tensorflow 中使用迁移学习训练 Mobilnet V2

2024年09月07日29luoye11

各位,

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

祝你好运