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

标签 tensorflow video-processing image-segmentation semantic-segmentation mscoco

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

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

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

最佳答案

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

请查看他们关于现成分割的出色演示:

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

我也找到了这篇博文

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

玩得开心!

关于tensorflow - 如何在 COCO 数据集上加载预训练模型进行图像分割?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50619150/

相关文章:

python - NumPy - 从 2D numpy 数组创建 1-hot 张量

python - 如何访问 TF Hub 模块中的 BERT 中间层输出?

h.264 - 是否有免费/便宜的工具可以打印出 H.264 流上每个宏 block 的大小?

ruby - 理解 "/dev/video"输出

video - 如何从时间线位置数据制作视频?

c++ - 将数组拆分为子 block

python - 了解 TensorFlow LSTM 输入形状

python - Tensorflow 模型返回疯狂的损失值

python - 如何运行以张量为范围的循环? (在 tensorflow 中)

deep-learning - 分割掩码 RCNN 和 FPN