【caffe 深度学习】4.Caffe的python接口安装以及模型可视化

一. python接口安装

    1.pip install protobuf



    2.pip install pydot (用于绘制网络图)



    3.安装graphviz并添加环境变量

        网上很多,自行下载安装。添加如下环境变量:

        


    4.修改CommonSettings.props文件

        <PythonSupport>

        <PythonSupport>true</PythonSupport>

        <PythonDir>D:\Anaconda2\</PythonDir>




5.编译

    打开

    

    编译pycaffe

    点确定

   

       编译完成后多了下图文件:



    6.把caffe-windows\Build\x64\Release\pycaffe\caffe复制到Anaconda2的Lib\site-packages目录下




    7.把Build\x64\Release\pycaffe\caffe目录下的所有文件复制到caffe-windows\python\caffe目录下


    选替换目标中的文件


    8.测试import caffe


    如图 输入python →import caffe 

    若没有任何错误 就代表python接口编译好了并成功导入caffe


二.使用绘制网络结构

    1.使用draw_net.py

        文件位置如下图:

    

        python执行draw_net.py
    第一个参数:--rankdir TB
    TB表示TOP到BOTTOM。或者使用LR,从左到右。
    第二个参数:网络结构描述文件的路径
    第三个参数:网络图的输出路径

我这里输入:

python E:\graduate_student\deep_learning\caffe\new_Win_caffe\document\1\caffe-windows\caffe-windows\python\draw_net.py --rankdir TB E:\graduate_student\deep_learning\caffe\new_Win_caffe\document\1\caffe-windows\caffe-windows\examples\mnist\lenet_train_test.prototxt E:\graduate_student\deep_learning\caffe\new_Win_caffe\document\1\caffe-windows\caffe-windows\examples\mnist\draw_net\1.png

生成图片:

    


    2.使用在线画图软件(需要翻墙)
        http://ethereon.github.io/netscope/#/editor

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页