I have a question about test.py.
cslee99 opened this issue · 0 comments
cslee99 commented
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.