@@ -15,6 +15,7 @@ class MultiScaleSampler(Sampler):
first_bs,
divided_factor=32,
is_training=True,
shuffle=True,
seed=None):
"""
multi scale samper
...
...
@@ -23,7 +24,7 @@ class MultiScaleSampler(Sampler):
scales(list): several scales for image resolution
first_bs(int): batch size for the first scale in scales
divided_factor(int): ImageNet models down-sample images by a factor, ensure that width and height dimensions are multiples are multiple of devided_factor.
is_training(boolean): mode
is_training(boolean): mode
"""
# min. and max. spatial dimensions
self.data_source=data_source
...
...
@@ -47,7 +48,7 @@ class MultiScaleSampler(Sampler):