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

Bug fix. #178

Closed kinoshita-hidetoshi closed 2 years ago

kinoshita-hidetoshi commented 2 years ago

Your program "run_ssd_live_demo.py" line 76, 79 output cv2.error "Bad argument" and it didn't work properly, so I fixed this bug. Please merge this fix if you allow me to fix it.

My environment is described below.

--Python: 3.10.4 --PyTorch: 1.11.0 --OS: Windows 11 home 21H2

Best regards, Kinoshita Hidetoshi