如何训练 tensorflow deeplab 模型？

Question

我需要用我的鞋子数据集训练 tensorflow deeplab 模型。 然后我将使用这个模型来删除图像鞋的背景。 我怎么能训练它？ 你能一步一步解释吗？ 你有这种情况的例子吗？ 张量流/深度实验室

Answer 1

您将需要阅读 Deeplab 代码的某些部分

• 下载仓库
• 现在您需要以正确的格式将数据放入 tfrecord
• 使用https://github.com/tensorflow/models/tree/master/research/deeplab/datasets 中的一些脚本下载和生成示例数据集
• 为您的鞋子数据集准备类似的脚本
• 将有关数据的信息添加到 Deeplab 源文件https://github.com/tensorflow/models/blob/master/research/deeplab/datasets/data_generator.py以类似格式添加信息，如示例数据集
• 检查架构的标志https://github.com/tensorflow/models/blob/master/research/deeplab/common.py
• 检查特定标志，然后使用train.py 、 vis.py 、 export_model.py 、 eval.py在文件夹https://github.com/tensorflow/models/tree/master/research/deeplab 中训练、导出、统计统计或可视化