MXNet转换caffe模型报错及解决

在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