• Windows下使用python绘制caffe中.prototxt网络结构数据可视化


    准备工具:


    1. 已编译好的pycaffe

    2. Anaconda(python2.7)

    3. graphviz

    4. pydot 


    1. graphviz安装


    graphviz是贝尔实验室开发的一个开源的绘图工具,它可以很方便绘制结构化的图形网络,支持多种格式输出,如各种常见的图片格式(bmp、png等),PDF,SVG等。 graphviz使用dot作为脚本语言,只需要在dot脚本中定义图的顶点和边,以及形状、颜色、字体、填充等样式,graphviz就可以使用合适的布局算法对图形布局,使各顶点均匀分布在画布上,并且尽可能的减少边的交叉。


    graphviz for windows 下载地址:  http://www.graphviz.org/Download_windows.php

    下载完成之后解压出来是一个release的文件夹,把release下的bin文件夹的路径 C:ProgramDatagraphviz eleasein 添加到环境变量里,让系统可以找到graphviz的相关文件就可以了



    2. pydot安装


    pydot模块提供了python对graphviz dot语言的接口,使在python下可以调用graphviz工具绘图。

    pydot下载地址: https://pypi.python.org/pypi/pydot


    下载之后解压,在解压出来的pydot-1.2.3文件夹下有一个pydot.py的安装文件,在所在文件夹下运行cmd命令,输入python pydot.py install 完成安装。


    如果安装完之后还是提示找不到pydot模块的话,可以尝试直接在cmd命令下输入 pip install pydot,anaconda会自动下载安装:




    3.  .prtotxt网络结构可视化


    caffe中的lenet_train_test_prototxt文件定义了一个LeNet卷积神经网络模型,尝试使用python把该模型可视化。

    新建一个test.bat文件,输入以下内容:

    python D:SoftwareCaffecaffe-masterBuildx64Releasepycaffedraw_net.py 
    D:SoftwareCaffecaffe-masterexamplesmnistlenet_train_test.prototxt 
    D:SoftwareCaffeprototxtVisualizationprototxtV.pdf

    双击运行test.bat文件,执行完之后在相应文件夹路径下生成prototxtV.pdf文件,分成两部分显示:


  • 相关阅读:
    Linux命令:ssh
    Linux命令:sshpass
    Linux命令:ls
    Linux文件的时间
    Linux命令:findutils
    jfrog
    git
    git branch
    git remote
    java equals 和hashcode
  • 原文地址:https://www.cnblogs.com/mtcnn/p/9411850.html
Copyright © 2020-2023  润新知