From ebf56ea725b3ab0f7237c540f623a4c2db9ca843 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 4 Aug 2019 02:50:35 +0200
Subject: [PATCH] updates

---
 train.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 512e208c..c1175620 100644
--- a/train.py
+++ b/train.py
@@ -91,9 +91,10 @@ def train(cfg,
     multi_scale = opt.multi_scale
 
     if multi_scale:
-        img_sz_min = round(img_size / 32 / 1.5)
-        img_sz_max = round(img_size / 32 * 1.5)
+        img_sz_min = round(img_size / 32 / 1.5) + 1
+        img_sz_max = round(img_size / 32 * 1.5) - 1
         img_size = img_sz_max * 32  # initiate with maximum multi_scale size
+        print('Using multi-scale %g - %g' % (img_sz_min * 32, img_size))
 
     # Configure run
     data_dict = parse_data_cfg(data)