特定对象Tensorflow中的颜色检测

Question

我正在我的android应用程序上实现Tensorflow对象检测，我遵循了“Tensorflow-for-poets”的演示和教程，并成功地创建了一个模型。

我需要一个帮助，我有检测交通信号灯的需求，我有一个数据集，我已经创建了一个模型，在一般情况下，它工作得很好。

我想要的是检测交通信号灯的颜色是亮的，是绿色的还是红色的？

我添加了一个数据集，其中包含绿色和红色两种类型的点亮交通信号灯的图像，但它只是检测交通信号灯。

有没有人可以帮助我或指导我如何实现这一点？

Answer 1

你为什么要检测红绿灯的颜色？

因此，确定哪种灯在发光会更可靠，例如，创建两个类别“红色交通灯”和“绿色交通灯”，并在它们上训练你的模型。

sladomic在评论中引用的帖子中给出的答案对噪音并不是一成不变的。比方说，你在下午晚些时候有一张照片，当太阳下山时，你可能会有一个明亮的环境。因此，在检测到的交通灯的边界框内确定红色像素的数量可能会失败，因为由环境造成的红色像素的数量大于来自闪光灯的绿色像素的数量。