在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的版本不一样造成的结果。