caffe可视化网络结构
2017-11-16 本文已影响0人
小沂沂
下载了需要安装版本的graphviz（具体的软件可以直接百度）然后将graphviz的root目录和bin目录都添加到了环境变量中
cd E:\MachineLearning\caffe\caffe-windows\Build\x64\Release\pycaffe
python draw_net.py E:\MachineLearning\caffe\caffe-windows\models\dcc_crowdnet\train_val.prototxt E:\MachineLearning\caffe\caffe-windows\models\dcc_crowdnet\1.png（默认是从左到右显示）
python draw_net.py E:\MachineLearning\caffe\caffe-windows\models\dcc_crowdnet\train_val.prototxt E:\MachineLearning\caffe\caffe-windows\models\dcc_crowdnet\2.png --rankdir=BT（从下到上显示）
