在Windows10下利用Caffe的Python接口进行绘制神经网络模型图时,出现了如下问题:

1. 无法加载pydot

解决方法:

在安装了Anconda的前提下使用

conda install pydot

但是由于conda设置之类的事情说是没办法找到相应的channel和下载文件。Google了一下,发现需要提前安装一下graphviz,于是:

conda install graphviz

安装成功之后继续:

conda install pydot

发现还是出错了,头皮发麻,用了pip:

pip install pydot

终于提示安装成功。

至此解决了pydot和graphviz的问题。

2. 运行draw_net.py后有报错

pydot find_graphviz = pydot.graphviz.find_graphviz

'module' object has no attribute 'graphviz'

这个看了draw_net.py的源码,把文件中所有相应的pydot.graphviz改为pydot即可。

可能是由于pydot的版本不一样造成的结果。