sanghoon / pva-faster-rcnn

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

error #91

Open LXWDL opened 6 years ago

LXWDL commented 6 years ago

Hi, when i run the following command: python ./tools/train_net.py --gpu=0 --solver=./models/pvanet_obsolete/example_train_384/solver.prototxt --weights=./models/pvanet/pva9.1/pva9.1_pretrained_no_fc6.caffemodel --iters=110000 --cfg=./models/pvanet_obsolete/cfgs/train.yml --imdb voc_2007_trainval an error occurs as following: Loading pretrained model weights from ./models/pvanet/pva9.1/pva9.1_pretrained_no_fc6.caffemodel F1025 12:08:32.385498 21261 net.cpp:767] Check failed: target_blobs.size() == source_layer.blobs_size() (1 vs. 2) Incompatible number of blobs for layer conv2_1/1/conv Check failure stack trace: Has been abandoned (core has been dumped) Anyone who knows how to solve this problem, I will be very grateful.