【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
文件位置如下图:
第一个参数:--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