From c62b4f35aa2c59d7db0fd177affeed14b1ba4bcb Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 08 Sep 2016 07:04:39 +0000
Subject: [PATCH] adding coco models

---
 cfg/tiny-coco.cfg |  125 +++++++++++++++++++++++++++++++
 cfg/yolo-coco.cfg |   57 +++++++++-----
 2 files changed, 161 insertions(+), 21 deletions(-)

diff --git a/cfg/tiny-coco.cfg b/cfg/tiny-coco.cfg
new file mode 100644
index 0000000..e58c73a
--- /dev/null
+++ b/cfg/tiny-coco.cfg
@@ -0,0 +1,125 @@
+[net]
+batch=64
+subdivisions=2
+height=448
+width=448
+channels=3
+momentum=0.9
+decay=0.0005
+
+hue = .1
+saturation=.75
+exposure=.75
+
+learning_rate=0.0005
+policy=steps
+steps=200,400,600,800,100000,150000
+scales=2.5,2,2,2,.1,.1
+max_batches = 200000
+
+[convolutional]
+batch_normalize=1
+filters=16
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=32
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=64
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[maxpool]
+size=2
+stride=2
+
+[convolutional]
+batch_normalize=1
+size=3
+stride=1
+pad=1
+filters=1024
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+size=3
+stride=1
+pad=1
+filters=256
+activation=leaky
+
+[connected]
+output= 4655
+activation=linear
+
+[detection]
+classes=80
+coords=4
+rescore=1
+side=7
+num=3
+softmax=0
+sqrt=1
+jitter=.2
+
+object_scale=1
+noobject_scale=.5
+class_scale=1
+coord_scale=5
diff --git a/cfg/yolo-coco.cfg b/cfg/yolo-coco.cfg
index 0c13a31..ed3f2d6 100644
--- a/cfg/yolo-coco.cfg
+++ b/cfg/yolo-coco.cfg
@@ -7,21 +7,18 @@
 momentum=0.9
 decay=0.0005
 
-learning_rate=0.0001
-policy=steps
-steps=100,200,300,100000,150000
-scales=2.5,2,2,.1,.1
-max_batches = 300000
+hue = .1
+saturation=.75
+exposure=.75
 
-[crop]
-crop_width=448
-crop_height=448
-flip=0
-angle=0
-saturation = 1.5
-exposure = 1.5
+learning_rate=0.0005
+policy=steps
+steps=200,400,600,800,100000,150000
+scales=2.5,2,2,2,.1,.1
+max_batches = 200000
 
 [convolutional]
+batch_normalize=1
 filters=64
 size=7
 stride=2
@@ -33,6 +30,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=192
 size=3
 stride=1
@@ -44,6 +42,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=128
 size=1
 stride=1
@@ -51,6 +50,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=3
 stride=1
@@ -58,6 +58,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -65,6 +66,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -76,6 +78,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -83,6 +86,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -90,6 +94,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -97,6 +102,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -104,6 +110,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -111,6 +118,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -118,6 +126,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=256
 size=1
 stride=1
@@ -125,6 +134,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=3
 stride=1
@@ -132,6 +142,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -139,6 +150,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -150,6 +162,7 @@
 stride=2
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -157,6 +170,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
@@ -164,6 +178,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=512
 size=1
 stride=1
@@ -171,17 +186,17 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 filters=1024
 size=3
 stride=1
 pad=1
 activation=leaky
 
-
-#################################
-
+#######
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -189,6 +204,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=2
 pad=1
@@ -196,6 +212,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -203,6 +220,7 @@
 activation=leaky
 
 [convolutional]
+batch_normalize=1
 size=3
 stride=1
 pad=1
@@ -213,14 +231,11 @@
 size=3
 stride=1
 pad=1
-filters=192
+filters=256
 activation=leaky
 
-[dropout]
-probability=.5
-
 [connected]
-output= 4410
+output= 4655
 activation=linear
 
 [detection]
@@ -228,7 +243,7 @@
 coords=4
 rescore=1
 side=7
-num=2
+num=3
 softmax=0
 sqrt=1
 jitter=.2

--
Gitblit v1.10.0