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/yolo-coco.cfg |   57 ++++++++++++++++++++++++++++++++++++---------------------
 1 files changed, 36 insertions(+), 21 deletions(-)

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