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

How does the hue augmentation work? #4405

Open LukeAI opened 4 years ago

LukeAI commented 4 years ago

What is the hue data augmentation unit?

What would be a good value for hue and saturation if the colour of the object of interest is basically irrelevant and I want the detector to be able to find objects of the same type but different colour to most of the training set.

context: I have a very large open source dataset on car license plates (CCPD 2019, 250000 images) but they are all chinese and so all have a blue background. I want to be able to detect license plates from anywhere in the world. My plan is to start by training a large network just on this very large database of chinese number plates and then using that to automatically annotate a large, more diverse dataset of cars and trucks with unmarked license-plates which I can manually check and then use to train a small network.

LukeAI commented 4 years ago

eg. 0025-0_0-312 536_387 564-387 564_312 563_312 536_387 537-0_0_22_32_29_27_18-112-5

AlexeyAB commented 4 years ago

What is the hue data augmentation unit?

What would be a good value for hue and saturation if the colour of the object of interest is basically irrelevant and I want the detector to be able to find objects of the same type but different colour to most of the training set.

set hue=1.0

f.e.:

saturation = 3.0
exposure = 3.0
hue=1.0

Just check how does it work by training with flag -show_imgs