如何在COCO日期集上加载预先训练的模型进行图像分割？

Question

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

有人可以指导我怎么做吗？

P.S。我可以使用tensorflow模型动物园中的预训练模型进行物体检测，但我无法将其扩展到物体分割。

Answer 1

我对本周从iNaturalist.org下载的图像进行了语义分割，结果令人满意。我最终使用的模型是DeepLab v3模型，该模型可以在存储库中的tensorflow research文件夹中找到

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

请参阅他们针对现成细分的优秀演示：

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

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

我也发现了这篇博客文章

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

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

玩得开心！