WongKinYiu / ScaledYOLOv4

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

How to see mAPs using test.py? #75

Closed blackCmd closed 3 years ago

blackCmd commented 3 years ago

When I try

python test.py --img 896 --conf 0.001 --batch 8 --device 0 --data swoon.yaml --weights weights/yolov4-p5.pt

I've received the result below. It's done. image

My yaml is here : image

And my data is here : image

How can I get result like below. image

WongKinYiu commented 3 years ago

pip install pycocotools and check if the annotation file is put in correct directory https://github.com/WongKinYiu/ScaledYOLOv4/blob/yolov4-large/test.py#L228

blackCmd commented 3 years ago

pip install pycocotools and check if the annotation file is put in correct directory https://github.com/WongKinYiu/ScaledYOLOv4/blob/yolov4-large/test.py#L228

Oh I see Thank you : )