Cityscapes

config/yoeo-rev-7-anchor.cfg

@@ -0,0 +1,337 @@
+[net]
+# Testing
+#batch=1
+#subdivisions=1
+# Training
+batch=64
+subdivisions=8
+width=416
+height=416
+channels=3
+momentum=0.9
+decay=0.0005
+angle=0
+saturation = 1.5
+exposure = 1.5
+hue=.1
+
+learning_rate=0.001
+burn_in=100
+max_batches = 4000
+policy=steps
+steps=50000,60000
+scales=.1,.1
+
+####
+# Like YOEO rev 2 but with deeper skip connections
+####
+
+[convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=2
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=3
+stride=2
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers=-1
+groups=2
+group_id=1
+
+[convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -1,-2
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -6,-1
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers=-1
+groups=2
+group_id=1
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -1,-2
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -6,-1
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers=-1
+groups=2
+group_id=1
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -1,-2
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -6,-1
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
+###########
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+size=1
+stride=1
+pad=1
+filters=39
+activation=linear
+
+[yolo]
+mask = 3,4,5
+anchors = 7, 11, 15,24, 25, 52, 48,65, 90,132, 88,237
+classes=8
+num=6
+jitter=.3
+ignore_thresh = .7
+truth_thresh = 1
+random=1
+
+[route]
+layers = -4
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[upsample]
+stride=2
+
+[route]
+layers = -1, 24
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+size=1
+stride=1
+pad=1
+filters=39
+activation=linear
+
+[yolo]
+mask = 0,1,2
+anchors = 7, 11, 15,24, 25, 52, 48,65, 90,132, 88,237
+classes=8
+classes=8
+num=6
+jitter=.3
+ignore_thresh = .7
+truth_thresh = 1
+random=1
+
+[route]
+layers = 18
+
+[upsample]
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -1, 10
+
+[upsample]
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -1, 2
+
+[upsample]
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[route]
+layers = -1, 0
+
+[upsample]
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=6
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[seg]
+classes=5

config/yoeo-rev-7.cfg

@@ -19,8 +19,8 @@ learning_rate=0.001
 burn_in=100
 max_batches = 4000
 policy=steps
-steps=50000,60000
-scales=.1,.1
+steps=15000,30000
+scales=.2,.1
 
 ####
 # Like YOEO rev 2 but with deeper skip connections
@@ -213,15 +213,13 @@ activation=leaky
 size=1
 stride=1
 pad=1
-filters=24
+filters=39
 activation=linear
 
-
-
 [yolo]
 mask = 3,4,5
 anchors = 17, 32, 13,171, 37, 67, 30,224, 69,112, 116,212
-classes=3
+classes=8
 num=6
 jitter=.3
 ignore_thresh = .7
@@ -257,13 +255,13 @@ activation=leaky
 size=1
 stride=1
 pad=1
-filters=24
+filters=39
 activation=linear
 
 [yolo]
 mask = 0,1,2
 anchors = 17, 32, 13,171, 37, 67, 30,224, 69,112, 116,212
-classes=3
+classes=8
 num=6
 jitter=.3
 ignore_thresh = .7
@@ -328,11 +326,11 @@ activation=leaky
 
 [convolutional]
 batch_normalize=1
-filters=3
+filters=6
 size=1
 stride=1
 pad=1
 activation=leaky
 
 [seg]
-classes=2
+classes=5