yolo.cfg 2.6 KB

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