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

Question

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

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

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

Answer 1

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

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

多数民众赞成，你完成了:)