def albumentations(self, img):
img = random_crop(img,
crop_height=64,
crop_width=64,
h_start=0,
w_start=0)
return resize(img, height=512, width=512)
def test_resize_different_height_and_width(target):
img = np.ones((100, 100), dtype=np.uint8)
img = convert_2d_to_target_format([img], target=target)
resized_img = FGeometric.resize(img, height=20, width=30)
height, width = resized_img.shape[:2]
assert height == 20
assert width == 30
if target == "image":
num_channels = resized_img.shape[2]
assert num_channels == 3
def test_resize_nearest_interpolation(target):
img = np.array([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4]],
dtype=np.uint8)
expected = np.array([[1, 1], [3, 3]], dtype=np.uint8)
img, expected = convert_2d_to_target_format([img, expected], target=target)
resized_img = FGeometric.resize(img, 2, 2, interpolation=cv2.INTER_NEAREST)
height, width = resized_img.shape[:2]
assert height == 2
assert width == 2
assert np.array_equal(resized_img, expected)
def test_resize_nearest_interpolation_float(target):
img = np.array([[0.1, 0.1, 0.1, 0.1], [0.2, 0.2, 0.2, 0.2],
[0.3, 0.3, 0.3, 0.3], [0.4, 0.4, 0.4, 0.4]],
dtype=np.float32)
expected = np.array([[0.1, 0.1], [0.3, 0.3]], dtype=np.float32)
img, expected = convert_2d_to_target_format([img, expected], target=target)
resized_img = FGeometric.resize(img, 2, 2, interpolation=cv2.INTER_NEAREST)
height, width = resized_img.shape[:2]
assert height == 2
assert width == 2
assert np.array_equal(resized_img, expected)
def test_resize_default_interpolation_float(target):
img = np.array([[0.1, 0.1, 0.1, 0.1], [0.2, 0.2, 0.2, 0.2],
[0.3, 0.3, 0.3, 0.3], [0.4, 0.4, 0.4, 0.4]],
dtype=np.float32)
expected = np.array([[0.15, 0.15], [0.35, 0.35]], dtype=np.float32)
img, expected = convert_2d_to_target_format([img, expected], target=target)
resized_img = FGeometric.resize(img, 2, 2)
height, width = resized_img.shape[:2]
assert height == 2
assert width == 2
assert_array_almost_equal_nulp(resized_img, expected)
def albumentations(self, img):
return resize(img, height=512, width=512)
