diff --git a/train.py b/train.py index 69299fcb..0038db97 100644 --- a/train.py +++ b/train.py @@ -63,8 +63,8 @@ def train(): # Initialize init_seeds() if opt.multi_scale: - img_sz_min = round(img_size / 32 / 1.5) - 1 - img_sz_max = round(img_size / 32 * 1.5) + 1 + img_sz_min = round(img_size / 32 / 1.5) + img_sz_max = round(img_size / 32 * 1.5) img_size = img_sz_max * 32 # initiate with maximum multi_scale size print('Using multi-scale %g - %g' % (img_sz_min * 32, img_size))