How can I use yolo to color detection?

[Lapinchu]

Hi, I'm making a project to detect the color of traffic lights by YOLO.
Can the YOLO distinguish between green and red, which are the color of traffic lights? And how to do that?
I mean Is it possible for the YOLO to learn color detection?

[py-lacroix]

Maybe the simplest way is to recognize the shape of traffic light and check the position of the lighted lamp (no need to check colour). But be careful ! This position depends on the country.

[tarasevich-dmitry]

You need firstly to detect class of the object and the color, probably you need 2 networks for it
Also there is a solution in OpenCV for multiple color detection here

[AlexeyAB]

Hi, I'm making a project to detect the color of traffic lights by YOLO.
Can the YOLO distinguish between green and red, which are the color of traffic lights? And how to do that?
I mean Is it possible for the YOLO to learn color detection?

Yes, Yolo can distinguish between green and red.
You just should:

• label red and green objects as different classes
• disable color augmentation, set hue=0 instead of

  darknet/cfg/yolov3.cfg

  Line 16 in 61c9d02

  hue=.1

Then train the Yolo as usuall.

---

More about HSV: https://en.wikipedia.org/wiki/HSL_and_HSV

[seojupal]

@AlexeyAB
Thanks! and one more question.
Can yolo distinguish all the colors not only green or red?

[nugrahari]

can I training without fliping image? example the direction of the road