[caffe]深度学习之图像分类模型VGG解读

一、简介

vgg和googlenet是2014年imagenet竞赛的双雄，这两类模型结构有一个共同特点是go deeper。跟googlenet不同的是，vgg继承了lenet以及alexnet的一些框架，尤其是跟alexnet框架非常像，vgg也是5个group的卷积、2层fc图像特征、一层fc分类特征，可以看做和alexnet一样总共8个part。根据前5个卷积group，每个group中的不同配置，vgg论文中给出了A~E这五种配置，卷积层数从8到16递增。从论文中可以看到从8到16随着卷积层的一步步加深，貌似通过加深卷积层数也已经到达准确率提升的瓶颈了。后面的有一些论文针对卷积层输入的前处理（例如batch normalization）和输出的后处理（例如prelu）做了研究。再进一步的提升方向会是什么呢？这个值得大家去深入思考。

二、网络分析

我根据http://cs.stanford.edu/people/karpathy/vgg_train_val.prototxt配置文件以及vgg论文指导修改得到了vgg_A网络结构。

在修改的过程中你会发现vgg为了做不同深度网络之间的对比，然后又不至于太多的修改网络，vgg中给所有的卷积层以及pool层都设置了一样的层操作参数，确保了每个group出来的shape都是一致的，不管你在卷积group中加多少层的卷积。

三、网络日志

以下是具体的shape log：

i0701 17:01:10.548092 26739 data_layer.cpp:85] output data size: 100,3,224,224i0701 17:01:10.736845 26739 net.cpp:206] top shape: 100 3 224 224 (15052800)i0701 17:01:10.736912 26739 net.cpp:206] top shape: 100 (100)i0701 17:01:10.736929 26739 layer_factory.hpp:75] creating layer conv1_1i0701 17:01:10.736968 26739 net.cpp:166] creating layer conv1_1i0701 17:01:10.736979 26739 net.cpp:496] conv1_1 <- datai0701 17:01:10.737004 26739 net.cpp:452] conv1_1 -> conv1_1i0701 17:01:10.737030 26739 net.cpp:197] setting up conv1_1i0701 17:01:10.738733 26739 net.cpp:206] top shape: 100 64 224 224 (321126400)i0701 17:01:10.738770 26739 layer_factory.hpp:75] creating layer relu1_1i0701 17:01:10.738786 26739 net.cpp:166] creating layer relu1_1i0701 17:01:10.738824 26739 net.cpp:496] relu1_1 <- conv1_1i0701 17:01:10.738838 26739 net.cpp:439] relu1_1 -> conv1_1 (in-place)i0701 17:01:10.738853 26739 net.cpp:197] setting up relu1_1i0701 17:01:10.738867 26739 net.cpp:206] top shape: 100 64 224 224 (321126400)i0701 17:01:10.738878 26739 layer_factory.hpp:75] creating layer pool1i0701 17:01:10.738890 26739 net.cpp:166] creating layer pool1i0701 17:01:10.738900 26739 net.cpp:496] pool1 <- conv1_1i0701 17:01:10.738914 26739 net.cpp:452] pool1 -> pool1i0701 17:01:10.738930 26739 net.cpp:197] setting up pool1i0701 17:01:10.738963 26739 net.cpp:206] top shape: 100 64 112 112 (80281600)i0701 17:01:10.738975 26739 layer_factory.hpp:75] creating layer conv2_1i0701 17:01:10.738992 26739 net.cpp:166] creating layer conv2_1i0701 17:01:10.739001 26739 net.cpp:496] conv2_1 <- pool1i0701 17:01:10.739017 26739 net.cpp:452] conv2_1 -> conv2_1i0701 17:01:10.739030 26739 net.cpp:197] setting up conv2_1i0701 17:01:10.746640 26739 net.cpp:206] top shape: 100 128 112 112 (160563200)i0701 17:01:10.746669 26739 layer_factory.hpp:75] creating layer relu2_1i0701 17:01:10.746682 26739 net.cpp:166] creating layer relu2_1i0701 17:01:10.746691 26739 net.cpp:496] relu2_1 <- conv2_1i0701 17:01:10.746702 26739 net.cpp:439] relu2_1 -> conv2_1 (in-place)i0701 17:01:10.746714 26739 net.cpp:197] setting up relu2_1i0701 17:01:10.746726 26739 net.cpp:206] top shape: 100 128 112 112 (160563200)i0701 17:01:10.746734 26739 layer_factory.hpp:75] creating layer pool2i0701 17:01:10.746749 26739 net.cpp:166] creating layer pool2i0701 17:01:10.746759 26739 net.cpp:496] pool2 <- conv2_1i0701 17:01:10.746770 26739 net.cpp:452] pool2 -> pool2i0701 17:01:10.746783 26739 net.cpp:197] setting up pool2i0701 17:01:10.746798 26739 net.cpp:206] top shape: 100 128 56 56 (40140800)i0701 17:01:10.746809 26739 layer_factory.hpp:75] creating layer conv3_1i0701 17:01:10.746809 26739 layer_factory.hpp:75] creating layer conv3_1i0701 17:01:10.746825 26739 net.cpp:166] creating layer conv3_1i0701 17:01:10.746835 26739 net.cpp:496] conv3_1 <- pool2i0701 17:01:10.746846 26739 net.cpp:452] conv3_1 -> conv3_1i0701 17:01:10.746860 26739 net.cpp:197] setting up conv3_1i0701 17:01:10.747910 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)i0701 17:01:10.747939 26739 layer_factory.hpp:75] creating layer relu3_1i0701 17:01:10.747954 26739 net.cpp:166] creating layer relu3_1i0701 17:01:10.747963 26739 net.cpp:496] relu3_1 <- conv3_1i0701 17:01:10.747974 26739 net.cpp:439] relu3_1 -> conv3_1 (in-place)i0701 17:01:10.747985 26739 net.cpp:197] setting up relu3_1i0701 17:01:10.747997 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)i0701 17:01:10.748009 26739 layer_factory.hpp:75] creating layer conv3_2i0701 17:01:10.748021 26739 net.cpp:166] creating layer conv3_2i0701 17:01:10.748030 26739 net.cpp:496] conv3_2 <- conv3_1i0701 17:01:10.748045 26739 net.cpp:452] conv3_2 -> conv3_2i0701 17:01:10.748060 26739 net.cpp:197] setting up conv3_2i0701 17:01:10.750586 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)i0701 17:01:10.750610 26739 layer_factory.hpp:75] creating layer relu3_2i0701 17:01:10.750624 26739 net.cpp:166] creating layer relu3_2i0701 17:01:10.750635 26739 net.cpp:496] relu3_2 <- conv3_2i0701 17:01:10.750648 26739 net.cpp:439] relu3_2 -> conv3_2 (in-place)i0701 17:01:10.750669 26739 net.cpp:197] setting up relu3_2i0701 17:01:10.750681 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)i0701 17:01:10.750690 26739 layer_factory.hpp:75] creating layer pool3i0701 17:01:10.750702 26739 net.cpp:166] creating layer pool3i0701 17:01:10.750710 26739 net.cpp:496] pool3 <- conv3_2i0701 17:01:10.750725 26739 net.cpp:452] pool3 -> pool3i0701 17:01:10.750740 26739 net.cpp:197] setting up pool3i0701 17:01:10.750756 26739 net.cpp:206] top shape: 100 256 28 28 (20070400)i0701 17:01:10.750764 26739 layer_factory.hpp:75] creating layer conv4_1i0701 17:01:10.750779 26739 net.cpp:166] creating layer conv4_1i0701 17:01:10.750788 26739 net.cpp:496] conv4_1 <- pool3i0701 17:01:10.750800 26739 net.cpp:452] conv4_1 -> conv4_1i0701 17:01:10.750825 26739 net.cpp:197] setting up conv4_1i0701 17:01:10.756436 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)i0701 17:01:10.756474 26739 layer_factory.hpp:75] creating layer relu4_1i0701 17:01:10.756489 26739 net.cpp:166] creating layer relu4_1i0701 17:01:10.756499 26739 net.cpp:496] relu4_1 <- conv4_1i0701 17:01:10.756510 26739 net.cpp:439] relu4_1 -> conv4_1 (in-place)i0701 17:01:10.756523 26739 net.cpp:197] setting up relu4_1i0701 17:01:10.756536 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)i0701 17:01:10.756546 26739 layer_factory.hpp:75] creating layer conv4_2i0701 17:01:10.756559 26739 net.cpp:166] creating layer conv4_2i0701 17:01:10.756568 26739 net.cpp:496] conv4_2 <- conv4_1i0701 17:01:10.756583 26739 net.cpp:452] conv4_2 -> conv4_2i0701 17:01:10.756597 26739 net.cpp:197] setting up conv4_2i0701 17:01:10.766434 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)i0701 17:01:10.766474 26739 layer_factory.hpp:75] creating layer relu4_2i0701 17:01:10.766490 26739 net.cpp:166] creating layer relu4_2i0701 17:01:10.766500 26739 net.cpp:496] relu4_2 <- conv4_2i0701 17:01:10.766513 26739 net.cpp:439] relu4_2 -> conv4_2 (in-place)i0701 17:01:10.766531 26739 net.cpp:197] setting up relu4_2i0701 17:01:10.766543 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)i0701 17:01:10.766552 26739 layer_factory.hpp:75] creating layer pool4i0701 17:01:10.766573 26739 net.cpp:166] creating layer pool4i0701 17:01:10.766582 26739 net.cpp:496] pool4 <- conv4_2i0701 17:01:10.766595 26739 net.cpp:452] pool4 -> pool4i0701 17:01:10.766608 26739 net.cpp:197] setting up pool4i0701 17:01:10.766624 26739 net.cpp:206] top shape: 100 512 14 14 (10035200)I0701 17:18:56.158187 29940 layer_factory.hpp:75] Creating layer conv5_1I0701 17:18:56.158201 29940 net.cpp:166] Creating Layer conv5_1I0701 17:18:56.158210 29940 net.cpp:496] conv5_1 <- pool4I0701 17:18:56.158222 29940 net.cpp:452] conv5_1 -> conv5_1I0701 17:18:56.158234 29940 net.cpp:197] Setting up conv5_1I0701 17:18:56.168265 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)I0701 17:18:56.168303 29940 layer_factory.hpp:75] Creating layer relu5_1I0701 17:18:56.168320 29940 net.cpp:166] Creating Layer relu5_1I0701 17:18:56.168329 29940 net.cpp:496] relu5_1 <- conv5_1I0701 17:18:56.168341 29940 net.cpp:439] relu5_1 -> conv5_1 (in-place)I0701 17:18:56.168355 29940 net.cpp:197] Setting up relu5_1I0701 17:18:56.168368 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)I0701 17:18:56.168378 29940 layer_factory.hpp:75] Creating layer conv5_2I0701 17:18:56.168395 29940 net.cpp:166] Creating Layer conv5_2I0701 17:18:56.168406 29940 net.cpp:496] conv5_2 <- conv5_1I0701 17:18:56.168417 29940 net.cpp:452] conv5_2 -> conv5_2I0701 17:18:56.168431 29940 net.cpp:197] Setting up conv5_2I0701 17:18:56.178441 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)I0701 17:18:56.178478 29940 layer_factory.hpp:75] Creating layer relu5_2I0701 17:18:56.178493 29940 net.cpp:166] Creating Layer relu5_2I0701 17:18:56.178501 29940 net.cpp:496] relu5_2 <- conv5_2I0701 17:18:56.178514 29940 net.cpp:439] relu5_2 -> conv5_2 (in-place)I0701 17:18:56.178527 29940 net.cpp:197] Setting up relu5_2I0701 17:18:56.178539 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)I0701 17:18:56.178547 29940 layer_factory.hpp:75] Creating layer pool5I0701 17:18:56.178561 29940 net.cpp:166] Creating Layer pool5I0701 17:18:56.178570 29940 net.cpp:496] pool5 <- conv5_2I0701 17:18:56.178581 29940 net.cpp:452] pool5 -> pool5I0701 17:18:56.178596 29940 net.cpp:197] Setting up pool5I0701 17:18:56.178611 29940 net.cpp:206] Top shape: 100 512 7 7 (2508800)I0701 17:18:56.178621 29940 layer_factory.hpp:75] Creating layer fc6i0701 17:01:10.796613 26739 net.cpp:166] creating layer fc6i0701 17:01:10.796622 26739 net.cpp:496] fc6 <- pool5i0701 17:01:10.796634 26739 net.cpp:452] fc6 -> fc6i0701 17:01:10.796650 26739 net.cpp:197] setting up fc6i0701 17:01:11.236284 26739 net.cpp:206] top shape: 100 4096 (409600)i0701 17:01:11.236351 26739 layer_factory.hpp:75] creating layer relu6i0701 17:01:11.236373 26739 net.cpp:166] creating layer relu6i0701 17:01:11.236384 26739 net.cpp:496] relu6 <- fc6i0701 17:01:11.236404 26739 net.cpp:439] relu6 -> fc6 (in-place)i0701 17:01:11.236423 26739 net.cpp:197] setting up relu6i0701 17:01:11.236435 26739 net.cpp:206] top shape: 100 4096 (409600)i0701 17:01:11.236444 26739 layer_factory.hpp:75] creating layer drop6i0701 17:01:11.236464 26739 net.cpp:166] creating layer drop6i0701 17:01:11.236472 26739 net.cpp:496] drop6 <- fc6i0701 17:01:11.236486 26739 net.cpp:439] drop6 -> fc6 (in-place)i0701 17:01:11.236500 26739 net.cpp:197] setting up drop6i0701 17:01:11.236524 26739 net.cpp:206] top shape: 100 4096 (409600)i0701 17:01:11.236534 26739 layer_factory.hpp:75] creating layer fc7i0701 17:01:11.236549 26739 net.cpp:166] creating layer fc7i0701 17:01:11.236557 26739 net.cpp:496] fc7 <- fc6i0701 17:01:11.236569 26739 net.cpp:452] fc7 -> fc7i0701 17:01:11.236585 26739 net.cpp:197] setting up fc7i0701 17:01:11.301771 26739 net.cpp:206] top shape: 100 4096 (409600)i0701 17:01:11.301842 26739 layer_factory.hpp:75] creating layer relu7i0701 17:01:11.301864 26739 net.cpp:166] creating layer relu7i0701 17:01:11.301877 26739 net.cpp:496] relu7 <- fc7i0701 17:01:11.301898 26739 net.cpp:439] relu7 -> fc7 (in-place)i0701 17:01:11.301916 26739 net.cpp:197] setting up relu7i0701 17:01:11.301929 26739 net.cpp:206] top shape: 100 4096 (409600)i0701 17:01:11.301939 26739 layer_factory.hpp:75] creating layer drop7i0701 17:01:11.301954 26739 net.cpp:166] creating layer drop7i0701 17:01:11.301962 26739 net.cpp:496] drop7 <- fc7i0701 17:01:11.301972 26739 net.cpp:439] drop7 -> fc7 (in-place)i0701 17:01:11.301985 26739 net.cpp:197] setting up drop7i0701 17:01:11.302000 26739 net.cpp:206] top shape: 100 4096 (409600)i0701 17:01:11.302008 26739 layer_factory.hpp:75] creating layer fc8i0701 17:01:11.302023 26739 net.cpp:166] creating layer fc8i0701 17:01:11.302032 26739 net.cpp:496] fc8 <- fc7i0701 17:01:11.302044 26739 net.cpp:452] fc8 -> fc8i0701 17:01:11.302058 26739 net.cpp:197] setting up fc8i0701 17:01:11.464764 26739 net.cpp:206] top shape: 100 10000 (1000000)

shape的计算方式可以参考[caffe]深度学习之图像分类模型AlexNet解读。