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:


  • 相关阅读:
    Django(二)
    Django(一)
    MYSQL理论知识汇总
    默认参数
    深浅拷贝和赋值关系
    bootstrap常用知识
    jQuery常用功能代码
    java集合框架知识总结
    Mysql数据库SQL语句整理
    基于IO流的模拟下载文件的操作
  • 原文地址:https://www.cnblogs.com/mtcnn/p/9411827.html
Copyright © 2011-2022 走看看