stouputils.data_science.data_processing.image.rotation module

rotate_image(image: ndarray[Any, dtype[Any]], angle: float | int, ignore_dtype: bool = False) → ndarray[Any, dtype[Any]]

Rotate an image by a given angle.

Parameters:

• image (NDArray[Any]) – Image to rotate

• angle (float|int) – Angle in degrees to rotate the image (between -360 and 360)

• ignore_dtype (bool) – Ignore the dtype check

Returns:

Rotated image

Return type:

NDArray[Any]

>>> ## Basic tests
>>> image = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> rotate_image(image.astype(np.int16), 90).tolist()
[[3, 6, 9], [2, 5, 8], [1, 4, 7]]

>>> rotate_image(image.astype(np.float32), 90).tolist()
[[3.0, 6.0, 9.0], [2.0, 5.0, 8.0], [1.0, 4.0, 7.0]]

>>> rotate_image(image.astype(np.uint8), 45).tolist()
[[1, 4, 4], [2, 5, 8], [2, 6, 5]]

>>> rotate_image(image.astype(np.float32), 45).tolist()
[[1.1875, 3.5625, 3.5625], [2.125, 5.0, 7.875], [2.375, 6.4375, 4.75]]

>>> ## Test invalid inputs
>>> rotate_image([1,2,3], 90)
Traceback (most recent call last):
        ...
AssertionError: Image must be a numpy array

>>> rotate_image(np.array([1,2,3]), 90)
Traceback (most recent call last):
        ...
AssertionError: Image must have at least 2 dimensions

>>> rotate_image(image.astype(np.uint8), "90")
Traceback (most recent call last):
        ...
AssertionError: Angle must be a number, got <class 'str'>

>>> rotate_image(image.astype(np.uint8), 400)
Traceback (most recent call last):
        ...
AssertionError: Angle must be between -360 and 360 degrees, got 400

>>> rotate_image(image.astype(np.int32), 90)
Traceback (most recent call last):
        ...
AssertionError: Image must be of type [<class 'numpy.uint8'>, <class 'numpy.int16'>, <class 'numpy.float32'>, <class 'numpy.float64'>]