特定对象 Tensorflow 中的颜色检测

Question

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

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

我想要的是检测交通信号灯的颜色是什么颜色，这意味着它是绿色还是红色？

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

谁能帮助我或指导我如何实现这一目标？

Answer 1

为什么要检测交通灯的颜色？

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

在 sladomic 的 cmets 中引用的帖子中给出的答案对于噪声不是不变的。比如说，你有一个下午晚些时候的图像，当太阳落山时，你可能会有一个偏红的明亮环境。因此，确定检测到的红绿灯边界框内的红色像素数量可能会失败，因为环境导致的红色像素数量大于来自闪光灯的绿色像素数量。