zoukankan      html  css  js  c++  java
  • python绘制caffe中网络模型

    caffe-master/python/draw_net.py 实现绘制caffe中定义的网络模型功能,将.prototxt文件可视化。


    需要先安装pydot和protobuf工具


    通过Anaconda安装pydot和protobuf工具:

    sduo chmod 777 -R ~/anaconda2
    conda install protobuf
    conda install pydot




    如果还缺少其他工具,也可以通过这种方式安装。



    绘制Lenet网络模型


    MNIST手写体网络模型Lenet:

    cd ~/caffe-master
    ./python/draw_net.py ./examples/mnist/lenet_train_test.prototxt ./python/lenet.png




    cifar:

    cd ~/caffe-master
    ./python/draw_net.py ./examples/cifar10/cifar10_quick_train_test.prototxt ./python/cifar.png




    更改网络模型方向显示


    draw_net.py带三个参数:

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

    更改绘制方向为从下到上:

    Lenet:





    cifar:


  • 相关阅读:
    eclipse c++
    smb
    osx mount nfs/smb
    0927用的
    0926 隐藏地址栏
    0921笔记
    生成文件并下载
    在线图标
    react redux
    electron
  • 原文地址:https://www.cnblogs.com/mtcnn/p/9411827.html
Copyright © 2011-2022 走看看