使用 Tensorflow 对象检测 API 检测对象并按颜色对对象进行分类答案

【问题标题】：Using Tensorflow object detection API to detect objects and classify objects by color使用 Tensorflow 对象检测 API 检测对象并按颜色对对象进行分类
【发布时间】：2018-01-03 08:49:58
【问题描述】：

我可以使用 Tensorflow 在我自己的数据集上训练模型。比如我训练了一个模型只检测安全帽，结果很好。

我下一步的计划是按颜色对已识别的安全头盔进行分类。但我仍在寻找方法。

我想知道是否应该使用不同的标签图重新训练模型，例如：[item1 red_helmet] [item2 blue_helmet] 并分别标记我的训练数据集？或者有没有其他棘手的方法可以达到同样的效果？

【问题讨论】：

能在这里找到真好！可以分享一下你的材料吗？我刚开始研究同一个主题（安全头盔检测）。我寻找了一个没有成功的数据集。也许你有一个 git-hub 回购？

标签： tensorflow

【解决方案1】：

您已经拥有图片中的感兴趣区域。

您所需要的只是从图片中提取头盔并将裁剪后的图像传递给可以检测颜色的 openCV 例程。

就是这样，你完成了:)

【讨论】：