def testResizeImageWorks(self, max_image_size, min_image_size, expected_shape,
expected_scale_factor):
# Construct image of size 4x2x3.
image = np.array([[[0, 0, 0], [1, 1, 1]], [[2, 2, 2], [3, 3, 3]],
[[4, 4, 4], [5, 5, 5]], [[6, 6, 6], [7, 7, 7]]],
dtype='uint8')
# Set up config.
config = delf_config_pb2.DelfConfig(
max_image_size=max_image_size, min_image_size=min_image_size)
resized_image, scale_factor = extractor.ResizeImage(image, config)
self.assertAllEqual(resized_image.shape, expected_shape)
self.assertAllClose(scale_factor, expected_scale_factor)
def testResizeImageRoundingWorks(self, max_image_size, min_image_size,
resize_factor, square_output,
expected_shape, expected_scale_factors):
# Construct image of size 3x2x3.
image = np.array([[[0, 0, 0], [1, 1, 1]], [[2, 2, 2], [3, 3, 3]],
[[4, 4, 4], [5, 5, 5]]],
dtype='uint8')
# Set up config.
config = delf_config_pb2.DelfConfig(max_image_size=max_image_size,
min_image_size=min_image_size)
resized_image, scale_factors = extractor.ResizeImage(
image, config, resize_factor, square_output)
self.assertAllEqual(resized_image.shape, expected_shape)
self.assertAllClose(scale_factors, expected_scale_factors)