WongKinYiu/yolor

I have a question about test.py.

cslee99 opened this issue · 0 comments

Test was driven by the script below.
test Script
python test.py --data data/data.yaml --cfg cfg/yolor_p6.cfg --weights runs/train/yolor_custom3/weights/best.pt --device 0 --img-size 512 --names data/data.names --conf 0.05

The results show labels and pred in jpg format in runs/test/expXX.

However, _label.jpg shows the correct bounding box, but _pred.jpg doesn't create a bounding box.

Is it because prediction failed?

The result of running detect.py is generating the correct bounding box.

Please help.