some more stuff

cfg/darknet.cfg

@@ -11,9 +11,10 @@ max_crop=320
 learning_rate=0.1
 policy=poly
 power=4
-max_batches=500000
+max_batches=1600000
 
 [convolutional]
+batch_normalize=1
 filters=16
 size=3
 stride=1
@@ -25,6 +26,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=32
 size=3
 stride=1
@@ -36,6 +38,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=3
 stride=1
@@ -47,6 +50,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=3
 stride=1
@@ -58,6 +62,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -69,6 +74,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -80,18 +86,22 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
 pad=1
 activation=leaky
 
-[avgpool]
-
-[connected]
-output=1000
+[convolutional]
+filters=1000
+size=1
+stride=1
+pad=1
 activation=leaky
 
+[avgpool]
+
 [softmax]
 groups=1
 

cfg/extraction.cfg

@@ -1,26 +1,20 @@
 [net]
 batch=128
 subdivisions=1
-height=256
-width=256
+height=224
+width=224
+max_crop=320
 channels=3
 momentum=0.9
 decay=0.0005
 
-learning_rate=0.5
+learning_rate=0.1
 policy=poly
-power=6
-max_batches=500000
-
-[crop]
-crop_height=224
-crop_width=224
-flip=1
-saturation=1
-exposure=1
-angle=0
+power=4
+max_batches=1600000
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=7
 stride=2
@@ -32,6 +26,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=192
 size=3
 stride=1
@@ -43,6 +38,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=1
 stride=1
@@ -50,6 +46,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -57,6 +54,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -64,6 +62,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -75,6 +74,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -82,6 +82,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -89,6 +90,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -96,6 +98,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -103,6 +106,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -110,6 +114,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -117,6 +122,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -124,6 +130,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -131,6 +138,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -138,6 +146,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -149,6 +158,7 @@ size=2
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -156,6 +166,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -163,6 +174,7 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -170,18 +182,22 @@ pad=1
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
 pad=1
 activation=leaky
 
+[convolutional]
+filters=1000
+size=1
+stride=1
+pad=1
+activation=leaky
+
 [avgpool]
 
-[connected]
-output=1000
-activation=leaky
-
 [softmax]
 groups=1