updates
This commit is contained in:
parent
1646b8de4b
commit
ebf56ea725
5
train.py
5
train.py
|
@ -91,9 +91,10 @@ def train(cfg,
|
||||||
multi_scale = opt.multi_scale
|
multi_scale = opt.multi_scale
|
||||||
|
|
||||||
if multi_scale:
|
if multi_scale:
|
||||||
img_sz_min = 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)
|
img_sz_max = round(img_size / 32 * 1.5) - 1
|
||||||
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))
|
||||||
|
|
||||||
# Configure run
|
# Configure run
|
||||||
data_dict = parse_data_cfg(data)
|
data_dict = parse_data_cfg(data)
|
||||||
|
|
Loading…
Reference in New Issue