🔥 🔥 yolo v2 🔥 🔥

cfg/coco.data

@@ -0,0 +1,8 @@
+classes= 80
+train  = /home/pjreddie/data/coco/trainvalno5k.txt
+valid  = coco_testdev
+#valid = data/coco_val_5k.list
+names = data/coco.names
+backup = /home/pjreddie/backup/
+eval=coco
+

cfg/voc.data

@@ -0,0 +1,6 @@
+classes= 20
+train  = /home/pjreddie/data/voc/train.txt
+valid  = /home/pjreddie/data/voc/2007_test.txt
+names = data/pascal.names
+backup = /home/pjreddie/backup/
+

cfg/yolo.cfg

@@ -1,26 +1,27 @@
 [net]
-batch=1
-subdivisions=1
-height=448
-width=448
+batch=64
+subdivisions=8
+height=416
+width=416
 channels=3
 momentum=0.9
 decay=0.0005
-saturation=1.5
-exposure=1.5
+angle=0
+saturation = 1.5
+exposure = 1.5
 hue=.1
 
-learning_rate=0.0005
+learning_rate=0.001
+max_batches = 120000
 policy=steps
-steps=200,400,600,20000,30000
-scales=2.5,2,2,.1,.1
-max_batches = 40000
+steps=-1,100,80000,100000
+scales=.1,10,.1,.1
 
 [convolutional]
 batch_normalize=1
-filters=64
-size=7
-stride=2
+filters=32
+size=3
+stride=1
 pad=1
 activation=leaky
 
@@ -30,7 +31,7 @@ stride=2
 
 [convolutional]
 batch_normalize=1
-filters=192
+filters=64
 size=3
 stride=1
 pad=1
@@ -43,30 +44,22 @@ stride=2
 [convolutional]
 batch_normalize=1
 filters=128
-size=1
-stride=1
-pad=1
-activation=leaky
-
-[convolutional]
-batch_normalize=1
-filters=256
 size=3
 stride=1
 pad=1
 activation=leaky
 
 [convolutional]
 batch_normalize=1
-filters=256
+filters=64
 size=1
 stride=1
 pad=1
 activation=leaky
 
 [convolutional]
 batch_normalize=1
-filters=512
+filters=128
 size=3
 stride=1
 pad=1
@@ -79,27 +72,31 @@ stride=2
 [convolutional]
 batch_normalize=1
 filters=256
-size=1
+size=3
 stride=1
 pad=1
 activation=leaky
 
 [convolutional]
 batch_normalize=1
-filters=512
-size=3
+filters=128
+size=1
 stride=1
 pad=1
 activation=leaky
 
 [convolutional]
 batch_normalize=1
 filters=256
-size=1
+size=3
 stride=1
 pad=1
 activation=leaky
 
+[maxpool]
+size=2
+stride=2
+
 [convolutional]
 batch_normalize=1
 filters=512
@@ -140,13 +137,9 @@ stride=1
 pad=1
 activation=leaky
 
-[convolutional]
-batch_normalize=1
-filters=512
-size=1
-stride=1
-pad=1
-activation=leaky
+[maxpool]
+size=2
+stride=2
 
 [convolutional]
 batch_normalize=1
@@ -156,10 +149,6 @@ stride=1
 pad=1
 activation=leaky
 
-[maxpool]
-size=2
-stride=2
-
 [convolutional]
 batch_normalize=1
 filters=512
@@ -192,6 +181,7 @@ stride=1
 pad=1
 activation=leaky
 
+
 #######
 
 [convolutional]
@@ -202,14 +192,6 @@ pad=1
 filters=1024
 activation=leaky
 
-[convolutional]
-batch_normalize=1
-size=3
-stride=2
-pad=1
-filters=1024
-activation=leaky
-
 [convolutional]
 batch_normalize=1
 size=3
@@ -218,6 +200,15 @@ pad=1
 filters=1024
 activation=leaky
 
+[route]
+layers=-9
+
+[reorg]
+stride=2
+
+[route]
+layers=-1,-3
+
 [convolutional]
 batch_normalize=1
 size=3
@@ -226,32 +217,28 @@ pad=1
 filters=1024
 activation=leaky
 
-[local]
-size=3
+[convolutional]
+size=1
 stride=1
 pad=1
-filters=256
-activation=leaky
-
-[dropout]
-probability=.5
-
-[connected]
-output= 1715
+filters=425
 activation=linear
 
-[detection]
-classes=20
+[region]
+anchors = 0.738768,0.874946,  2.42204,2.65704,  4.30971,7.04493,  10.246,4.59428,  12.6868,11.8741
+bias_match=1
+classes=80
 coords=4
-rescore=1
-side=7
-num=3
-softmax=0
-sqrt=1
+num=5
+softmax=1
 jitter=.2
+rescore=1
 
-object_scale=1
-noobject_scale=.5
+object_scale=5
+noobject_scale=1
 class_scale=1
-coord_scale=5
+coord_scale=1
 
+absolute=1
+thresh = .6
+random=0