Caffe 使用Python绘制网络结构图是遇到的错误

Caffe 使用Python绘制网络结构图是遇到的错误

1. make Python时
错误：CXX/LD -o python/caffe/_caffe.so python/caffe/_caffe.cpp
python/caffe/_caffe.cpp:10:31: fatal error: numpy/arrayobject.h: 没有那个文件或目录
解决：检查是否安装了numpy，sudo apt-get install python-numpy
如果已安装查看Makefile.config文件中PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
安装路径可能时/usr/lib/python2.7/dist-packages
将上面的代码改成PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include

2. 绘制绘图时
make通过后，运行python ./python/draw_net.py ./models/bvlc_reference_caffenet/train_val.prototxt ~/Desktop/caffenet.png绘图。
错误：Traceback (most recent call last):
File “./python/draw_net.py”, line 9, in
import caffe.draw
File “/home/lee/caffe/python/caffe/draw.py”, line 22, in
import pydot
ImportError: No module named pydot
解决：sudo pip install pydot2安装pydot
错误：Couldn’t import dot_parser, loading of dot files will not be possible.
解决：sudo pip install pyparsing==1.5.7
错误：Drawing net to /home/lee/caffenet.png
Traceback (most recent call last):
File “./python/draw_net.py”, line 58, in
main()
File “./python/draw_net.py”, line 54, in main
phase)
File “/home/lee/caffe/python/caffe/draw.py”, line 244, in draw_net_to_file
fid.write(draw_net(caffe_net, rankdir, ext, phase))
File “/home/lee/caffe/python/caffe/draw.py”, line 223, in draw_net
return get_pydot_graph(caffe_net, rankdir, phase=phase).create(format=ext)
File “/usr/local/lib/python2.7/dist-packages/pydot.py”, line 1953, in create
‘GraphViz\’s executables not found’ )
pydot.InvocationException: GraphViz’s executables not found
解决：sudo apt-get install graphviz