From 725762b9377744856b6392dc0b7efd725bb115a6 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 13 Oct 2019 17:41:49 +0200 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index b14d67bb..ab1791cf 100644 --- a/train.py +++ b/train.py @@ -65,8 +65,8 @@ def train(): multi_scale = opt.multi_scale if multi_scale: - img_sz_min = 14 # round(img_size / 32 / 1.5) + 1 - img_sz_max = 27 # round(img_size / 32 * 1.5) - 1 + 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))