51dev.com IT技术开发者社区

51dev.com 技术开发者社区

drawnet.py绘制网络结构

代码星球阅读(33)2020-10-13 收藏0次评论

脚本在caffe的python目录里,需要安装pydot,graphviz

安装pydot:
$ sudo apt-get install python-pydot 

安装graphviz:

 

$ sudo apt-get install graphviz  

 

生成的命令行:

python draw_net.py + 网络prototxt + 生成图片名

$ python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

–rankdir选项确定了输出图片的摆放位置,如TB (Top-Bottom) 就是网络从上到下拜访,同理RL (Right-Left)就是从右向左,LR (Left-Right) 就是从左向右摆放。默认使用的是LR方式。注意是--。

生成的可以是jpg,也可以是png,但生成的文件名必须是绝对路径,不能直接写成lenet_structure.jpg,那样会识别不了。

 

 

绘制模型会报的常见错误:

http://blog.csdn.net/yiranyhy/article/details/55509304

 

这里有一个在线的绘制:

http://ethereon.github.io/netscope/#/editor

以上就是drawnet.py绘制网络结构的全部内容。