From f3aced12fcc41415851ffb463675930cea8dd086 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sun, 16 Sep 2018 02:01:19 +0000
Subject: [PATCH] 10 class weight

---
 cfg/tiny-yolo.cfg |   66 ++++++++++++++++++--------------
 1 files changed, 37 insertions(+), 29 deletions(-)

diff --git a/cfg/tiny-yolo.cfg b/cfg/tiny-yolo.cfg
index ac4b346..5580098 100644
--- a/cfg/tiny-yolo.cfg
+++ b/cfg/tiny-yolo.cfg
@@ -1,21 +1,21 @@
 [net]
 batch=64
-subdivisions=2
-height=448
-width=448
+subdivisions=8
+width=416
+height=416
 channels=3
 momentum=0.9
 decay=0.0005
+angle=0
+saturation = 1.5
+exposure = 1.5
+hue=.1
 
-saturation=.75
-exposure=.75
-hue = .1
-
-learning_rate=0.0005
+learning_rate=0.001
+max_batches = 120000
 policy=steps
-steps=200,400,600,800,20000,30000
-scales=2.5,2,2,2,.1,.1
-max_batches = 40000
+steps=-1,100,80000,100000
+scales=.1,10,.1,.1
 
 [convolutional]
 batch_normalize=1
@@ -87,7 +87,17 @@
 
 [maxpool]
 size=2
-stride=2
+stride=1
+
+[convolutional]
+batch_normalize=1
+filters=1024
+size=3
+stride=1
+pad=1
+activation=leaky
+
+###########
 
 [convolutional]
 batch_normalize=1
@@ -98,29 +108,27 @@
 activation=leaky
 
 [convolutional]
-batch_normalize=1
-size=3
+size=1
 stride=1
 pad=1
-filters=256
-activation=leaky
-
-[connected]
-output= 1470
+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=2
-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=1

--
Gitblit v1.10.0