caffe 視覺化網路及resnet50結構
阿新 • • 發佈:2019-01-01
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之類可查。
可以看出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處理。