extraction.cfg 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  1. [net]
  2. # Training
  3. # batch=128
  4. # subdivisions=4
  5. # Testing
  6. batch=1
  7. subdivisions=1
  8. height=224
  9. width=224
  10. max_crop=320
  11. channels=3
  12. momentum=0.9
  13. decay=0.0005
  14. learning_rate=0.1
  15. policy=poly
  16. power=4
  17. max_batches=1600000
  18. [convolutional]
  19. batch_normalize=1
  20. filters=64
  21. size=7
  22. stride=2
  23. pad=1
  24. activation=leaky
  25. [maxpool]
  26. size=2
  27. stride=2
  28. [convolutional]
  29. batch_normalize=1
  30. filters=192
  31. size=3
  32. stride=1
  33. pad=1
  34. activation=leaky
  35. [maxpool]
  36. size=2
  37. stride=2
  38. [convolutional]
  39. batch_normalize=1
  40. filters=128
  41. size=1
  42. stride=1
  43. pad=1
  44. activation=leaky
  45. [convolutional]
  46. batch_normalize=1
  47. filters=256
  48. size=3
  49. stride=1
  50. pad=1
  51. activation=leaky
  52. [convolutional]
  53. batch_normalize=1
  54. filters=256
  55. size=1
  56. stride=1
  57. pad=1
  58. activation=leaky
  59. [convolutional]
  60. batch_normalize=1
  61. filters=512
  62. size=3
  63. stride=1
  64. pad=1
  65. activation=leaky
  66. [maxpool]
  67. size=2
  68. stride=2
  69. [convolutional]
  70. batch_normalize=1
  71. filters=256
  72. size=1
  73. stride=1
  74. pad=1
  75. activation=leaky
  76. [convolutional]
  77. batch_normalize=1
  78. filters=512
  79. size=3
  80. stride=1
  81. pad=1
  82. activation=leaky
  83. [convolutional]
  84. batch_normalize=1
  85. filters=256
  86. size=1
  87. stride=1
  88. pad=1
  89. activation=leaky
  90. [convolutional]
  91. batch_normalize=1
  92. filters=512
  93. size=3
  94. stride=1
  95. pad=1
  96. activation=leaky
  97. [convolutional]
  98. batch_normalize=1
  99. filters=256
  100. size=1
  101. stride=1
  102. pad=1
  103. activation=leaky
  104. [convolutional]
  105. batch_normalize=1
  106. filters=512
  107. size=3
  108. stride=1
  109. pad=1
  110. activation=leaky
  111. [convolutional]
  112. batch_normalize=1
  113. filters=256
  114. size=1
  115. stride=1
  116. pad=1
  117. activation=leaky
  118. [convolutional]
  119. batch_normalize=1
  120. filters=512
  121. size=3
  122. stride=1
  123. pad=1
  124. activation=leaky
  125. [convolutional]
  126. batch_normalize=1
  127. filters=512
  128. size=1
  129. stride=1
  130. pad=1
  131. activation=leaky
  132. [convolutional]
  133. batch_normalize=1
  134. filters=1024
  135. size=3
  136. stride=1
  137. pad=1
  138. activation=leaky
  139. [maxpool]
  140. size=2
  141. stride=2
  142. [convolutional]
  143. batch_normalize=1
  144. filters=512
  145. size=1
  146. stride=1
  147. pad=1
  148. activation=leaky
  149. [convolutional]
  150. batch_normalize=1
  151. filters=1024
  152. size=3
  153. stride=1
  154. pad=1
  155. activation=leaky
  156. [convolutional]
  157. batch_normalize=1
  158. filters=512
  159. size=1
  160. stride=1
  161. pad=1
  162. activation=leaky
  163. [convolutional]
  164. batch_normalize=1
  165. filters=1024
  166. size=3
  167. stride=1
  168. pad=1
  169. activation=leaky
  170. [convolutional]
  171. filters=1000
  172. size=1
  173. stride=1
  174. pad=1
  175. activation=leaky
  176. [avgpool]
  177. [softmax]
  178. groups=1