当前位置: 代码迷 >> 综合 >> caffe 网络模型可视化python例程
  详细解决方案

caffe 网络模型可视化python例程

热度:93   发布时间:2023-10-17 03:39:10.0

所有目录与文件均是基于caffe根目录。

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

(1)安装graphviz(graphviz的是一款图形绘制工具,用来被python程序调用绘制图片)

未安装graphviz会出现:Syntax error"dot" not found in path.

安装graphviz不要用pip install安装,否则还是会找不到可执行程序

执行:sudo apt-get install graphviz

(2)安装pydot

执行:sudo pip install pydot

绘制网络模型图

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

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

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

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

执行:python python/draw_net.py models/bvlc_reference_caffenet/train_val.prototxt caffenet.png

(caffe根目录下,在terminal里运行,不是进入Python再运行)生成的图片在caffe目录下,也可以自己指定目录