tensorflow - 如何在 COCO 数据集上加载预训练模型进行图像分割？

Question

我想对视频文件中的对象进行语义分割。我更喜欢在 COCO 数据集（或 COCO stuff 数据集）上使用预训练模型，并开始使用它对我自己的视频文件进行语义分割和对象检测。我遇到的大多数主题都是关于 COCO 数据集的训练算法。我正在寻找一个可以直接用于分割我自己的视频文件的预训练模型（一个冻结的图形文件）。

谁能指导我如何做到这一点？

PS 我可以使用来自 tensorflow 模型动物园的预训练模型进行对象检测，但我无法将其扩展为对象分割。

Answer 1

本周我对从 iNaturalist.org 下载的图像进行了语义分割，结果令人鼓舞。我最终使用的模型是 DeepLab v3 模型，它可以在存储库的 tensorflow 研究文件夹中轻松获得

• https://github.com/tensorflow/models/tree/master/research/deeplab。

请查看他们关于现成细分的出色演示：

• https://colab.research.google.com/github/tensorflow/models/blob/master/research/deeplab/deeplab_demo.ipynb。

只是通过笔记本。您实际上可以通过提供图像 url 在浏览器中对其进行测试，如笔记本中所述，以查看它是否满足您的需求。之后，如果您熟悉 tensorflow，手动下载其中一个预训练模型应该相当容易，就像他们在演示中所做的那样，并设置在您的计算机上运行的东西。

我也找到了这篇博文

• http://blog.qure.ai/notes/semantic-segmentation-deep-learning-review，

非常了解当今可用的模型以及它们各自的优缺点

玩得开心！