search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.22k stars 4.17k forks source link

Why the iou_thred params in train.py and readme.md are different #1603

Open Yuxinyi-Qiyu opened 1 year ago

Yuxinyi-Qiyu commented 1 year ago

hey, i find that the iou_thred in train.py (https://github.com/WongKinYiu/yolov7/blob/main/train.py#L499) and readme.md (https://github.com/WongKinYiu/yolov7/tree/main#testing) are different. One is 0.70, and the latter is 0.65. i wonder if any meaning in setting iou_thred in this way? thanks!

TusharChauhaan commented 1 year ago

Bro these threshold are for NMS see in the test.py file. You can put anything.