• caffe学习笔记(十一)绘制网络模型


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

    在绘制之前,需要先安装两个库

    1、安装GraphViz

    apt-get install GraphViz

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

    2 、安装pydot

    pip install pydot

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

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

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

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

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

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

    例:绘制Lenet模型

    python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

    例:绘制cifar10的模型

    python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT
  • 相关阅读:
    tcp socket
    Spring MVC 复习笔记01
    Spring 复习笔记01
    mybatis 复习笔记03
    ActiveMQ实现负载均衡+高可用部署方案(转)
    JVM复习笔记
    JAVA NIO复习笔记
    java IO复习笔记
    并发编程复习笔记
    杂记复习笔记
  • 原文地址:https://www.cnblogs.com/zhoulixue/p/6639733.html
Copyright © 2020-2023  润新知