From 9ff40baa4f0b005d50c4e3302b4103551160943f Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 07 Feb 2018 22:04:40 +0000
Subject: [PATCH] Fixes for training Yolo on small objects
---
src/detector.c | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/src/detector.c b/src/detector.c
index d79fbcc..4e22055 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -82,6 +82,7 @@
args.classes = classes;
args.jitter = jitter;
args.num_boxes = l.max_boxes;
+ args.small_object = l.small_object;
args.d = &buffer;
args.type = DETECTION_DATA;
args.threads = 4;// 8;
--
Gitblit v1.10.0