1
0

vgg-16.cfg 1.4 KB

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