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

Question

关闭。这个问题需要更多 focused .它目前不接受答案。

---

想要改进这个问题吗？ 更新问题，使其只关注一个问题 editing this post .

关闭 4 年前。

Improve this question

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

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

附:我可以使用 tensorflow model zoo 中的预训练模型进行对象检测，但我无法将其扩展为对象分割。

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 ,

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

玩得开心!