1. 程式人生 > >繪製caffe中的prototxt網路結構圖

繪製caffe中的prototxt網路結構圖

(1)準備python環境

(2)編譯pycaffe

make clean

make -j 

make pycaffe

(3)繪製網路結構圖

python draw_net.py ../models/bvlc_alexnet/train_val.prototxt alexnet.png