weiliu89 / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
4.77k stars 1.67k forks source link

ssd_detect.py error 2 #615

Open FLsur opened 7 years ago

FLsur commented 7 years ago

Hello everyone. I tried to reproduce the results of the ssd_detect.ipynb, so I converted it into ssd_detect.py. However, when I execute it, it throws me the following error:

I0524 11:50:22.202144 8771 net.cpp:100] Creating Layer detection_out I0524 11:50:22.202153 8771 net.cpp:434] detection_out <- mbox_loc I0524 11:50:22.202162 8771 net.cpp:434] detection_out <- mbox_conf_flatten I0524 11:50:22.202172 8771 net.cpp:434] detection_out <- mbox_priorbox I0524 11:50:22.202183 8771 net.cpp:408] detection_out -> detection_out F0524 11:50:22.203439 8771 detection_output_layer.cpp:98] Check failed: num_testimage <= names_.size() (4952 vs. 2613) Check failure stack trace: Aborted (core dumped)

I think there is a problem with the .caffemodel loaded (the same as the example), but I am not very sure.

I am using caffe in CPU mode, in a standard PC without an nvidia GPU.

Thank you very much

suji104 commented 7 years ago

Hi FLsur. If you do not mind, could you see the full code of ssd_detect.py? thank you.

weiliu89 commented 7 years ago

You have to change the name_size_file

FLsur commented 7 years ago

Hi. Already solved by replacing:

num_test_image: 4952

for

num_test_image: 2613

in file deploy.prototxt.

suji104, my ssd_detect.py is basically this https://github.com/weiliu89/caffe/blob/ssd/examples/ssd_detect.ipynb with little to no modifications.

Thank you very much.