Closed fengshikun closed 7 years ago
You need to compile the train-caffe provided by the author rather than using the official caffe. They implemented their own cpm_data_layer.
@kevin-ssy ,Thanks for your reply! However, I have used the author's caffe(https://github.com/CMU-Perceptual-Computing-Lab/caffe_train.git), and set the caffe root to this, and then this error occurs :(
Have you made the pycaffe? You need to set your caffe root dir to caffe_dir + '/python'. More information about compiling pycaffe please refer to http://caffe.berkeleyvision.org/installation.html#compilation
Yes, I have made the pycaffe wrapper, and I just change a segment of code in setLayers.py to this:
caffe_path = '/home/kunkun/DL/pose_estimate/caffe_train'
#caffe_path = '/media/ssd/humanPose/caffe/'
import sys, os
sys.path.insert(0, os.path.join(caffe_path, 'python'))
import caffe
The directory '/home/kunkun/DL/pose_estimate/caffe_train' is where I clone the project "(https://github.com/CMU-Perceptual-Computing-Lab/caffe_train.git)"
I have searched the string "cpm_transform_param" in the project https://github.com/CMU-Perceptual-Computing-Lab/caffe_train.git and find nothing...
Yes, I found there's no such parameter defined in caffe.proto. Maybe the author just tried before and deleted them all, as they only use the data in coco and I guess such transformation is to transform the data in MPII pose dataset. I found the param transform_param contains all the flags required by coco. You may delete all the code related to cpm_transform_param and use transform_param for transformation instead.
@ZheC Could you please check the bug mentioned above?
@kevin-ssy ,Thanks!
@ZheC I met same problem. Have you solved it? Thanks a lot
@fengshikun @NIEYALI @kevin-ssy Thank you for reporting those issues. I have fixed them with an updated caffe_train(https://github.com/CMU-Perceptual-Computing-Lab/caffe_train). Also, it uses a newer version of caffe so that it is compatible with CUDA 7.5/ 8.0 and cudnn v5. Let me know if you meet other issues.
@ZheC Thank you for your kind explanation. I had the same problem for weeks and I download the updated caffe_train(https://github.com/CMU-Perceptual-Computing-Lab/caffe_train). but it doesn't fix and I have the same problem again when using setlayer.py. Can I ask you to share the build version of this caffe_train? Or the output files after running setlayer.py?
Hello, i want to train a model follow your training steps, when execute "python setLayers.py --exp 1", Generate errors likes this:
Looking forward to your help, Thanks!