Train on KITTI dataset about squeezedet HOT 8 CLOSED

@goan15910 What's your command for launching the training thread? Did you make any modification to the config file?

from squeezedet.

I use the vgg16 version in the train.sh and all config are the same as original.

from squeezedet.

When I tried to train SqueezeDet on KITTI dataset, I met the problem about joblib as following:

Traceback (most recent call last):
File "./src/train.py", line 281, in
tf.app.run()
File "/home/lsk/anaconda2/envs/tensorflow/lib/python2.7/site-packages/tensorflow/python/platform/app.py", line 43, in run
sys.exit(main(sys.argv[:1] + flags_passthrough))
File "./src/train.py", line 277, in main
train()
File "./src/train.py", line 122, in train
model = SqueezeDet(mc, FLAGS.gpu)
File "/home/lsk/my_tensorflow/squeezeDet/src/nets/squeezeDet.py", line 24, in init
self._add_forward_graph()
File "/home/lsk/my_tensorflow/squeezeDet/src/nets/squeezeDet.py", line 38, in _add_forward_graph
self.caffemodel_weight = joblib.load(mc.PRETRAINED_MODEL_PATH)
File "/home/lsk/anaconda2/envs/tensorflow/lib/python2.7/site-packages/joblib/numpy_pickle.py", line 459, in load
obj = unpickler.load()
File "/home/lsk/anaconda2/envs/tensorflow/lib/python2.7/pickle.py", line 864, in load
dispatchkey
File "/home/lsk/anaconda2/envs/tensorflow/lib/python2.7/pickle.py", line 1096, in load_global
klass = self.find_class(module, name)
File "/home/lsk/anaconda2/envs/tensorflow/lib/python2.7/pickle.py", line 1132, in find_class
klass = getattr(mod, name)
AttributeError: 'module' object has no attribute 'NumpyArrayWrapper'

My tensorflow version is 0.12, easydict is 1.6 joblib is 0.10.3.
I can also run the demo code, have you solved your problem? @goan15910

from squeezedet.

Hi, @samlong-yang
I made the pretrained-weights pkl file myself and it works.
I used this repository caffe-tensorflow to convert caffe model weight into npy file and dump it into pkl file. If you are going to use this repo, make sure not to transpose the channels as the original code of src/nn_skeleton.py, since the repo already switch them from caffe order into tensorflow order.

from squeezedet.

Thanks for your reply and suggestions very much! I will try for that.

from squeezedet.

Your solution works for me @goan15910 .
I find that there is a file in the project ./src/utils/caffemodel2pkl.py, which is also helpful for converting the caffemodel into pkl weight file. It also works fine to me.

from squeezedet.

This is probably because you used different version of joblib 0.10.3.
I just had the same problem with @samlong-yang when using joblib 0.9.4. I updated to 0.10.3 and everything is fine.

from squeezedet.

Dear @samlong-yang how do you convert the squeezenet_v1.1.caffemodel to pkl by caffemodel2pkl.py?
should the prototxt be modified in some way?
Thanks

from squeezedet.