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在线可视化,本人亲自玩了几次,发现网址不好用了。

     

     

  • 相关阅读:
    java当前时间
    @Repository、@Service、@Controller 和 @Component
    spring mvc controller间跳转 重定向 传参
    MVC思想
    AJAX
    MySQL优化大总结
    MySQL数据库优化
    java JDBM2 的几个简单实例
    集群环境下如何防止定时任务重复执行?
    Java应用集群下的定时任务处理方案(mysql)
  • 原文地址:https://www.cnblogs.com/carle-09/p/9086793.html
Copyright © 2011-2022 走看看