search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.63k stars 7.95k forks source link

Traffic light color detection #2451

Open aravindvel opened 5 years ago

aravindvel commented 5 years ago

Hi,

I am trying to find traffic light and its color. I am able to detect and draw the traffic light but I am not able to find the status of the traffic light whether it is red, green or yellow.

Pls help me to find the traffic light status.

AlexeyAB commented 5 years ago

@aravindvel Hi, Just label each light as different objects: 0 - red, 1 - green, yellow -2 and train it. Also before training set hue=0.05 saturation=1.1 exposure 1.5

arkjiang commented 5 years ago

@aravindvel Hi, Just label each light as different objects: 0 - red, 1 - green, yellow -2 and train it. Also before training set hue=0.05 saturation=1.1 exposure 1.5

Hi Alex, Could you explain a bit more what the hsv parameter adjustment does ? Thanks .

AlexeyAB commented 5 years ago

@arkjiang This is color data augmentation - it randomly changes color of image during training. So you should reduce it slightly.