From 481d9d98abc8ef1225feac45d04a9514935832bf Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 6 May 2018 10:40:30 -0700
Subject: [PATCH] yolov3-tiny :fire: :fire: :fire:

---
 cfg/yolov3-tiny.cfg | 182 ++++++++++++++++++++++++++++++++++++++++++++
 cfg/yolov3.cfg      |   6 +-
 src/yolo_layer.c    |   2 +-
 3 files changed, 186 insertions(+), 4 deletions(-)
 create mode 100644 cfg/yolov3-tiny.cfg

diff --git a/cfg/yolov3-tiny.cfg b/cfg/yolov3-tiny.cfg
new file mode 100644
index 00000000..42c0fcf9
--- /dev/null
+++ b/cfg/yolov3-tiny.cfg
@@ -0,0 +1,182 @@
+[net]
+# Testing
+batch=1
+subdivisions=1
+# Training
+# batch=64
+# subdivisions=2
+width=416
+height=416
+channels=3
+momentum=0.9
+decay=0.0005
+angle=0
+saturation = 1.5
+exposure = 1.5
+hue=.1
+
+learning_rate=0.001
+burn_in=1000
+max_batches = 500200
+policy=steps
+steps=400000,450000
+scales=.1,.1
+
+[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=1
+
+[convolutional]
+batch_normalize=1
+filters=1024
+size=3
+stride=1
+pad=1
+activation=leaky
+
+###########
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+batch_normalize=1
+filters=512
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+size=1
+stride=1
+pad=1
+filters=255
+activation=linear
+
+
+
+[yolo]
+mask = 3,4,5
+anchors = 10,14,  23,27,  37,58,  81,82,  135,169,  344,319
+classes=80
+num=6
+jitter=.3
+ignore_thresh = .7
+truth_thresh = 1
+random=1
+
+[route]
+layers = -4
+
+[convolutional]
+batch_normalize=1
+filters=128
+size=1
+stride=1
+pad=1
+activation=leaky
+
+[upsample]
+stride=2
+
+[route]
+layers = -1, 8
+
+[convolutional]
+batch_normalize=1
+filters=256
+size=3
+stride=1
+pad=1
+activation=leaky
+
+[convolutional]
+size=1
+stride=1
+pad=1
+filters=255
+activation=linear
+
+[yolo]
+mask = 1,2,3
+anchors = 10,14,  23,27,  37,58,  81,82,  135,169,  344,319
+classes=80
+num=6
+jitter=.3
+ignore_thresh = .7
+truth_thresh = 1
+random=1
diff --git a/cfg/yolov3.cfg b/cfg/yolov3.cfg
index 5f3ab621..4a0ecc33 100644
--- a/cfg/yolov3.cfg
+++ b/cfg/yolov3.cfg
@@ -610,7 +610,7 @@ anchors = 10,13,  16,30,  33,23,  30,61,  62,45,  59,119,  116,90,  156,198,  37
 classes=80
 num=9
 jitter=.3
-ignore_thresh = .5
+ignore_thresh = .7
 truth_thresh = 1
 random=1
 
@@ -696,7 +696,7 @@ anchors = 10,13,  16,30,  33,23,  30,61,  62,45,  59,119,  116,90,  156,198,  37
 classes=80
 num=9
 jitter=.3
-ignore_thresh = .5
+ignore_thresh = .7
 truth_thresh = 1
 random=1
 
@@ -783,7 +783,7 @@ anchors = 10,13,  16,30,  33,23,  30,61,  62,45,  59,119,  116,90,  156,198,  37
 classes=80
 num=9
 jitter=.3
-ignore_thresh = .5
+ignore_thresh = .7
 truth_thresh = 1
 random=1
 
diff --git a/src/yolo_layer.c b/src/yolo_layer.c
index af714d46..c3380363 100644
--- a/src/yolo_layer.c
+++ b/src/yolo_layer.c
@@ -54,7 +54,7 @@ layer make_yolo_layer(int batch, int w, int h, int n, int total, int *mask, int
     l.delta_gpu = cuda_make_array(l.delta, batch*l.outputs);
 #endif
 
-    fprintf(stderr, "detection\n");
+    fprintf(stderr, "yolo\n");
     srand(0);
 
     return l;