From b36512ea2e7e6d2ec36a3241c1bf751e4e074fe1 Mon Sep 17 00:00:00 2001
From: vinjn <vinjn.z@gmail.com>
Date: Sat, 26 May 2018 13:42:53 +0000
Subject: [PATCH] classifier.c - add the awesome training chart and make sure "top" is not bigger than "classes" in datacfg file.

---
 cfg/darknet.cfg |   30 ++++++++++++------------------
 1 files changed, 12 insertions(+), 18 deletions(-)

diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg
index 53d1ec9..60b939a 100644
--- a/cfg/darknet.cfg
+++ b/cfg/darknet.cfg
@@ -1,24 +1,17 @@
 [net]
 batch=128
 subdivisions=1
-height=256
-width=256
+height=224
+width=224
 channels=3
 momentum=0.9
 decay=0.0005
+max_crop=320
 
 learning_rate=0.1
 policy=poly
 power=4
-max_batches=500000
-
-[crop]
-crop_height=224
-crop_width=224
-flip=1
-angle=0
-saturation=1
-exposure=1
+max_batches=1600000
 
 [convolutional]
 batch_normalize=1
@@ -91,6 +84,7 @@
 [maxpool]
 size=2
 stride=2
+padding=1
 
 [convolutional]
 batch_normalize=1
@@ -100,15 +94,15 @@
 pad=1
 activation=leaky
 
-[avgpool]
-
-[dropout]
-probability=.5
-
-[connected]
-output=1000
+[convolutional]
+filters=1000
+size=1
+stride=1
+pad=1
 activation=leaky
 
+[avgpool]
+
 [softmax]
 groups=1
 

--
Gitblit v1.10.0