vgg_7.prototxt 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. # adapted from https://github.com/lltcggie/waifu2x-caffe
  2. name: "vgg_7"
  3. layer {
  4. name: "input"
  5. type: "Input"
  6. top: "input"
  7. input_param { shape: { dim: 1 dim: 3 dim: 142 dim: 142 } }
  8. }
  9. layer {
  10. name: "conv1_layer"
  11. type: "Convolution"
  12. bottom: "input"
  13. top: "conv1"
  14. convolution_param {
  15. num_output: 32
  16. kernel_size: 3
  17. stride: 1
  18. weight_filler {
  19. type: "gaussian"
  20. std: 0.01
  21. }
  22. }
  23. }
  24. layer {
  25. name: "conv1_relu_layer"
  26. type: "ReLU"
  27. bottom: "conv1"
  28. top: "conv1"
  29. relu_param {
  30. negative_slope: 0.1
  31. }
  32. }
  33. layer {
  34. name: "conv2_layer"
  35. type: "Convolution"
  36. bottom: "conv1"
  37. top: "conv2"
  38. convolution_param {
  39. num_output: 32
  40. kernel_size: 3
  41. stride: 1
  42. weight_filler {
  43. type: "gaussian"
  44. std: 0.01
  45. }
  46. }
  47. }
  48. layer {
  49. name: "conv2_relu_layer"
  50. type: "ReLU"
  51. bottom: "conv2"
  52. top: "conv2"
  53. relu_param {
  54. negative_slope: 0.1
  55. }
  56. }
  57. layer {
  58. name: "conv3_layer"
  59. type: "Convolution"
  60. bottom: "conv2"
  61. top: "conv3"
  62. convolution_param {
  63. num_output: 64
  64. kernel_size: 3
  65. stride: 1
  66. weight_filler {
  67. type: "gaussian"
  68. std: 0.01
  69. }
  70. }
  71. }
  72. layer {
  73. name: "conv3_relu_layer"
  74. type: "ReLU"
  75. bottom: "conv3"
  76. top: "conv3"
  77. relu_param {
  78. negative_slope: 0.1
  79. }
  80. }
  81. layer {
  82. name: "conv4_layer"
  83. type: "Convolution"
  84. bottom: "conv3"
  85. top: "conv4"
  86. convolution_param {
  87. num_output: 64
  88. kernel_size: 3
  89. stride: 1
  90. weight_filler {
  91. type: "gaussian"
  92. std: 0.01
  93. }
  94. }
  95. }
  96. layer {
  97. name: "conv4_relu_layer"
  98. type: "ReLU"
  99. bottom: "conv4"
  100. top: "conv4"
  101. relu_param {
  102. negative_slope: 0.1
  103. }
  104. }
  105. layer {
  106. name: "conv5_layer"
  107. type: "Convolution"
  108. bottom: "conv4"
  109. top: "conv5"
  110. convolution_param {
  111. num_output: 128
  112. kernel_size: 3
  113. stride: 1
  114. weight_filler {
  115. type: "gaussian"
  116. std: 0.01
  117. }
  118. }
  119. }
  120. layer {
  121. name: "conv5_relu_layer"
  122. type: "ReLU"
  123. bottom: "conv5"
  124. top: "conv5"
  125. relu_param {
  126. negative_slope: 0.1
  127. }
  128. }
  129. layer {
  130. name: "conv6_layer"
  131. type: "Convolution"
  132. bottom: "conv5"
  133. top: "conv6"
  134. convolution_param {
  135. num_output: 128
  136. kernel_size: 3
  137. stride: 1
  138. weight_filler {
  139. type: "gaussian"
  140. std: 0.01
  141. }
  142. }
  143. }
  144. layer {
  145. name: "conv6_relu_layer"
  146. type: "ReLU"
  147. bottom: "conv6"
  148. top: "conv6"
  149. relu_param {
  150. negative_slope: 0.1
  151. }
  152. }
  153. layer {
  154. name: "conv7_layer"
  155. type: "Convolution"
  156. bottom: "conv6"
  157. top: "conv7"
  158. convolution_param {
  159. num_output: 3
  160. kernel_size: 3
  161. stride: 1
  162. weight_filler {
  163. type: "gaussian"
  164. std: 0.01
  165. }
  166. }
  167. }
  168. layer {
  169. name: "target"
  170. type: "MemoryData"
  171. top: "target"
  172. top: "dummy_label2"
  173. memory_data_param {
  174. batch_size: 1
  175. channels: 3
  176. height: 142
  177. width: 142
  178. }
  179. include: { phase: TRAIN }
  180. }
  181. layer {
  182. name: "loss"
  183. type: "EuclideanLoss"
  184. bottom: "conv7"
  185. bottom: "target"
  186. top: "loss"
  187. include: { phase: TRAIN }
  188. }