This commit is contained in:
Glenn Jocher 2019-08-04 02:50:35 +02:00
parent 1646b8de4b
commit ebf56ea725
1 changed files with 3 additions and 2 deletions

View File

@ -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)