This commit is contained in:
Glenn Jocher 2019-11-27 15:50:29 -10:00
parent 82b62c9855
commit 9e9a6a1425
1 changed files with 2 additions and 2 deletions

View File

@ -63,8 +63,8 @@ def train():
# Initialize # Initialize
init_seeds() init_seeds()
if opt.multi_scale: if opt.multi_scale:
img_sz_min = 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) + 1 img_sz_max = round(img_size / 32 * 1.5)
img_size = img_sz_max * 32 # initiate with maximum multi_scale size img_size = img_sz_max * 32 # initiate with maximum multi_scale size
print('Using multi-scale %g - %g' % (img_sz_min * 32, img_size)) print('Using multi-scale %g - %g' % (img_sz_min * 32, img_size))