search

AlexeyAB / darknet

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

max_batches in yolov4-custom.cfg #8871

Open TidelyPom opened 7 months ago

TidelyPom commented 7 months ago

Hi. I am training for custom objects of very different sizes in the image. As recommended I am using yolov4-custom.cfg instead of yolo-obj.cfg but there is something very different in yolov4-custom.cfg from yolo-obj.cfg : on lines 20 and 22 the values are 500500 for max_batches and 400000, 450000 for steps instead of 6000 and 4800, 5400. Is this the recommended values ? I have only 2 types of objects now and maybe 3 in the future. The problem with these values is that the training time is estimated at 450 hours, which is a bit long. Thank you

stephanecharette commented 7 months ago

You need to re-read the steps on training. See https://github.com/hank-ai/darknet#training

stephanecharette commented 7 months ago

Also of possible interest to you: https://www.ccoderun.ca/programming/yolo_faq/#configuration_template