WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

I have a question about test.py. #256

Open cslee99 opened 2 years ago

cslee99 commented 2 years ago

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.