AlexeyAB / darknet

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

Custom training: All nan with 0 counts #1634

Open omar391 opened 6 years ago

omar391 commented 6 years ago

I have marked all the images with Yolo_mark tools and create a weight with yolov3.cfg. However, when I tried to use the tiny config ( I followed this: https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects) the training goes wrong. ALL NAN and 0 counts training_1

Config: yolov3-tiny_obj.cfg.zip data+names.zip

1 0.565234 0.547222 0.108594 0.094444 0 0.680078 0.549306 0.097656 0.106944 0 0.261328 0.509722 0.052344 0.080556 0 0.286719 0.470833 0.048437 0.075000 0 0.257422 0.440972 0.049219 0.068056

What should I do? I have tried to check similar issues with no luck.

omar391 commented 6 years ago

@AlexeyAB Could you please help with my issue..

BlcaKHat commented 6 years ago

for me also, after 900 iteration, the avg loss become 0.000 and average IOU is nan. do you have a local machine for training

omar391 commented 6 years ago

I am using a local machine for training. I have avg loss =0, all NAN, all count =0. So it seems there must a problem somewhere.

mohhao commented 6 years ago

I have the same problem,have you solved it?

omar391 commented 6 years ago

No, I am still trying. I am currently checking if any of my yolo mark annotations is wrong and then recalculate anchors,

dexception commented 6 years ago

Use yolo_mark tool and check your dataset.

AlexeyAB commented 6 years ago

I just trained model with your cfg-file and my dataset, and I got avg loss 0.48 after 3000 iterations:

chart

clydebailey commented 5 years ago

What is bad.list and bad_label.list?