用draw_net.py绘制网络图时的小问题

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

你可能感兴趣的:(用draw_net.py绘制网络图时的小问题)