From 5d8cb13190718a094d45c34a41b2d40693bad22e Mon Sep 17 00:00:00 2001
From: Basuki74 <73452927+Basuki74@users.noreply.github.com>
Date: Wed, 5 Jan 2022 16:00:36 +0700
Subject: [PATCH] Update yolov4-custom.cfg

---
 cfg/yolov4-custom.cfg | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/cfg/yolov4-custom.cfg b/cfg/yolov4-custom.cfg
index 32516bdc120..f2279141e86 100644
--- a/cfg/yolov4-custom.cfg
+++ b/cfg/yolov4-custom.cfg
@@ -5,8 +5,8 @@
 # Training
 batch=64
 subdivisions=16
-width=608
-height=608
+width=416
+height=416
 channels=3
 momentum=0.949
 decay=0.0005
@@ -17,9 +17,9 @@ hue=.1
 
 learning_rate=0.001
 burn_in=1000
-max_batches = 500500
+max_batches = 2000
 policy=steps
-steps=400000,450000
+steps=1600,1800
 scales=.1,.1
 
 #cutmix=1
@@ -960,14 +960,14 @@ activation=leaky
 size=1
 stride=1
 pad=1
-filters=255
+filters=18
 activation=linear
 
 
 [yolo]
 mask = 0,1,2
 anchors = 12, 16, 19, 36, 40, 28, 36, 75, 76, 55, 72, 146, 142, 110, 192, 243, 459, 401
-classes=80
+classes=1
 num=9
 jitter=.3
 ignore_thresh = .7
@@ -1048,14 +1048,14 @@ activation=leaky
 size=1
 stride=1
 pad=1
-filters=255
+filters=18
 activation=linear
 
 
 [yolo]
 mask = 3,4,5
 anchors = 12, 16, 19, 36, 40, 28, 36, 75, 76, 55, 72, 146, 142, 110, 192, 243, 459, 401
-classes=80
+classes=1
 num=9
 jitter=.3
 ignore_thresh = .7
@@ -1136,14 +1136,14 @@ activation=leaky
 size=1
 stride=1
 pad=1
-filters=255
+filters=18
 activation=linear
 
 
 [yolo]
 mask = 6,7,8
 anchors = 12, 16, 19, 36, 40, 28, 36, 75, 76, 55, 72, 146, 142, 110, 192, 243, 459, 401
-classes=80
+classes=1
 num=9
 jitter=.3
 ignore_thresh = .7