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

 [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：

 

[plain] view plaincopyprint?
 
  1. i0701 17:01:10.548092 26739 data_layer.cpp:85] output data size: 100,3,224,224  
  2. i0701 17:01:10.736845 26739 net.cpp:206] top shape: 100 3 224 224 (15052800)  
  3. i0701 17:01:10.736912 26739 net.cpp:206] top shape: 100 (100)  
  4. i0701 17:01:10.736929 26739 layer_factory.hpp:75] creating layer conv1_1  
  5. i0701 17:01:10.736968 26739 net.cpp:166] creating layer conv1_1  
  6. i0701 17:01:10.736979 26739 net.cpp:496] conv1_1 <- data  
  7. i0701 17:01:10.737004 26739 net.cpp:452] conv1_1 -> conv1_1  
  8. i0701 17:01:10.737030 26739 net.cpp:197] setting up conv1_1  
  9. i0701 17:01:10.738733 26739 net.cpp:206] top shape: 100 64 224 224 (321126400)  
  10. i0701 17:01:10.738770 26739 layer_factory.hpp:75] creating layer relu1_1  
  11. i0701 17:01:10.738786 26739 net.cpp:166] creating layer relu1_1  
  12. i0701 17:01:10.738824 26739 net.cpp:496] relu1_1 <- conv1_1  
  13. i0701 17:01:10.738838 26739 net.cpp:439] relu1_1 -> conv1_1 (in-place)  
  14. i0701 17:01:10.738853 26739 net.cpp:197] setting up relu1_1  
  15. i0701 17:01:10.738867 26739 net.cpp:206] top shape: 100 64 224 224 (321126400)  
  16. i0701 17:01:10.738878 26739 layer_factory.hpp:75] creating layer pool1  
  17. i0701 17:01:10.738890 26739 net.cpp:166] creating layer pool1  
  18. i0701 17:01:10.738900 26739 net.cpp:496] pool1 <- conv1_1  
  19. i0701 17:01:10.738914 26739 net.cpp:452] pool1 -> pool1  
  20. i0701 17:01:10.738930 26739 net.cpp:197] setting up pool1  
  21. i0701 17:01:10.738963 26739 net.cpp:206] top shape: 100 64 112 112 (80281600)  
  22. i0701 17:01:10.738975 26739 layer_factory.hpp:75] creating layer conv2_1  
  23. i0701 17:01:10.738992 26739 net.cpp:166] creating layer conv2_1  
  24. i0701 17:01:10.739001 26739 net.cpp:496] conv2_1 <- pool1  
  25. i0701 17:01:10.739017 26739 net.cpp:452] conv2_1 -> conv2_1  
  26. i0701 17:01:10.739030 26739 net.cpp:197] setting up conv2_1  
  27. i0701 17:01:10.746640 26739 net.cpp:206] top shape: 100 128 112 112 (160563200)  
  28. i0701 17:01:10.746669 26739 layer_factory.hpp:75] creating layer relu2_1  
  29. i0701 17:01:10.746682 26739 net.cpp:166] creating layer relu2_1  
  30. i0701 17:01:10.746691 26739 net.cpp:496] relu2_1 <- conv2_1  
  31. i0701 17:01:10.746702 26739 net.cpp:439] relu2_1 -> conv2_1 (in-place)  
  32. i0701 17:01:10.746714 26739 net.cpp:197] setting up relu2_1  
  33. i0701 17:01:10.746726 26739 net.cpp:206] top shape: 100 128 112 112 (160563200)  
  34. i0701 17:01:10.746734 26739 layer_factory.hpp:75] creating layer pool2  
  35. i0701 17:01:10.746749 26739 net.cpp:166] creating layer pool2  
  36. i0701 17:01:10.746759 26739 net.cpp:496] pool2 <- conv2_1  
  37. i0701 17:01:10.746770 26739 net.cpp:452] pool2 -> pool2  
  38. i0701 17:01:10.746783 26739 net.cpp:197] setting up pool2  
  39. i0701 17:01:10.746798 26739 net.cpp:206] top shape: 100 128 56 56 (40140800)  
  40. i0701 17:01:10.746809 26739 layer_factory.hpp:75] creating layer conv3_1  
  41. i0701 17:01:10.746809 26739 layer_factory.hpp:75] creating layer conv3_1  
  42. i0701 17:01:10.746825 26739 net.cpp:166] creating layer conv3_1  
  43. i0701 17:01:10.746835 26739 net.cpp:496] conv3_1 <- pool2  
  44. i0701 17:01:10.746846 26739 net.cpp:452] conv3_1 -> conv3_1  
  45. i0701 17:01:10.746860 26739 net.cpp:197] setting up conv3_1  
  46. i0701 17:01:10.747910 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)  
  47. i0701 17:01:10.747939 26739 layer_factory.hpp:75] creating layer relu3_1  
  48. i0701 17:01:10.747954 26739 net.cpp:166] creating layer relu3_1  
  49. i0701 17:01:10.747963 26739 net.cpp:496] relu3_1 <- conv3_1  
  50. i0701 17:01:10.747974 26739 net.cpp:439] relu3_1 -> conv3_1 (in-place)  
  51. i0701 17:01:10.747985 26739 net.cpp:197] setting up relu3_1  
  52. i0701 17:01:10.747997 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)  
  53. i0701 17:01:10.748009 26739 layer_factory.hpp:75] creating layer conv3_2  
  54. i0701 17:01:10.748021 26739 net.cpp:166] creating layer conv3_2  
  55. i0701 17:01:10.748030 26739 net.cpp:496] conv3_2 <- conv3_1  
  56. i0701 17:01:10.748045 26739 net.cpp:452] conv3_2 -> conv3_2  
  57. i0701 17:01:10.748060 26739 net.cpp:197] setting up conv3_2  
  58. i0701 17:01:10.750586 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)  
  59. i0701 17:01:10.750610 26739 layer_factory.hpp:75] creating layer relu3_2  
  60. i0701 17:01:10.750624 26739 net.cpp:166] creating layer relu3_2  
  61. i0701 17:01:10.750635 26739 net.cpp:496] relu3_2 <- conv3_2  
  62. i0701 17:01:10.750648 26739 net.cpp:439] relu3_2 -> conv3_2 (in-place)  
  63. i0701 17:01:10.750669 26739 net.cpp:197] setting up relu3_2  
  64. i0701 17:01:10.750681 26739 net.cpp:206] top shape: 100 256 56 56 (80281600)  
  65. i0701 17:01:10.750690 26739 layer_factory.hpp:75] creating layer pool3  
  66. i0701 17:01:10.750702 26739 net.cpp:166] creating layer pool3  
  67. i0701 17:01:10.750710 26739 net.cpp:496] pool3 <- conv3_2  
  68. i0701 17:01:10.750725 26739 net.cpp:452] pool3 -> pool3  
  69. i0701 17:01:10.750740 26739 net.cpp:197] setting up pool3  
  70. i0701 17:01:10.750756 26739 net.cpp:206] top shape: 100 256 28 28 (20070400)  
  71. i0701 17:01:10.750764 26739 layer_factory.hpp:75] creating layer conv4_1  
  72. i0701 17:01:10.750779 26739 net.cpp:166] creating layer conv4_1  
  73. i0701 17:01:10.750788 26739 net.cpp:496] conv4_1 <- pool3  
  74. i0701 17:01:10.750800 26739 net.cpp:452] conv4_1 -> conv4_1  
  75. i0701 17:01:10.750825 26739 net.cpp:197] setting up conv4_1  
  76. i0701 17:01:10.756436 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)  
  77. i0701 17:01:10.756474 26739 layer_factory.hpp:75] creating layer relu4_1  
  78. i0701 17:01:10.756489 26739 net.cpp:166] creating layer relu4_1  
  79. i0701 17:01:10.756499 26739 net.cpp:496] relu4_1 <- conv4_1  
  80. i0701 17:01:10.756510 26739 net.cpp:439] relu4_1 -> conv4_1 (in-place)  
  81. i0701 17:01:10.756523 26739 net.cpp:197] setting up relu4_1  
  82. i0701 17:01:10.756536 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)  
  83. i0701 17:01:10.756546 26739 layer_factory.hpp:75] creating layer conv4_2  
  84. i0701 17:01:10.756559 26739 net.cpp:166] creating layer conv4_2  
  85. i0701 17:01:10.756568 26739 net.cpp:496] conv4_2 <- conv4_1  
  86. i0701 17:01:10.756583 26739 net.cpp:452] conv4_2 -> conv4_2  
  87. i0701 17:01:10.756597 26739 net.cpp:197] setting up conv4_2  
  88. i0701 17:01:10.766434 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)  
  89. i0701 17:01:10.766474 26739 layer_factory.hpp:75] creating layer relu4_2  
  90. i0701 17:01:10.766490 26739 net.cpp:166] creating layer relu4_2  
  91. i0701 17:01:10.766500 26739 net.cpp:496] relu4_2 <- conv4_2  
  92. i0701 17:01:10.766513 26739 net.cpp:439] relu4_2 -> conv4_2 (in-place)  
  93. i0701 17:01:10.766531 26739 net.cpp:197] setting up relu4_2  
  94. i0701 17:01:10.766543 26739 net.cpp:206] top shape: 100 512 28 28 (40140800)  
  95. i0701 17:01:10.766552 26739 layer_factory.hpp:75] creating layer pool4  
  96. i0701 17:01:10.766573 26739 net.cpp:166] creating layer pool4  
  97. i0701 17:01:10.766582 26739 net.cpp:496] pool4 <- conv4_2  
  98. i0701 17:01:10.766595 26739 net.cpp:452] pool4 -> pool4  
  99. i0701 17:01:10.766608 26739 net.cpp:197] setting up pool4  
  100. i0701 17:01:10.766624 26739 net.cpp:206] top shape: 100 512 14 14 (10035200)  
  101. I0701 17:18:56.158187 29940 layer_factory.hpp:75] Creating layer conv5_1  
  102. I0701 17:18:56.158201 29940 net.cpp:166] Creating Layer conv5_1  
  103. I0701 17:18:56.158210 29940 net.cpp:496] conv5_1 <- pool4  
  104. I0701 17:18:56.158222 29940 net.cpp:452] conv5_1 -> conv5_1  
  105. I0701 17:18:56.158234 29940 net.cpp:197] Setting up conv5_1  
  106. I0701 17:18:56.168265 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)  
  107. I0701 17:18:56.168303 29940 layer_factory.hpp:75] Creating layer relu5_1  
  108. I0701 17:18:56.168320 29940 net.cpp:166] Creating Layer relu5_1  
  109. I0701 17:18:56.168329 29940 net.cpp:496] relu5_1 <- conv5_1  
  110. I0701 17:18:56.168341 29940 net.cpp:439] relu5_1 -> conv5_1 (in-place)  
  111. I0701 17:18:56.168355 29940 net.cpp:197] Setting up relu5_1  
  112. I0701 17:18:56.168368 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)  
  113. I0701 17:18:56.168378 29940 layer_factory.hpp:75] Creating layer conv5_2  
  114. I0701 17:18:56.168395 29940 net.cpp:166] Creating Layer conv5_2  
  115. I0701 17:18:56.168406 29940 net.cpp:496] conv5_2 <- conv5_1  
  116. I0701 17:18:56.168417 29940 net.cpp:452] conv5_2 -> conv5_2  
  117. I0701 17:18:56.168431 29940 net.cpp:197] Setting up conv5_2  
  118. I0701 17:18:56.178441 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)  
  119. I0701 17:18:56.178478 29940 layer_factory.hpp:75] Creating layer relu5_2  
  120. I0701 17:18:56.178493 29940 net.cpp:166] Creating Layer relu5_2  
  121. I0701 17:18:56.178501 29940 net.cpp:496] relu5_2 <- conv5_2  
  122. I0701 17:18:56.178514 29940 net.cpp:439] relu5_2 -> conv5_2 (in-place)  
  123. I0701 17:18:56.178527 29940 net.cpp:197] Setting up relu5_2  
  124. I0701 17:18:56.178539 29940 net.cpp:206] Top shape: 100 512 14 14 (10035200)  
  125. I0701 17:18:56.178547 29940 layer_factory.hpp:75] Creating layer pool5  
  126. I0701 17:18:56.178561 29940 net.cpp:166] Creating Layer pool5  
  127. I0701 17:18:56.178570 29940 net.cpp:496] pool5 <- conv5_2  
  128. I0701 17:18:56.178581 29940 net.cpp:452] pool5 -> pool5  
  129. I0701 17:18:56.178596 29940 net.cpp:197] Setting up pool5  
  130. I0701 17:18:56.178611 29940 net.cpp:206] Top shape: 100 512 7 7 (2508800)  
  131. I0701 17:18:56.178621 29940 layer_factory.hpp:75] Creating layer fc6  
  132. i0701 17:01:10.796613 26739 net.cpp:166] creating layer fc6  
  133. i0701 17:01:10.796622 26739 net.cpp:496] fc6 <- pool5  
  134. i0701 17:01:10.796634 26739 net.cpp:452] fc6 -> fc6  
  135. i0701 17:01:10.796650 26739 net.cpp:197] setting up fc6  
  136. i0701 17:01:11.236284 26739 net.cpp:206] top shape: 100 4096 (409600)  
  137. i0701 17:01:11.236351 26739 layer_factory.hpp:75] creating layer relu6  
  138. i0701 17:01:11.236373 26739 net.cpp:166] creating layer relu6  
  139. i0701 17:01:11.236384 26739 net.cpp:496] relu6 <- fc6  
  140. i0701 17:01:11.236404 26739 net.cpp:439] relu6 -> fc6 (in-place)  
  141. i0701 17:01:11.236423 26739 net.cpp:197] setting up relu6  
  142. i0701 17:01:11.236435 26739 net.cpp:206] top shape: 100 4096 (409600)  
  143. i0701 17:01:11.236444 26739 layer_factory.hpp:75] creating layer drop6  
  144. i0701 17:01:11.236464 26739 net.cpp:166] creating layer drop6  
  145. i0701 17:01:11.236472 26739 net.cpp:496] drop6 <- fc6  
  146. i0701 17:01:11.236486 26739 net.cpp:439] drop6 -> fc6 (in-place)  
  147. i0701 17:01:11.236500 26739 net.cpp:197] setting up drop6  
  148. i0701 17:01:11.236524 26739 net.cpp:206] top shape: 100 4096 (409600)  
  149. i0701 17:01:11.236534 26739 layer_factory.hpp:75] creating layer fc7  
  150. i0701 17:01:11.236549 26739 net.cpp:166] creating layer fc7  
  151. i0701 17:01:11.236557 26739 net.cpp:496] fc7 <- fc6  
  152. i0701 17:01:11.236569 26739 net.cpp:452] fc7 -> fc7  
  153. i0701 17:01:11.236585 26739 net.cpp:197] setting up fc7  
  154. i0701 17:01:11.301771 26739 net.cpp:206] top shape: 100 4096 (409600)  
  155. i0701 17:01:11.301842 26739 layer_factory.hpp:75] creating layer relu7  
  156. i0701 17:01:11.301864 26739 net.cpp:166] creating layer relu7  
  157. i0701 17:01:11.301877 26739 net.cpp:496] relu7 <- fc7  
  158. i0701 17:01:11.301898 26739 net.cpp:439] relu7 -> fc7 (in-place)  
  159. i0701 17:01:11.301916 26739 net.cpp:197] setting up relu7  
  160. i0701 17:01:11.301929 26739 net.cpp:206] top shape: 100 4096 (409600)  
  161. i0701 17:01:11.301939 26739 layer_factory.hpp:75] creating layer drop7  
  162. i0701 17:01:11.301954 26739 net.cpp:166] creating layer drop7  
  163. i0701 17:01:11.301962 26739 net.cpp:496] drop7 <- fc7  
  164. i0701 17:01:11.301972 26739 net.cpp:439] drop7 -> fc7 (in-place)  
  165. i0701 17:01:11.301985 26739 net.cpp:197] setting up drop7  
  166. i0701 17:01:11.302000 26739 net.cpp:206] top shape: 100 4096 (409600)  
  167. i0701 17:01:11.302008 26739 layer_factory.hpp:75] creating layer fc8  
  168. i0701 17:01:11.302023 26739 net.cpp:166] creating layer fc8  
  169. i0701 17:01:11.302032 26739 net.cpp:496] fc8 <- fc7  
  170. i0701 17:01:11.302044 26739 net.cpp:452] fc8 -> fc8  
  171. i0701 17:01:11.302058 26739 net.cpp:197] setting up fc8  
  172. i0701 17:01:11.464764 26739 net.cpp:206] top shape: 100 10000 (1000000)  



 

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

版权声明：本文为博主原创文章，未经博主允许不得转载。

posted @ 2015-09-11 19:55  菜鸡一枚  阅读(1691)  评论(0)  编辑  收藏  举报