search

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
571 stars 166 forks source link

failed when run : python demo.py --cfg=$file --demo=./experiments/person.jpg #14

Closed liwei46 closed 6 years ago

liwei46 commented 6 years ago

./experiments/cfgs/ssd_lite_mobilenetv2_train_voc.yml ===> Building model ==>Feature map size: [(19, 19), (10, 10), (5, 5), (3, 3), (2, 2), (1, 1)] Utilize GPUs for computation Number of GPU available 1 => loading checkpoint ./weights/ssd_lite/mobilenet_v2_ssd_lite_voc_73.2.pth /home/wei_li/anaconda3/lib/python3.6/site-packages/torch/tensor.py:321: UserWarning: self and other not broadcastable, but have the same number of elements. Falling back to deprecated pointwise behavior. return self.mul(other) ASSERT: "false" in file qasciikey.cpp, line 501 ./time_benchmark.sh: line 9: 16177 Aborted (core dumped) python demo.py --cfg=$file --demo=./experiments/person.jpg

foreverYoungGitHub commented 6 years ago

Which pytorch version are you using now. Currently, it only support 0.3.1. It will support 0.4.0 later.

liwei46 commented 6 years ago

Thanks for the reply. I am using 0.3.1, maybe due to my GPU memory not enough... sometimes it works. I will close it.

rawandeheliah commented 5 years ago

i have the same problem ? when i run a python file uses opencv , how did you fix this problem ?