search

qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad
MIT License
1.39k stars 529 forks source link

cv2.error: OpenCV(4.5.5) : -1 : error: (-5:Bad argument) in function 'rectangle' #188

Closed bryanbocao closed 1 year ago

bryanbocao commented 1 year ago 
python3 run_ssd_example.py mb2-ssd-lite models/mb2-ssd-lite-mp-0_686.pth models/voc-model-labels.txt coco.jpg
Inference time:  0.025010347366333008
Traceback (most recent call last):
  File "run_ssd_example.py", line 60, in <module>
    cv2.rectangle(orig_image, (box[0], box[1]), (box[2], box[3]), (255, 255, 0), 4)
cv2.error: OpenCV(4.5.5) :-1: error: (-5:Bad argument) in function 'rectangle'
> Overload resolution failed:
>  - Can't parse 'pt1'. Sequence item with index 0 has a wrong type
>  - Can't parse 'pt1'. Sequence item with index 0 has a wrong type
>  - Can't parse 'rec'. Expected sequence length 4, got 2
>  - Can't parse 'rec'. Expected sequence length 4, got 2
bryanbocao commented 1 year ago

Fixed by casting box type to int: 9a20fb6b3ef1f7a5709fcd160ebb18c236d60e54 https://github.com/qfgaohao/pytorch-ssd/pull/189/commits