search

sanghoon / pva-faster-rcnn

Demo code for PVANet
https://arxiv.org/abs/1611.08588
Other
651 stars 241 forks source link

Finetune #28

Open rajiv235 opened 7 years ago

rajiv235 commented 7 years ago

I am trying to finetune pvanet on my data using solver under example_finetune. However I get the following error. I1201 22:09:59.937469 32013 net.cpp:270] This network produces output rpn_cls_loss I1201 22:09:59.937480 32013 net.cpp:270] This network produces output rpn_loss_bbox I1201 22:09:59.937691 32013 net.cpp:283] Network initialization done. I1201 22:09:59.938531 32013 solver.cpp:60] Solver scaffolding done. Loading pretrained model weights from data/ImagenetTrainedpvanetComppressed.model I1201 22:10:00.516289 32013 net.cpp:761] Ignoring source layer data I1201 22:10:00.516345 32013 net.cpp:761] Ignoring source layer label_data_1_split I1201 22:10:00.519294 32013 net.cpp:761] Ignoring source layer pool5 F1201 22:10:00.775359 32013 net.cpp:774] Cannot copy param 0 weights from layer 'fc6'; shape mismatch. Source param shape is 4096 13824 (56623104); target param shape is 4096 18432 (75497472). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

Has someone seen this error before? Can someone help me with this. I am using Imagenet pretrained model for initial weights.

Thanks

catsdogone commented 7 years ago

I think you should use the model in data/pvanet as pretrained model if you use the prototxt in example_finetune.

rajiv235 commented 7 years ago

Hey @catsdogone. Thanks for getting back. I tried that too but got the same error.