zoukankan      html  css  js  c++  java
  • Caffe---自带工具进行网络结构(xxx.prototxt)可视化

    Caffe---自带绘图工具(draw_net.py)绘制网络结构图(xxx.prototxt)

    目录:

    一,安装依赖库。

    二,draw_net.py使用说明。

    正文:

    一,安装依赖库。

    在绘制之前,需要先安装相关的依赖库
    1、安装GraphViz
    # sudo apt-get install GraphViz
    注意,这里用的是apt-get来安装,而不是pip.

    2 、安装pydot
    # sudo pip install pydot
    用的是pip来安装,而不是apt-get
    安装好了,就可以调用脚本来绘制图片了

    二,draw_net.py使用说明。
    draw_net.py执行的时候带三个参数
    第一个参数:网络模型的prototxt文件
    第二个参数:保存的图片路径及名字
    第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。

    例:绘制Lenet模型
    sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=LR

    #cd caffe
    #python python/draw_net.py examples/mnist/lenet_train_test.prototxt my_show_NetStructure/mnist_lenet_train_test.png --rankdir=LR

    #python python/draw_net.py examples/mnist/mnist_autoencoder.prototxt my_show_NetStructure/mnist_autoencoder.png --rankdir=LR

    #python python/draw_net.py examples/cifar10/cifar10_quick_train_test.prototxt my_show_NetStructure/cifar10_quick_train_test.png --rankdir=LR


    最后补充,在Caffe中可视化prototxt格式网络结构的方法:目前除了可以使用Caffe提供的【1】draw_net.py,还可以使用【2】Netscope在线可视化。

    Netscope是个支持prototxt格式描述的神经网络结构的在线可视工具,即:http://ethereon.github.io/netscope/quickstart.html

    使用方法:首先打开这个地址:http://ethereon.github.io/netscope/#/editor,然后把你的描述神经网络结构的prototxt文件复制到该编辑框里,最后按shift-enter,这样就可以直接以图形方式显示网络的结构 。

    使用Netscope在线可视化,本人亲自玩了几次,发现网址不好用了。

     

     

  • 相关阅读:
    51nod 1087 1 10 100 1000(找规律+递推+stl)
    51nod 1082 与7无关的数 (打表预处理)
    51 nod 1080 两个数的平方和
    1015 水仙花数(水题)
    51 nod 1003 阶乘后面0的数量
    51nod 1002 数塔取数问题
    51 nod 1001 数组中和等于K的数对
    51 nod 1081 子段求和
    51nod 1134 最长递增子序列 (O(nlogn)算法)
    51nod 1174 区间中最大的数(RMQ)
  • 原文地址:https://www.cnblogs.com/carle-09/p/9086793.html
Copyright © 2011-2022 走看看