caffe可视化网络结构

下载了需要安装版本的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(从下到上显示)

你可能感兴趣的:(caffe可视化网络结构)