yolov1-tiny.cfg 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  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=.75
  14. exposure=.75
  15. hue = .1
  16. learning_rate=0.0005
  17. policy=steps
  18. steps=200,400,600,800,20000,30000
  19. scales=2.5,2,2,2,.1,.1
  20. max_batches = 40000
  21. [convolutional]
  22. batch_normalize=1
  23. filters=16
  24. size=3
  25. stride=1
  26. pad=1
  27. activation=leaky
  28. [maxpool]
  29. size=2
  30. stride=2
  31. [convolutional]
  32. batch_normalize=1
  33. filters=32
  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=64
  44. size=3
  45. stride=1
  46. pad=1
  47. activation=leaky
  48. [maxpool]
  49. size=2
  50. stride=2
  51. [convolutional]
  52. batch_normalize=1
  53. filters=128
  54. size=3
  55. stride=1
  56. pad=1
  57. activation=leaky
  58. [maxpool]
  59. size=2
  60. stride=2
  61. [convolutional]
  62. batch_normalize=1
  63. filters=256
  64. size=3
  65. stride=1
  66. pad=1
  67. activation=leaky
  68. [maxpool]
  69. size=2
  70. stride=2
  71. [convolutional]
  72. batch_normalize=1
  73. filters=512
  74. size=3
  75. stride=1
  76. pad=1
  77. activation=leaky
  78. [maxpool]
  79. size=2
  80. stride=2
  81. [convolutional]
  82. batch_normalize=1
  83. size=3
  84. stride=1
  85. pad=1
  86. filters=1024
  87. activation=leaky
  88. [convolutional]
  89. batch_normalize=1
  90. size=3
  91. stride=1
  92. pad=1
  93. filters=256
  94. activation=leaky
  95. [connected]
  96. output= 1470
  97. activation=linear
  98. [detection]
  99. classes=20
  100. coords=4
  101. rescore=1
  102. side=7
  103. num=2
  104. softmax=0
  105. sqrt=1
  106. jitter=.2
  107. object_scale=1
  108. noobject_scale=.5
  109. class_scale=1
  110. coord_scale=5