search

pjreddie / darknet

Convolutional Neural Networks
http://pjreddie.com/darknet/
Other
25.84k stars 21.33k forks source link

How to detect color using YOLO ? #1267

Open varmatangoit opened 5 years ago

varmatangoit commented 5 years ago

Using YOLO I can detect an object, but I need to detect a color of an object which is present in an image. Any idea to fix this?

LilianYe commented 5 years ago

You can label different objects with the color to the same class. Then train your yolo with one class.

theshawnz commented 5 years ago

Sounds like the object comes in different colors. Why not train the different colors of the objects as separate classes? this would be helpful if an image has more than one objects within it.