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

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

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

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

1、安装GraphViz

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值