Caffe下使用draw_net.py绘制lenet网络结构图

HMHx2_YY

于 2017-11-03 00:40:10 发布

阅读量741

点赞数 1

分类专栏： caffe学习文章标签： caffe学习

本文链接：https://blog.csdn.net/HMH2_YY/article/details/78430875

版权

本文介绍了如何在Caffe环境下使用draw_net.py绘制Lenet网络结构图。首先，需要安装GraphViz和pydot这两个库。在安装完成后，通过调用draw_net.py脚本可以实现网络模型的可视化。但在运行过程中可能会遇到错误，解决办法可参考相关链接。

摘要由CSDN通过智能技术生成

一.安装库：
python/draw_net.py, 这个文件，就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。

在绘制之前，需要先安装两个库

１、安装ＧraphViz

sudo su

sudo apt-get insall graphviz ＃这里用的是apt-get来安装，而不是pip.

2 、安装pydot

sudo pip install pydot　＃pip安装pydot
pip uninstall pyparsing

二．使用draw_net.py绘制lenet网络结构图
安装好了，就可以调用脚本来绘制图片了，draw_net.py在/caffe/python路径下，用于可视化Net

cd caffe/python  #进入draw_net.py目录下

python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../e

关注

专栏目录