在Ubuntu环境下使用命令./run.sh vgg16转换caffe模型时出现报错。
1.报错
converting
Traceback (most recent call last):
File "./convert_model.py", line 5, in <module>
from convert_symbol import proto2symbol
File "/home/weiliu/mxnet/tools/caffe_converter/convert_symbol.py", line 12, in <module>
import caffe_parse.caffe_pb2
ImportError: No module named caffe_pb2
weiliu@jxas-NF5280M4:~/mxnet/tools/caffe_converter$ ./run.sh vgg16
converting
Traceback (most recent call last):
File "./convert_model.py", line 5, in <module>
from convert_symbol import proto2symbol
File "/home/weiliu/mxnet/tools/caffe_converter/convert_symbol.py", line 12, in <module>
import caffe_parse.caffe_pb2
ImportError: No module named caffe_pb2
2.原因
没有安装pycaffe
3.解决
3.1 下载caffe
http://caffe.berkeleyvision.org/installation.html
3.2 修改Makefile.config,并编译
make all
make test
make runtest
3.3 编译pycaffe
在caffe/python/caffe目录下执行:
for req in $(cat requirements.txt); do sudo pip install $req; done
返回到上两级目录,在caffe目录下编译pycaffe:
make pycaffe
3.4 修改环境变量
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
其它报错:
1.转换ResNet模型时的报错
https://github.com/dmlc/mxnet/issues/3138