tiny-yolo.cfg 1.2 KB

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