search

WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network
GNU General Public License v3.0
2.02k stars 572 forks source link

How to test "mAP" on specific iou? #80

Closed blackCmd closed 3 years ago

blackCmd commented 3 years ago

image I want to test not mAP@0.5 or mAP@0.5:0.95 but only mAP@iou=0.8. Plz help me.. How can I do that? Where do I edit code?

ADD QUESTION

I want to train using IoU 0.8. Is it possible? If it possible, How can I do that?

WongKinYiu commented 3 years ago

https://github.com/WongKinYiu/ScaledYOLOv4/blob/yolov4-large/test.py#L195 as i remember, ap[:, 0], ap[:, 1], ap[:, 2], ... are ap50, ap55, ap60... you could try to change the index to get ap80, and use it in fitness function to save best.pt with best map80.

blackCmd commented 3 years ago

Ok. I'll try it ! Thank you