gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-04 02:24:00 26 4
gpt4 key购买 nike

各位,

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

祝你好运

关于python - 在 Tensorflow 中使用迁移学习训练 Mobilnet V2,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50918521/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com