vgg-16.cfg 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153
  1. [net]
  2. batch=128
  3. subdivisions=4
  4. height=256
  5. width=256
  6. channels=3
  7. learning_rate=0.00001
  8. momentum=0.9
  9. decay=0.0005
  10. [crop]
  11. crop_height=224
  12. crop_width=224
  13. flip=1
  14. exposure=1
  15. saturation=1
  16. angle=0
  17. [convolutional]
  18. filters=64
  19. size=3
  20. stride=1
  21. pad=1
  22. activation=relu
  23. [convolutional]
  24. filters=64
  25. size=3
  26. stride=1
  27. pad=1
  28. activation=relu
  29. [maxpool]
  30. size=2
  31. stride=2
  32. [convolutional]
  33. filters=128
  34. size=3
  35. stride=1
  36. pad=1
  37. activation=relu
  38. [convolutional]
  39. filters=128
  40. size=3
  41. stride=1
  42. pad=1
  43. activation=relu
  44. [maxpool]
  45. size=2
  46. stride=2
  47. [convolutional]
  48. filters=256
  49. size=3
  50. stride=1
  51. pad=1
  52. activation=relu
  53. [convolutional]
  54. filters=256
  55. size=3
  56. stride=1
  57. pad=1
  58. activation=relu
  59. [convolutional]
  60. filters=256
  61. size=3
  62. stride=1
  63. pad=1
  64. activation=relu
  65. [maxpool]
  66. size=2
  67. stride=2
  68. [convolutional]
  69. filters=512
  70. size=3
  71. stride=1
  72. pad=1
  73. activation=relu
  74. [convolutional]
  75. filters=512
  76. size=3
  77. stride=1
  78. pad=1
  79. activation=relu
  80. [convolutional]
  81. filters=512
  82. size=3
  83. stride=1
  84. pad=1
  85. activation=relu
  86. [maxpool]
  87. size=2
  88. stride=2
  89. [convolutional]
  90. filters=512
  91. size=3
  92. stride=1
  93. pad=1
  94. activation=relu
  95. [convolutional]
  96. filters=512
  97. size=3
  98. stride=1
  99. pad=1
  100. activation=relu
  101. [convolutional]
  102. filters=512
  103. size=3
  104. stride=1
  105. pad=1
  106. activation=relu
  107. [maxpool]
  108. size=2
  109. stride=2
  110. [connected]
  111. output=4096
  112. activation=relu
  113. [dropout]
  114. probability=.5
  115. [connected]
  116. output=4096
  117. activation=relu
  118. [dropout]
  119. probability=.5
  120. [connected]
  121. output=1000
  122. activation=linear
  123. [softmax]
  124. groups=1
  125. [cost]
  126. type=sse