Advertisement
Guest User

Untitled

a guest
Mar 23rd, 2016
300
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 6.61 KB | None | 0 0
  1. name: "lstm_joints"
  2. layer {
  3. name: "data"
  4. type: "Python"
  5. top: "data"
  6. top: "label"
  7. top: "clip_markers"
  8. python_param {
  9. module: "sequence_input_layer"
  10. layer: "videoReadTrain_flow"
  11. }
  12. include: { phase: TRAIN }
  13. }
  14.  
  15. layer {
  16. name: "data"
  17. type: "Python"
  18. top: "data"
  19. top: "label"
  20. top: "clip_markers"
  21. python_param {
  22. module: "sequence_input_layer"
  23. layer: "videoReadTest_flow"
  24. }
  25. include: { phase: TEST stage: "test-on-test" }
  26. }
  27. layer {
  28. name: "conv1"
  29. type: "Convolution"
  30. bottom: "data"
  31. top: "conv1"
  32. param {
  33. lr_mult: 1
  34. decay_mult: 1
  35. }
  36. param {
  37. lr_mult: 2
  38. decay_mult: 0
  39. }
  40. convolution_param {
  41. num_output: 96
  42. kernel_size: 7
  43. stride: 2
  44. weight_filler {
  45. type: "gaussian"
  46. std: 0.01
  47. }
  48. bias_filler {
  49. type: "constant"
  50. value: 0.1
  51. }
  52. }
  53. }
  54. layer {
  55. name: "relu1"
  56. type: "ReLU"
  57. bottom: "conv1"
  58. top: "conv1"
  59. }
  60. layer {
  61. name: "pool1"
  62. type: "Pooling"
  63. bottom: "conv1"
  64. top: "pool1"
  65. pooling_param {
  66. pool: MAX
  67. kernel_size: 3
  68. stride: 2
  69. }
  70. }
  71. layer {
  72. name: "norm1"
  73. type: "LRN"
  74. bottom: "pool1"
  75. top: "norm1"
  76. lrn_param {
  77. local_size: 5
  78. alpha: 0.0001
  79. beta: 0.75
  80. }
  81. }
  82. layer {
  83. name: "conv2"
  84. type: "Convolution"
  85. bottom: "norm1"
  86. top: "conv2"
  87. param {
  88. lr_mult: 1
  89. decay_mult: 1
  90. }
  91. param {
  92. lr_mult: 2
  93. decay_mult: 0
  94. }
  95. convolution_param {
  96. num_output: 384
  97. kernel_size: 5
  98. group: 2
  99. stride: 2
  100. weight_filler {
  101. type: "gaussian"
  102. std: 0.01
  103. }
  104. bias_filler {
  105. type: "constant"
  106. value: 0.1
  107. }
  108. }
  109. }
  110. layer {
  111. name: "relu2"
  112. type: "ReLU"
  113. bottom: "conv2"
  114. top: "conv2"
  115. }
  116. layer {
  117. name: "pool2"
  118. type: "Pooling"
  119. bottom: "conv2"
  120. top: "pool2"
  121. pooling_param {
  122. pool: MAX
  123. kernel_size: 3
  124. stride: 2
  125. }
  126. }
  127. layer {
  128. name: "norm2"
  129. type: "LRN"
  130. bottom: "pool2"
  131. top: "norm2"
  132. lrn_param {
  133. local_size: 5
  134. alpha: 0.0001
  135. beta: 0.75
  136. }
  137. }
  138. layer {
  139. name: "conv3"
  140. type: "Convolution"
  141. bottom: "norm2"
  142. top: "conv3"
  143. param {
  144. lr_mult: 1
  145. decay_mult: 1
  146. }
  147. param {
  148. lr_mult: 2
  149. decay_mult: 0
  150. }
  151. convolution_param {
  152. num_output: 512
  153. pad: 1
  154. kernel_size: 3
  155. weight_filler {
  156. type: "gaussian"
  157. std: 0.01
  158. }
  159. bias_filler {
  160. type: "constant"
  161. value: 0.1
  162. }
  163. }
  164. }
  165. layer {
  166. name: "relu3"
  167. type: "ReLU"
  168. bottom: "conv3"
  169. top: "conv3"
  170. }
  171. layer {
  172. name: "conv4"
  173. type: "Convolution"
  174. bottom: "conv3"
  175. top: "conv4"
  176. param {
  177. lr_mult: 1
  178. decay_mult: 1
  179. }
  180. param {
  181. lr_mult: 2
  182. decay_mult: 0
  183. }
  184. convolution_param {
  185. num_output: 512
  186. pad: 1
  187. kernel_size: 3
  188. group: 2
  189. weight_filler {
  190. type: "gaussian"
  191. std: 0.01
  192. }
  193. bias_filler {
  194. type: "constant"
  195. value: 0.1
  196. }
  197. }
  198. }
  199. layer {
  200. name: "relu4"
  201. type: "ReLU"
  202. bottom: "conv4"
  203. top: "conv4"
  204. }
  205. layer {
  206. name: "conv5"
  207. type: "Convolution"
  208. bottom: "conv4"
  209. top: "conv5"
  210. param {
  211. lr_mult: 1
  212. decay_mult: 1
  213. }
  214. param {
  215. lr_mult: 2
  216. decay_mult: 0
  217. }
  218. convolution_param {
  219. num_output: 384
  220. pad: 1
  221. kernel_size: 3
  222. group: 2
  223. weight_filler {
  224. type: "gaussian"
  225. std: 0.01
  226. }
  227. bias_filler {
  228. type: "constant"
  229. value: 0.1
  230. }
  231. }
  232. }
  233. layer {
  234. name: "relu5"
  235. type: "ReLU"
  236. bottom: "conv5"
  237. top: "conv5"
  238. }
  239. layer {
  240. name: "pool5"
  241. type: "Pooling"
  242. bottom: "conv5"
  243. top: "pool5"
  244. pooling_param {
  245. pool: MAX
  246. kernel_size: 3
  247. stride: 2
  248. }
  249. }
  250. layer {
  251. name: "fc6"
  252. type: "InnerProduct"
  253. bottom: "pool5"
  254. top: "fc6"
  255. param {
  256. lr_mult: 1
  257. decay_mult: 1
  258. }
  259. param {
  260. lr_mult: 2
  261. decay_mult: 0
  262. }
  263. inner_product_param {
  264. num_output: 4096
  265. weight_filler {
  266. type: "gaussian"
  267. std: 0.01
  268. }
  269. bias_filler {
  270. type: "constant"
  271. value: 0.1
  272. }
  273. }
  274. }
  275. layer {
  276. name: "relu6"
  277. type: "ReLU"
  278. bottom: "fc6"
  279. top: "fc6"
  280. }
  281. layer {
  282. name: "drop6"
  283. type: "Dropout"
  284. bottom: "fc6"
  285. top: "fc6"
  286. dropout_param {
  287. dropout_ratio: 0.9
  288. }
  289. }
  290. layer{
  291. name: "reshape-data"
  292. type: "Reshape"
  293. bottom: "fc6"
  294. top: "fc6-reshape"
  295. reshape_param{
  296. shape{
  297. dim: 16
  298. dim: 24
  299. dim: 4096
  300. }
  301. }
  302. include: { phase: TRAIN }
  303. }
  304. layer{
  305. name: "reshape-data"
  306. type: "Reshape"
  307. bottom: "fc6"
  308. top: "fc6-reshape"
  309. reshape_param{
  310. shape{
  311. dim: 16
  312. dim: 3
  313. dim: 4096
  314. }
  315. }
  316. include: { phase: TEST stage: "test-on-test" }
  317. }
  318. layer{
  319. name: "reshape-label"
  320. type: "Reshape"
  321. bottom: "label"
  322. top: "reshape-label"
  323. reshape_param{
  324. shape{
  325. dim: 16
  326. dim: 24
  327. }
  328. }
  329. include: { phase: TRAIN }
  330. }
  331. layer{
  332. name: "reshape-label"
  333. type: "Reshape"
  334. bottom: "label"
  335. top: "reshape-label"
  336. reshape_param{
  337. shape{
  338. dim: 16
  339. dim: 3
  340. }
  341. }
  342. include: { phase: TEST stage: "test-on-test" }
  343. }
  344. layer{
  345. name: "reshape-cm"
  346. type: "Reshape"
  347. bottom: "clip_markers"
  348. top: "reshape-cm"
  349. reshape_param{
  350. shape{
  351. dim: 16
  352. dim: 24
  353. }
  354. }
  355. include: { phase: TRAIN }
  356. }
  357. layer{
  358. name: "reshape-cm"
  359. type: "Reshape"
  360. bottom: "clip_markers"
  361. top: "reshape-cm"
  362. reshape_param{
  363. shape{
  364. dim: 16
  365. dim: 3
  366. }
  367. }
  368. include: { phase: TEST stage: "test-on-test" }
  369. }
  370. layer {
  371. name: "lstm1"
  372. type: "LSTM"
  373. bottom: "fc6-reshape"
  374. bottom: "reshape-cm"
  375. top: "lstm1"
  376. recurrent_param {
  377. num_output: 256
  378. weight_filler {
  379. type: "uniform"
  380. min: -0.01
  381. max: 0.01
  382. }
  383. bias_filler {
  384. type: "constant"
  385. value: 0
  386. }
  387. }
  388. }
  389. layer {
  390. name: "lstm1-drop"
  391. type: "Dropout"
  392. bottom: "lstm1"
  393. top: "lstm1-drop"
  394. dropout_param {
  395. dropout_ratio: 0.5
  396. }
  397. }
  398. layer {
  399. name: "fc8-final"
  400. type: "InnerProduct"
  401. bottom: "lstm1-drop"
  402. top: "fc8-final"
  403. param {
  404. lr_mult: 10
  405. decay_mult: 1
  406. }
  407. param {
  408. lr_mult: 20
  409. decay_mult: 0
  410. }
  411. inner_product_param {
  412. num_output: 101
  413. weight_filler {
  414. type: "gaussian"
  415. std: 0.01
  416. }
  417. bias_filler {
  418. type: "constant"
  419. value: 0
  420. }
  421. axis: 2
  422. }
  423. }
  424. layer {
  425. name: "loss"
  426. type: "SoftmaxWithLoss"
  427. bottom: "fc8-final"
  428. bottom: "reshape-label"
  429. top: "loss"
  430. softmax_param {
  431. axis: 2
  432. }
  433. }
  434. layer {
  435. name: "accuracy"
  436. type: "Accuracy"
  437. bottom: "fc8-final"
  438. bottom: "reshape-label"
  439. top: "accuracy"
  440. accuracy_param {
  441. axis: 2
  442. }
  443. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement