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

How to train this net using my own dataset? #21

Open
Kongsea opened this issue Feb 9, 2018 · 1 comment
Open

How to train this net using my own dataset? #21

Kongsea opened this issue Feb 9, 2018 · 1 comment

Comments

@Kongsea
Copy link

Kongsea commented Feb 9, 2018

I want to train this net using my own dataset.

I have successfully set the data, however, the number of class need to be modified and some model layers cannot be restored accordingly.

For example, when I run the code, it raised the following error:

Cannot copy param 0 weights from layer 'cls_score_readonly'; shape mismatch. Source param shape is 21 4096 (86016); target param shape is 146 4096 (598016). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

Cannot copy param 0 weights from layer 'cls_score'; shape mismatch. Source param shape is 21 4096 (86016); target param shape is 146 4096 (598016). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

So could anybody give me some advice on how to fix it?
Thank you!

@Kongsea
Copy link
Author

Kongsea commented Feb 9, 2018

Besides, where can I download the pretrained VGG16.V2.caffemodel?

unable to open file: name = 'data/imagenet_models/VGG16.v2.caffemodel', errno = 2, error message = 'No such file or directory'

We can download it from here.

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

1 participant