AlexeyAB / darknet

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

Custom training goes wrong with "good hyperparameters" #4751

Closed xevolesi closed 4 years ago

xevolesi commented 4 years ago

Hi @AlexeyAB !

I trained YOLOv3 with custom data set. Classes:

I used YOLOv3 .cfg file with 416x416 input and added "good hyperparameters" from #4430

Here is my chart.png

chart

Here is my yolov3.cfg: yolov3-default-416-mosaic-scale-ciou.cfg.txt

Some image examples from training and validation sets: Снимок экрана от 2020-01-24 10-47-00 Снимок экрана от 2020-01-24 10-47-11

I checked my labels - it seems like it's correct, because earlier i trained default YOLOv3 and it gave me about 77% mAP. Could you, please, tell me what could happen with my training?

Thanks!

AlexeyAB commented 4 years ago

Try to:


If it doesn't help try to

xevolesi commented 4 years ago

Thanks!

I'll try both options and will notice you.

xevolesi commented 4 years ago

Hi, @AlexeyAB !

Your first suggestion worked well.

Here is final chart:

Снимок экрана от 2020-01-31 12-42-45

Do you think this is the effect of max_delta or stopbackward removal?

AlexeyAB commented 4 years ago

Do you think this is the effect of max_delta or stopbackward removal?

I don't know what did you do

xevolesi commented 4 years ago

Hi, @AlexeyAB !

Try to:

* remove `stopbackward=1`

* and add `max_delta=10` for each `[yolo]` layer

i did this