AlexeyAB / darknet

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

Advice for hyperparameter search? #5228

Open rgov opened 4 years ago

rgov commented 4 years ago

The code in this repo and the thorough instructions are great. I'd like to see if I can increase the quality of detection on my data set. I was thinking about searching across the hyper parameter space for a parameter set that works a little better.

Are there particular hyper parameters I should focus on? For instance, the image resizing?

In my case, my data has a number of adverse properties (low light, motion blur, etc.) To augment the data first I run it through algorithms to segment the foreground, and I dedicate one channel to optical flow data. So the images look different than most.

For example (ignore the bounding boxes which are not correct due to an earlier bug):

PastedGraphic-1

AlexeyAB commented 4 years ago

I dedicate one channel to optical flow data.

rgov commented 4 years ago

There are 3 channels;

AlexeyAB commented 4 years ago

comment these lines: https://github.com/AlexeyAB/darknet/blob/342a8d1561c19317f2d5fda0f099449b79b51716/cfg/yolov3.cfg#L14-L16

rgov commented 4 years ago

Thanks! Do you recommend searching any other parameters, such as width and height?

AlexeyAB commented 4 years ago

If all your trainin and test images have the same size 400x960, then use [net] width=416 height=960