Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training Caffe on my own dataset for object localization #57

Open
mtrth opened this issue Jun 26, 2015 · 1 comment
Open

Training Caffe on my own dataset for object localization #57

mtrth opened this issue Jun 26, 2015 · 1 comment

Comments

@mtrth
Copy link

mtrth commented Jun 26, 2015

I am new to caffe. I am trying to use caffe on my own dataset to perform object localization.I am following this blog to train caffe on my own dataset http://sites.duke.edu/rachelmemo/2015/04/03/train-and-test-lenet-on-your-own-dataset/

I was able to train and test the model as mentioned in the blog. For the next step object localization I am using the example to https://github.com/BVLC/caffe/blob/master/examples/detection.ipynb . But when I run the command to convert the data to .h5 format as mentioned in example using the command

./python/detect.py --crop_mode=selective_search --pretrained_model=./examples/trial/trial_iter_10000.caffemodel --model_def=./examples/trial/trial.prototxt --gpu --raw_scale=255 _temp/det_input.txt _temp/det_output.h5

where,

pretrained_model=./examples/trial/trial_iter_10000.caffemodel (new model based on my dataset) and

--model_def=./examples/trial/trial.prototxt (new model def )

I am getting the error:

Check failed: ShapeEquals(proto) shape mismatch (reshape not set)
*** Check failure stack trace: ***

But all my inputs are of same size 64*64.

@mtrth mtrth changed the title Training Caffe on my own datasert for object localization Training Caffe on my own dataset for object localization Jun 26, 2015
@catsdogone
Copy link

Does your label-file right?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants