search

chuanqi305 / MobileNet-SSD

Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.
MIT License
2k stars 1.18k forks source link

Caffe model training keeps aborting #176

Open rajat-bhagat opened 4 years ago

rajat-bhagat commented 4 years ago

I have followed all the steps to create my own LMDB dataset, also I have created the model using the gen_model.sh script but then when i run the train.sh file it starts and then gives an abort error directly. Screenshot from 2020-03-06 12-28-18

Suraj520 commented 4 years ago

Yes @rajat-bhagat , I too encountered the same error ! @chuanqi305 Could you please check for a resolution ?

weishinyi commented 3 years ago

You can modify the "math_functions.cpp" and "sampler.cpp" of caffe ssd source code file to solver the problem.

Step1: find the file [caffe_ssd_root]/src/caffe/util/math_functions.cpp mark the line 250 as shown below: image

Step2: find the file [caffe_ssd_root]/src/caffe/util/sample.cpp add the codes after line 107 as shown below: image

Step3: rebuild the caffe ssd library.