Re-adding multiple windows_per_image for sampler_resize
@@ -26,6 +26,7 @@ class ResizeSampler(Layer, InputBatchQueueRunner):
shuffle_buffer=True):
self.reader = reader
+ self.windows_per_image = windows_per_image
self.shuffle = bool(shuffle_buffer)
Layer.__init__(self, name='input_buffer')
@@ -69,6 +70,7 @@ class ResizeSampler(Layer, InputBatchQueueRunner):
# for resize sampler the coordinates are not used
# simply use the spatial dims of the input image
all_coordinates = dummy_coordinates(image_id, image_shapes)
+ for _ in range(self.windows_per_image):
output_dict = {}
for name in list(data):
# prepare output dictionary keys
