1. 程式人生 > >caffe 視覺化網路及resnet50結構

caffe 視覺化網路及resnet50結構

sudo python /home/lu/caffe/python/draw_net.py /home/lu/caffe/examples/mnist/lenet_train_test.prototxt /home/lu/caffe/lenet.png --rankdir=BT

複製prototxt,shift+enter 即可 
這裡寫圖片描述

或者利用caffe/python 中的 draw_net.py。

python draw_net.py –rankdir TB /Users/eclipsycn/Documents/resnet-protofiles-master/ResNet_50_train_val.prototxt /Users/eclipsycn/Documents/resnet-protofiles-master/ResNet_50_train_val.jpg

–rankdir TB 是上下結構,其他結構LR之類可查。 
resnet

可以看出Resnet50結構: 
conv1 + (1) 
3*res2*3(branch a-c) + (9) 
3*res3*4(branch a-d) + (12) 
3*res4*6(branch a-f) + (18) 
3*res5*3(branch a-c) + (9) 
fc (1) 
=50 
3 4 6 3 每一階段第一個block 都會將輸入接1*1conv後+輸出,階段剩下(3-1)、(4-1)、(6-1)、(3-1)個block按正常block處理。