def test_extract_patch_for_padding(self):
"""Test patch extraction which is not completely inside the original image, hence needing padding."""
input_image = Image(np.random.rand(100, 71, 3))
patch_size = 10
"""Test with patch which hangs outside the image on the top left."""
patch_center_x = 3
patch_center_y = 4
computed_patch = input_image.extract_patch(patch_center_x,
patch_center_y, patch_size)
# check the patch dimensions
self.assertEqual(computed_patch.width, patch_size)
self.assertEqual(computed_patch.height, patch_size)
# check for zeros in area out of bounds
np.testing.assert_allclose(computed_patch.value_array[:1, :], 0)
np.testing.assert_allclose(computed_patch.value_array[:, :2], 0)
# check the patch contents with the original input
np.testing.assert_allclose(
computed_patch.value_array[1:, 2:],
input_image.value_array[:patch_center_y +
(patch_size + 1) // 2, :patch_center_x +
(patch_size + 1) // 2, ],
)
"""Test with patch which hangs outside the image on bottom right."""
patch_center_x = 70
patch_center_y = 96
computed_patch = input_image.extract_patch(patch_center_x,
patch_center_y, patch_size)
# check the patch dimensions
self.assertEqual(computed_patch.width, patch_size)
self.assertEqual(computed_patch.height, patch_size)
# check for zeros in area out of bounds
np.testing.assert_allclose(computed_patch.value_array[-1:, :], 0)
np.testing.assert_allclose(computed_patch.value_array[:, -4:], 0)
# check the patch contents with the original input
np.testing.assert_allclose(
computed_patch.value_array[:-1, :-4],
input_image.value_array[patch_center_y - patch_size // 2:,
patch_center_x - patch_size // 2:, ],
)
def test_extract_patch_fully_inside(self):
"""Test patch extraction which is fully inside the original image."""
input_image = Image(np.random.rand(100, 71, 3))
patch_center_x = 21
patch_center_y = 22
"""Test with even patch size."""
patch_size = 10
computed_patch = input_image.extract_patch(patch_center_x,
patch_center_y, patch_size)
# check the patch dimensions
self.assertEqual(computed_patch.width, patch_size)
self.assertEqual(computed_patch.height, patch_size)
# check the patch contents
np.testing.assert_allclose(
computed_patch.value_array,
input_image.value_array[patch_center_y -
patch_size // 2:patch_center_y +
(patch_size + 1) // 2, patch_center_x -
patch_size // 2:patch_center_x +
(patch_size + 1) // 2, ],
)
"""Test with odd patch size."""
patch_size = 11
computed_patch = input_image.extract_patch(patch_center_x,
patch_center_y, patch_size)
# check the patch dimensions
self.assertEqual(computed_patch.width, patch_size)
self.assertEqual(computed_patch.height, patch_size)
# check the patch contents
np.testing.assert_allclose(
computed_patch.value_array,
input_image.value_array[patch_center_y -
patch_size // 2:patch_center_y +
(patch_size + 1) // 2, patch_center_x -
patch_size // 2:patch_center_x +
(patch_size + 1) // 2, ],
)