首页 > 解决方案 > 如何使用 tensorflow/models/deeplab 在自己的数据集上进行训练?

问题描述

这是github 上存储库的链接。我可以运行示例脚本来训练和冻结模型。但是,这只能在 pascal_voc 的数据集或其他一些公共数据集上运行。我需要在我自己的数据集上运行它,但我不知道我应该为该--tf_initial_checkpoint选项提供什么。当我的数据集与 pascal_voc 有不同数量的类时,我不能给它一个 initial_checkpoint。

标签: pythontensorflowdeep-learningimage-segmentation

解决方案


推荐阅读