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