【caffe】绘制网络结构图

简介: @tags caffe 网络结构 可视化当拿到一份网络定义文件net.prototxt,可以用工具画出网络结构。最快速的方法是使用在线工具netscope,粘贴内容后shift+回车就可以看结果了。caffe也自带了网络结构绘制工具,需要稍微配置下,并确保你用的caffe版本中实现了网络中涉及到的层。

@tags caffe 网络结构 可视化

当拿到一份网络定义文件net.prototxt,可以用工具画出网络结构。最快速的方法是使用在线工具netscope,粘贴内容后shift+回车就可以看结果了。

caffe也自带了网络结构绘制工具,需要稍微配置下,并确保你用的caffe版本中实现了网络中涉及到的层。以下是在windows下尝试并配置的步骤。

<caffe_root>/python目录下,看起来draw_net.py脚本提供了绘制网络结构的功能。不过不能直接执行它,需要在用visual studio编译caffe项目的时候,配置python支持(推荐装anaconda2并且配置阿里云的pip源),然后:

1 环境变量

创建系统环境变量PythonPath,内容为:

<caffe_root>\Build\x64\Release\pycaffe

<caffe_root>换成你自己的目录,比如我的是d:\lib\caffe-master

2 pip包

用pip安装一些必要的包

pip install protobuf pydot

3 安装graphviz

http://www.graphviz.org/Download_windows.php 到这里下载。安装后把/bin目录追加到系统环境变量PATH上

4 绘图

重开一个cmd控制台,根据prototxt来绘制网络图片,

draw_net.py执行的时候带三个参数

第一个参数:网络模型的prototxt文件

第二个参数:保存的图片路径及名字

第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。

d:
cd d:\lib\caffe-master\Build\x64\Release\pycaffe
# python draw_net.py prototxt文件名 保存图片文件名 --rankdir=方向,比如:
python draw_net.py d:\lib\caffe-master\examples\mnist\lenet_train_test.prototxt d:\lib\caffe-master\examples\mnist\lenet_train_test.png --rankdir=LR

看看lenet_train_test.prototxt网络结构绘制出来是什么样子的:

img_04d63aa5dd6f08a0a767a330fd1adc11.png

目录
相关文章
|
3月前
|
机器学习/深度学习 编解码 并行计算
【FasterVIT】试图从FasterVIT网络结构中窥探出一些有用的信息
【FasterVIT】试图从FasterVIT网络结构中窥探出一些有用的信息
25 0
【FasterVIT】试图从FasterVIT网络结构中窥探出一些有用的信息
|
3月前
|
机器学习/深度学习 存储 算法
【复现】尝试使用numpy对卷积神经网络中各经典结构进行改写复现
【复现】尝试使用numpy对卷积神经网络中各经典结构进行改写复现
40 0
【复现】尝试使用numpy对卷积神经网络中各经典结构进行改写复现
|
25天前
|
负载均衡 算法 应用服务中间件
面试题:Nginx有哪些负载均衡算法?Nginx位于七层网络结构中的哪一层?
字节跳动面试题:Nginx有哪些负载均衡算法?Nginx位于七层网络结构中的哪一层?
35 0
|
3月前
|
机器学习/深度学习 编解码 Go
YOLOv8改进 | Conv篇 | 利用YOLO-MS的MSBlock轻量化网络结构(既轻量又长点)
YOLOv8改进 | Conv篇 | 利用YOLO-MS的MSBlock轻量化网络结构(既轻量又长点)
123 0
|
2月前
|
机器学习/深度学习 PyTorch 测试技术
|
27天前
|
机器学习/深度学习 存储 人工智能
深度学习第1天:深度学习入门-Keras与典型神经网络结构
深度学习第1天:深度学习入门-Keras与典型神经网络结构
35 0
|
5天前
网络拓扑结构
网络拓扑结构主要包括星型、环型、总线型、树型和网状型。星型拓扑以中心节点控制全网,简单易管理,但依赖中央节点。环型拓扑信息沿环路传递,故障可能影响全网。总线型结构简洁适合小规模网络。树型拓扑是多星型结构的组合,适合复杂网络环境,具有扩展性。网状拓扑节点间多路径连接,提高可靠性,但结构复杂,成本高。
14 2
|
6天前
|
数据可视化 算法 数据挖掘
R语言SIR模型网络结构扩散过程模拟SIR模型(Susceptible Infected Recovered )代码实例
R语言SIR模型网络结构扩散过程模拟SIR模型(Susceptible Infected Recovered )代码实例
13 0
|
18天前
|
运维 网络架构
软件体系结构 - 网络拓扑结构
【4月更文挑战第14天】软件体系结构 - 网络拓扑结构
13 0
|
19天前
|
机器学习/深度学习
yolov7论文学习——创新点解析、网络结构图
yolov7论文学习——创新点解析、网络结构图