pjreddie / darknet

Convolutional Neural Networks
http://pjreddie.com/darknet/
Other
25.54k stars 21.33k forks source link

Official YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors #2557

Open AlexeyAB opened 2 years ago

AlexeyAB commented 2 years ago

Official YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors


Darknet cfg/weights file - currently tested for inference only:

Test FPS on: https://github.com/AlexeyAB/darknet


YOLOv7 is more accurate and faster than YOLOv5 by 120% FPS, than YOLOX by 180% FPS, than Dual-Swin-T by 1200% FPS, than ConvNext by 550% FPS, than SWIN-L by 500% FPS, than PPYOLOE-X by 150% FPS.

YOLOv7 surpasses all known object detectors in both speed and accuracy in the range from 5 FPS to 160 FPS and has the highest accuracy 56.8% AP among all known real-time object detectors with 30 FPS or higher on GPU V100, batch=1.


image


image


yolov7_640_1280

ullrichthomas92 commented 1 year ago

I tried to train a custom dataset with yolov7-tiny and conv.87 Unfortunately it gets stuck at a loss above 100. Same configuration in yolov4-tiny works great. Can you please help?

ESJavadex commented 1 year ago

Are you going to launch the full yolov7.weights and yolov7.cfg? Or are there any option to convert it from .pt?

jhony2507 commented 1 year ago

Good Morning, I'm using yolov7 to detect diseases in papaya, but the results are horrible. I have approximately 20k samples, divided into 8 diseases, the annotations are correct and I still get a max mAP of 34% (this after over 20,000 iterations). This result is slightly WORSE than yolov4, which achieves a mAP of 37%. Does anyone have an idea what could be wrong? *** With the efficientDet-d3 I get a mAP of 65% for the same base (I don't have the computational capacity to use the D6,D7..)

Attached are the yolov4/v7 configuration files and sample images amostras yolov4-custom_cfg.txt yolov7-papaya_cfg.txt

developer239 commented 1 year ago

@AlexeyAB @ESJavadex @jhony2507 I would also like to know what how do I get the weights file for yolov7? Can I convert .pt to .weights somehow? (I guess not)

How do I train .weights then? How is it different from training V3/V4? 🙏

I guess we can convert .pt to . onnx though. 🤔

rishav1122 commented 1 year ago

I have the yolov7-w6 weights file in darknet format best.weights, Is there any way to convert it in pt. Also, is there a yolov7-W6 config file available (yolov7-w6.cfg)