必威体育Betway必威体育官网
当前位置:首页 > IT技术

caffe学习笔记(十三)--绘制网络结构图

时间:2019-10-13 12:14:30来源:IT技术作者:seo实验室小编阅读:74次「手机版」
 

网络结构图

小编推荐两种方法绘制深度学习网络结构图

方法一    通过Netscope

具体网址为http://ethereon.github.io/netscope/#/editor

PS:在linux系统可以正常打开,在windows系统可能需要翻墙。。

打开后界面如图

# Enter your network definition here.           在这里输入你的网络定义,也就是你的Prototxt的代码

# Use Shift+Enter to update the visualization.     使用Shift+Enter 进行更新可视化

下面举个例子:

使用caffe自带的mnist手写体识别的LeNet 网络结构绘制,部分截图如下:

方法二   调用caffe中的draw_net.py文件进行绘制结构图

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

在绘制之前,需要先安装两个库,以下代码都是在caffe根目录下执行

1、安装GraphViz

sudo apt-get install GraphViz

注意,这里用的是apt-get来安装,而不是pip.

2 、安装pydot

sudo pip install pydot

用的是pip来安装,而不是apt-get

安装好了,就可以调用脚本来绘制图片了

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

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

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

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

例:绘制cifar10的模型

sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT
就会在caffe根目录的netImage文件夹中产生cifar.png.

相关阅读

《java虚拟机原理图解》

转载自:http://blog.csdn.net/luanlouis/ 注:一下内容均针对jdk1.7之前的版本,新版中中可能有一些变动 JVM运行时数据区(JVM Runtime

绝地求生的高校地图!蚂蚁电竞ACUL全国总决赛来袭!

在2018 年8月30号的ANTGAMER蚂蚁电竞品牌暨产品发布会上,蚂蚁电竞官方正式启动了“ACUL蚂蚁电竞全国高校冠军联赛”。此

SPFA 算法详解( 强大图解,不会都难!)&&spfa优化——深度

https://blog.csdn.net/muxidreamtohit/article/details/7894298  适用范围:给定的图存在负权边,这时类似Dijkstra等算法便没有了

单片机原理及应用学习笔记(一)

*在学习51之前,最先接接触到的是Arduino,这个东西上手确实十分的简单,可以做出来一些小玩意,但是它的性能终究是不够强大,对于一些比较

图像增强的几个方法以及Matlab代码

1. 灰度线性变换 灰度线性变换, 是一种空域的方法, 直接对每一个像素的灰度值进行操作假设图像为III则其中每一个像素点的灰度值为I

分享到:

栏目导航

推荐阅读

热门阅读