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

The confidence level of multiple class models is lower than that of a single model #8866

Open Nanmei opened 7 months ago

Nanmei commented 7 months ago

I use weigth of yolov4.conv.137 to training max_batches = 5000 I trained a class of models with 1700 images , one class detection effect, with a confidence level of 0.98

2023-11-10 09-09-06 2023-11-10 09-26-31 chart_yolov4-custom voc-dp.data.txt yolov4-custom.cfg.txt yolov4-custom-test.cfg.txt

When I added 2 classes, these two classes are associated with 1400 images 2023-11-09 13-38-29 gap is where I added 2 classes of corresponding images,I added 2 classes of 1400 images. Now I have 3 classes.I use weigth of yolov4.conv.137 to training max_batches = 50000 2023-11-09 13-36-26 2023-11-10 09-25-14 chart_yolov4-custom_conv137_50000

voc-dp.data.txt

yolov4-custom.cfg.txt

yolov4-custom-test.cfg.txt

I use the weights obtained from training three classes to detect. The confidence level of a class is very low, with a result of 0.44. Even other images cannot be detected

The training of multiple classes has caused my detection confidence to be very low. I cannot find the reason and cannot solve this problem at present