yolov1.cfg 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. [net]
  2. # Testing
  3. batch=1
  4. subdivisions=1
  5. # Training
  6. # batch=64
  7. # subdivisions=8
  8. height=448
  9. width=448
  10. channels=3
  11. momentum=0.9
  12. decay=0.0005
  13. saturation=1.5
  14. exposure=1.5
  15. hue=.1
  16. learning_rate=0.0005
  17. policy=steps
  18. steps=200,400,600,20000,30000
  19. scales=2.5,2,2,.1,.1
  20. max_batches = 40000
  21. [convolutional]
  22. batch_normalize=1
  23. filters=64
  24. size=7
  25. stride=2
  26. pad=1
  27. activation=leaky
  28. [maxpool]
  29. size=2
  30. stride=2
  31. [convolutional]
  32. batch_normalize=1
  33. filters=192
  34. size=3
  35. stride=1
  36. pad=1
  37. activation=leaky
  38. [maxpool]
  39. size=2
  40. stride=2
  41. [convolutional]
  42. batch_normalize=1
  43. filters=128
  44. size=1
  45. stride=1
  46. pad=1
  47. activation=leaky
  48. [convolutional]
  49. batch_normalize=1
  50. filters=256
  51. size=3
  52. stride=1
  53. pad=1
  54. activation=leaky
  55. [convolutional]
  56. batch_normalize=1
  57. filters=256
  58. size=1
  59. stride=1
  60. pad=1
  61. activation=leaky
  62. [convolutional]
  63. batch_normalize=1
  64. filters=512
  65. size=3
  66. stride=1
  67. pad=1
  68. activation=leaky
  69. [maxpool]
  70. size=2
  71. stride=2
  72. [convolutional]
  73. batch_normalize=1
  74. filters=256
  75. size=1
  76. stride=1
  77. pad=1
  78. activation=leaky
  79. [convolutional]
  80. batch_normalize=1
  81. filters=512
  82. size=3
  83. stride=1
  84. pad=1
  85. activation=leaky
  86. [convolutional]
  87. batch_normalize=1
  88. filters=256
  89. size=1
  90. stride=1
  91. pad=1
  92. activation=leaky
  93. [convolutional]
  94. batch_normalize=1
  95. filters=512
  96. size=3
  97. stride=1
  98. pad=1
  99. activation=leaky
  100. [convolutional]
  101. batch_normalize=1
  102. filters=256
  103. size=1
  104. stride=1
  105. pad=1
  106. activation=leaky
  107. [convolutional]
  108. batch_normalize=1
  109. filters=512
  110. size=3
  111. stride=1
  112. pad=1
  113. activation=leaky
  114. [convolutional]
  115. batch_normalize=1
  116. filters=256
  117. size=1
  118. stride=1
  119. pad=1
  120. activation=leaky
  121. [convolutional]
  122. batch_normalize=1
  123. filters=512
  124. size=3
  125. stride=1
  126. pad=1
  127. activation=leaky
  128. [convolutional]
  129. batch_normalize=1
  130. filters=512
  131. size=1
  132. stride=1
  133. pad=1
  134. activation=leaky
  135. [convolutional]
  136. batch_normalize=1
  137. filters=1024
  138. size=3
  139. stride=1
  140. pad=1
  141. activation=leaky
  142. [maxpool]
  143. size=2
  144. stride=2
  145. [convolutional]
  146. batch_normalize=1
  147. filters=512
  148. size=1
  149. stride=1
  150. pad=1
  151. activation=leaky
  152. [convolutional]
  153. batch_normalize=1
  154. filters=1024
  155. size=3
  156. stride=1
  157. pad=1
  158. activation=leaky
  159. [convolutional]
  160. batch_normalize=1
  161. filters=512
  162. size=1
  163. stride=1
  164. pad=1
  165. activation=leaky
  166. [convolutional]
  167. batch_normalize=1
  168. filters=1024
  169. size=3
  170. stride=1
  171. pad=1
  172. activation=leaky
  173. #######
  174. [convolutional]
  175. batch_normalize=1
  176. size=3
  177. stride=1
  178. pad=1
  179. filters=1024
  180. activation=leaky
  181. [convolutional]
  182. batch_normalize=1
  183. size=3
  184. stride=2
  185. pad=1
  186. filters=1024
  187. activation=leaky
  188. [convolutional]
  189. batch_normalize=1
  190. size=3
  191. stride=1
  192. pad=1
  193. filters=1024
  194. activation=leaky
  195. [convolutional]
  196. batch_normalize=1
  197. size=3
  198. stride=1
  199. pad=1
  200. filters=1024
  201. activation=leaky
  202. [local]
  203. size=3
  204. stride=1
  205. pad=1
  206. filters=256
  207. activation=leaky
  208. [dropout]
  209. probability=.5
  210. [connected]
  211. output= 1715
  212. activation=linear
  213. [detection]
  214. classes=20
  215. coords=4
  216. rescore=1
  217. side=7
  218. num=3
  219. softmax=0
  220. sqrt=1
  221. jitter=.2
  222. object_scale=1
  223. noobject_scale=.5
  224. class_scale=1
  225. coord_scale=5