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
mnn/mnn-0.2.0.3/tools/converter/source/caffe/Convolution.cpp:30: Check failed: weightBlob.shape().dim_size() == 4 ==> Conv Weight Dimension ERROR! #215
Comments
Maybe your caffe model use the old caffe proto. Please give us the model link, and we test on it. |
【Converted Done!】 |
@czy2014hust |
Thanks for quick check. So you can convert without any error? |
Yes, I test on resnet50 which you provide without any error. |
可以在Github MNN钉钉群里说 |
Thanks. I will try again. |
这个问题的原因是caffe仓库中提供的alexnet模型是旧版本的模型,需要用caffe的更新工具upgrade_net_proto_binary进行更新。 git clone https://github.com/BVLC/caffe.git
cd caffe && mkdir build && cd build
cmake -DCPU_ONLY=ON -DBUILD_python=OFF -DBUILD_python_layer=OFF -DUSE_OPENCV=OFF -DUSE_LEVELDB=OFF -DUSE_LMDB=OFF ..
make -j$(nproc) 编译完成后在tools目录下会生成upgrade_net_proto_binary和upgrade_net_proto_text等文件。 ./upgrade_net_proto_binary ../../../MNN/build_ubuntu/bvlc_alexnet.caffemodel ./bvlc_alexnet_upgrade.caffemodel |
Hi,
I am having trouble converting standard cnn architecture models such as alexnet, vgg16, googlenet etc. Following is the error log.
Please help.
Thanks.
Best,
Deepak
The text was updated successfully, but these errors were encountered: