stouputils 1.14.0__py3-none-any.whl → 1.14.2__py3-none-any.whl

stouputils/data_science/data_processing/image/histogram_equalization.py

@@ -1,123 +1,123 @@
-
-# pyright: reportUnusedImport=false
-# ruff: noqa: F401
-
-# Imports
-from typing import Literal
-
-from .common import Any, NDArray, check_image, cv2, np
-
-# Constants
-VALID_SPACES: list[str] = ["lab", "ycbcr", "hsv"]
-
-# Color space conversion constants
-COLOR_SPACE_CONSTANTS: dict[str, tuple[int, int, int]] = {
-    "lab":    (cv2.COLOR_BGR2LAB, cv2.COLOR_LAB2BGR, 0),      # L channel index is 0
-    "ycbcr":  (cv2.COLOR_BGR2YCrCb, cv2.COLOR_YCrCb2BGR, 0),  # Y channel index is 0
-    "hsv":    (cv2.COLOR_BGR2HSV, cv2.COLOR_HSV2BGR, 2),      # V channel index is 2
-}
-
-
-# Functions
-def histogram_equalization_image(
-	image: NDArray[Any],
-	color_space: Literal["lab", "ycbcr", "hsv"] = "lab",
-	ignore_dtype: bool = False,
-) -> NDArray[Any]:
-	""" Apply standard histogram equalization to an image.
-
-	Histogram equalization improves the contrast in images by stretching
-	the intensity range to utilize the full range of intensity values.
-
-	Args:
-		image           (NDArray[Any]):  Image to apply histogram equalization to
-		color_space     (str):           Color space to use for equalization ("lab", "ycbcr", or "hsv")
-			"lab":   CIELab color space (perceptually uniform, best visual fidelity)
-			"ycbcr": YCbCr color space (fast, good balance)
-			"hsv":   HSV color space (intuitive, may cause color shifts)
-		ignore_dtype    (bool):        Ignore the dtype check
-	Returns:
-		NDArray[Any]: Image with histogram equalization applied
-
-	>>> ## Basic tests
-	>>> image = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
-	>>> histogram_equalization_image(image.astype(np.uint8)).tolist()
-	[[0, 32, 64], [96, 128, 159], [191, 223, 255]]
-
-	>>> img = np.full((5,5), 128, dtype=np.uint8)
-	>>> img[1:3, 1:3] = 200  # Create a bright region
-	>>> histogram_equalization_image(img).tolist()
-	[[0, 0, 0, 0, 0], [0, 255, 255, 0, 0], [0, 255, 255, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]]
-
-	>>> rgb = np.full((3,3,3), 128, dtype=np.uint8)
-	>>> rgb[1, 1, :] = 50  # Create a dark region
-	>>> equalized_rgb = histogram_equalization_image(rgb)
-	>>> bool(np.std(equalized_rgb) > np.std(rgb))  # Should enhance contrast
-	True
-	>>> equalized_rgb.tolist()
-	[[[255, 255, 255], [255, 255, 255], [255, 255, 255]], [[255, 255, 255], [0, 0, 0], [255, 255, 255]], [[255, 255, 255], [255, 255, 255], [255, 255, 255]]]
-
-	>>> ## Test each color space
-	>>> test_img = np.zeros((20, 20, 3), dtype=np.uint8)
-	>>> test_img[5:15, 5:15] = 200  # Add contrast region
-
-	>>> # Test LAB color space
-	>>> lab_result = histogram_equalization_image(test_img, color_space="lab")
-	>>> isinstance(lab_result, np.ndarray) and lab_result.shape == test_img.shape
-	True
-	>>> bool(np.std(lab_result) > np.std(test_img))  # Verify contrast enhancement
-	True
-
-	>>> # Test YCbCr color space
-	>>> ycbcr_result = histogram_equalization_image(test_img, color_space="ycbcr")
-	>>> isinstance(ycbcr_result, np.ndarray) and ycbcr_result.shape == test_img.shape
-	True
-	>>> bool(np.std(ycbcr_result) > np.std(test_img))  # Verify contrast enhancement
-	True
-
-	>>> # Test HSV color space
-	>>> hsv_result = histogram_equalization_image(test_img, color_space="hsv")
-	>>> isinstance(hsv_result, np.ndarray) and hsv_result.shape == test_img.shape
-	True
-	>>> bool(np.std(hsv_result) > np.std(test_img))  # Verify contrast enhancement
-	True
-
-	>>> ## Test invalid inputs
-	>>> histogram_equalization_image("not an image")
-	Traceback (most recent call last):
-		...
-	AssertionError: Image must be a numpy array
-
-	>>> histogram_equalization_image(rgb, "invalid_space")
-	Traceback (most recent call last):
-		...
-	AssertionError: color_space must be one of: lab, ycbcr, hsv
-	"""  # noqa: E501
-	# Check input data
-	check_image(image, ignore_dtype=ignore_dtype)
-	lowered_color_space = color_space.lower()
-	assert lowered_color_space in VALID_SPACES, f"color_space must be one of: {', '.join(VALID_SPACES)}"
-
-	# Handle different image types
-	if len(image.shape) == 2:
-		# Grayscale image - just apply histogram equalization directly
-		return cv2.equalizeHist(image)
-	else:
-		# Color image - apply equalization based on selected color space
-		convert_to, convert_from, channel_idx = COLOR_SPACE_CONSTANTS[lowered_color_space]
-
-		# Convert to target color space
-		converted: NDArray[Any] = cv2.cvtColor(image, convert_to)
-
-		# Split channels
-		channels: list[NDArray[Any]] = list(cv2.split(converted))
-
-		# Apply histogram equalization to the appropriate channel
-		channels[channel_idx] = cv2.equalizeHist(channels[channel_idx])
-
-		# Merge channels
-		result: NDArray[Any] = cv2.merge(channels)
-
-		# Convert back to BGR
-		return cv2.cvtColor(result, convert_from)
-
+
+# pyright: reportUnusedImport=false
+# ruff: noqa: F401
+
+# Imports
+from typing import Literal
+
+from .common import Any, NDArray, check_image, cv2, np
+
+# Constants
+VALID_SPACES: list[str] = ["lab", "ycbcr", "hsv"]
+
+# Color space conversion constants
+COLOR_SPACE_CONSTANTS: dict[str, tuple[int, int, int]] = {
+    "lab":    (cv2.COLOR_BGR2LAB, cv2.COLOR_LAB2BGR, 0),      # L channel index is 0
+    "ycbcr":  (cv2.COLOR_BGR2YCrCb, cv2.COLOR_YCrCb2BGR, 0),  # Y channel index is 0
+    "hsv":    (cv2.COLOR_BGR2HSV, cv2.COLOR_HSV2BGR, 2),      # V channel index is 2
+}
+
+
+# Functions
+def histogram_equalization_image(
+	image: NDArray[Any],
+	color_space: Literal["lab", "ycbcr", "hsv"] = "lab",
+	ignore_dtype: bool = False,
+) -> NDArray[Any]:
+	""" Apply standard histogram equalization to an image.
+
+	Histogram equalization improves the contrast in images by stretching
+	the intensity range to utilize the full range of intensity values.
+
+	Args:
+		image           (NDArray[Any]):  Image to apply histogram equalization to
+		color_space     (str):           Color space to use for equalization ("lab", "ycbcr", or "hsv")
+			"lab":   CIELab color space (perceptually uniform, best visual fidelity)
+			"ycbcr": YCbCr color space (fast, good balance)
+			"hsv":   HSV color space (intuitive, may cause color shifts)
+		ignore_dtype    (bool):        Ignore the dtype check
+	Returns:
+		NDArray[Any]: Image with histogram equalization applied
+
+	>>> ## Basic tests
+	>>> image = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
+	>>> histogram_equalization_image(image.astype(np.uint8)).tolist()
+	[[0, 32, 64], [96, 128, 159], [191, 223, 255]]
+
+	>>> img = np.full((5,5), 128, dtype=np.uint8)
+	>>> img[1:3, 1:3] = 200  # Create a bright region
+	>>> histogram_equalization_image(img).tolist()
+	[[0, 0, 0, 0, 0], [0, 255, 255, 0, 0], [0, 255, 255, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]]
+
+	>>> rgb = np.full((3,3,3), 128, dtype=np.uint8)
+	>>> rgb[1, 1, :] = 50  # Create a dark region
+	>>> equalized_rgb = histogram_equalization_image(rgb)
+	>>> bool(np.std(equalized_rgb) > np.std(rgb))  # Should enhance contrast
+	True
+	>>> equalized_rgb.tolist()
+	[[[255, 255, 255], [255, 255, 255], [255, 255, 255]], [[255, 255, 255], [0, 0, 0], [255, 255, 255]], [[255, 255, 255], [255, 255, 255], [255, 255, 255]]]
+
+	>>> ## Test each color space
+	>>> test_img = np.zeros((20, 20, 3), dtype=np.uint8)
+	>>> test_img[5:15, 5:15] = 200  # Add contrast region
+
+	>>> # Test LAB color space
+	>>> lab_result = histogram_equalization_image(test_img, color_space="lab")
+	>>> isinstance(lab_result, np.ndarray) and lab_result.shape == test_img.shape
+	True
+	>>> bool(np.std(lab_result) > np.std(test_img))  # Verify contrast enhancement
+	True
+
+	>>> # Test YCbCr color space
+	>>> ycbcr_result = histogram_equalization_image(test_img, color_space="ycbcr")
+	>>> isinstance(ycbcr_result, np.ndarray) and ycbcr_result.shape == test_img.shape
+	True
+	>>> bool(np.std(ycbcr_result) > np.std(test_img))  # Verify contrast enhancement
+	True
+
+	>>> # Test HSV color space
+	>>> hsv_result = histogram_equalization_image(test_img, color_space="hsv")
+	>>> isinstance(hsv_result, np.ndarray) and hsv_result.shape == test_img.shape
+	True
+	>>> bool(np.std(hsv_result) > np.std(test_img))  # Verify contrast enhancement
+	True
+
+	>>> ## Test invalid inputs
+	>>> histogram_equalization_image("not an image")
+	Traceback (most recent call last):
+		...
+	AssertionError: Image must be a numpy array
+
+	>>> histogram_equalization_image(rgb, "invalid_space")
+	Traceback (most recent call last):
+		...
+	AssertionError: color_space must be one of: lab, ycbcr, hsv
+	"""  # noqa: E501
+	# Check input data
+	check_image(image, ignore_dtype=ignore_dtype)
+	lowered_color_space = color_space.lower()
+	assert lowered_color_space in VALID_SPACES, f"color_space must be one of: {', '.join(VALID_SPACES)}"
+
+	# Handle different image types
+	if len(image.shape) == 2:
+		# Grayscale image - just apply histogram equalization directly
+		return cv2.equalizeHist(image)
+	else:
+		# Color image - apply equalization based on selected color space
+		convert_to, convert_from, channel_idx = COLOR_SPACE_CONSTANTS[lowered_color_space]
+
+		# Convert to target color space
+		converted: NDArray[Any] = cv2.cvtColor(image, convert_to)
+
+		# Split channels
+		channels: list[NDArray[Any]] = list(cv2.split(converted))
+
+		# Apply histogram equalization to the appropriate channel
+		channels[channel_idx] = cv2.equalizeHist(channels[channel_idx])
+
+		# Merge channels
+		result: NDArray[Any] = cv2.merge(channels)
+
+		# Convert back to BGR
+		return cv2.cvtColor(result, convert_from)
+

stouputils/data_science/data_processing/image/invert.py

@@ -1,64 +1,64 @@
-
-# Imports
-from .common import Any, NDArray, check_image, np
-
-
-# Function
-def invert_image(image: NDArray[Any], ignore_dtype: bool = False) -> NDArray[Any]:
-	""" Invert the colors of an image.
-
-	This function inverts the colors of the input image by subtracting each pixel value
-	from the maximum possible value based on the image type.
-
-	Args:
-		image         (NDArray[Any]):       Input image as a NumPy array.
-		ignore_dtype  (bool):             Ignore the dtype check.
-	Returns:
-		NDArray[Any]: Image with inverted colors.
-
-	>>> ## Basic tests
-	>>> image = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]], dtype=np.uint8)
-	>>> inverted = invert_image(image)
-	>>> inverted.tolist()
-	[[245, 235, 225], [215, 205, 195], [185, 175, 165]]
-
-	>>> # Test with floating point image
-	>>> float_img = np.array([[0.1, 0.2], [0.3, 0.4]], dtype=np.float32)
-	>>> [round(float(x), 1) for x in invert_image(float_img).flatten()]
-	[0.9, 0.8, 0.7, 0.6]
-
-	>>> # Test with RGB image
-	>>> rgb = np.zeros((2, 2, 3), dtype=np.uint8)
-	>>> rgb[0, 0] = [255, 0, 0]  # Red pixel
-	>>> inverted_rgb = invert_image(rgb)
-	>>> inverted_rgb[0, 0].tolist()
-	[0, 255, 255]
-
-	>>> ## Test invalid inputs
-	>>> invert_image("not an image")
-	Traceback (most recent call last):
-		...
-	AssertionError: Image must be a numpy array
-		"""
-	# Check input data
-	check_image(image, ignore_dtype=ignore_dtype)
-
-	# Get the maximum value based on the image's data type
-	if image.dtype == np.uint8:
-		max_value = 255
-	elif image.dtype == np.uint16:
-		max_value = 65535
-	elif image.dtype == np.float32 or image.dtype == np.float64:
-		# For float images, we assume range [0, 1]
-		max_value = 1.0
-	else:
-		# Default case, assuming 8-bit
-		max_value = 255
-		image = image.astype(np.uint8)
-
-	# Invert the image
-	inverted = max_value - image
-
-	# Ensure we return the same dtype as the input
-	return inverted.astype(image.dtype)
-
+
+# Imports
+from .common import Any, NDArray, check_image, np
+
+
+# Function
+def invert_image(image: NDArray[Any], ignore_dtype: bool = False) -> NDArray[Any]:
+	""" Invert the colors of an image.
+
+	This function inverts the colors of the input image by subtracting each pixel value
+	from the maximum possible value based on the image type.
+
+	Args:
+		image         (NDArray[Any]):       Input image as a NumPy array.
+		ignore_dtype  (bool):             Ignore the dtype check.
+	Returns:
+		NDArray[Any]: Image with inverted colors.
+
+	>>> ## Basic tests
+	>>> image = np.array([[10, 20, 30], [40, 50, 60], [70, 80, 90]], dtype=np.uint8)
+	>>> inverted = invert_image(image)
+	>>> inverted.tolist()
+	[[245, 235, 225], [215, 205, 195], [185, 175, 165]]
+
+	>>> # Test with floating point image
+	>>> float_img = np.array([[0.1, 0.2], [0.3, 0.4]], dtype=np.float32)
+	>>> [round(float(x), 1) for x in invert_image(float_img).flatten()]
+	[0.9, 0.8, 0.7, 0.6]
+
+	>>> # Test with RGB image
+	>>> rgb = np.zeros((2, 2, 3), dtype=np.uint8)
+	>>> rgb[0, 0] = [255, 0, 0]  # Red pixel
+	>>> inverted_rgb = invert_image(rgb)
+	>>> inverted_rgb[0, 0].tolist()
+	[0, 255, 255]
+
+	>>> ## Test invalid inputs
+	>>> invert_image("not an image")
+	Traceback (most recent call last):
+		...
+	AssertionError: Image must be a numpy array
+		"""
+	# Check input data
+	check_image(image, ignore_dtype=ignore_dtype)
+
+	# Get the maximum value based on the image's data type
+	if image.dtype == np.uint8:
+		max_value = 255
+	elif image.dtype == np.uint16:
+		max_value = 65535
+	elif image.dtype == np.float32 or image.dtype == np.float64:
+		# For float images, we assume range [0, 1]
+		max_value = 1.0
+	else:
+		# Default case, assuming 8-bit
+		max_value = 255
+		image = image.astype(np.uint8)
+
+	# Invert the image
+	inverted = max_value - image
+
+	# Ensure we return the same dtype as the input
+	return inverted.astype(image.dtype)
+

stouputils/data_science/data_processing/image/laplacian.py

@@ -1,60 +1,60 @@
-
-# pyright: reportUnknownMemberType=false
-# pyright: reportUnknownVariableType=false
-# pyright: reportUnknownArgumentType=false
-# pyright: reportAttributeAccessIssue=false
-# pyright: reportArgumentType=false
-# pyright: reportCallIssue=false
-
-# Imports
-from .common import Any, NDArray, check_image, cv2, np
-
-
-# Functions
-def laplacian_image(image: NDArray[Any], kernel_size: int = 3, ignore_dtype: bool = False) -> NDArray[Any]:
-	""" Apply Laplacian edge detection to an image.
-
-	Args:
-		image         (NDArray[Any]):  Image to apply Laplacian edge detection
-		kernel_size   (int):         Size of the kernel (must be odd)
-		ignore_dtype  (bool):        Ignore the dtype check
-	Returns:
-		NDArray[Any]: Image with Laplacian edge detection applied
-
-	>>> ## Basic tests
-	>>> image = np.array([[100, 150, 200], [50, 125, 175], [25, 75, 225]])
-	>>> edges = laplacian_image(image.astype(np.uint8))
-	>>> edges.shape == image.shape[:2]  # Laplacian returns single channel
-	True
-
-	>>> rgb = np.random.randint(0, 256, (3,3,3), dtype=np.uint8)
-	>>> edges_rgb = laplacian_image(rgb)
-	>>> edges_rgb.shape == rgb.shape[:2]  # Laplacian returns single channel
-	True
-
-	>>> ## Test invalid inputs
-	>>> laplacian_image("not an image")
-	Traceback (most recent call last):
-		...
-	AssertionError: Image must be a numpy array
-
-	>>> laplacian_image(image.astype(np.uint8), kernel_size=2)
-	Traceback (most recent call last):
-		...
-	AssertionError: kernel_size must be odd, got 2
-	"""
-	# Check input data
-	check_image(image, ignore_dtype=ignore_dtype)
-	assert kernel_size % 2 == 1, f"kernel_size must be odd, got {kernel_size}"
-
-	# Convert to grayscale if needed
-	if len(image.shape) > 2:
-		image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-
-	# Apply Laplacian edge detection
-	laplacian: NDArray[Any] = cv2.Laplacian(image, cv2.CV_64F, ksize=kernel_size)
-
-	# Convert back to uint8 and normalize to 0-255 range
-	normalized: NDArray[Any] = cv2.normalize(laplacian, None, alpha=0, beta=255, norm_type=cv2.NORM_MINMAX)
-	return normalized.astype(np.uint8)
-
+
+# pyright: reportUnknownMemberType=false
+# pyright: reportUnknownVariableType=false
+# pyright: reportUnknownArgumentType=false
+# pyright: reportAttributeAccessIssue=false
+# pyright: reportArgumentType=false
+# pyright: reportCallIssue=false
+
+# Imports
+from .common import Any, NDArray, check_image, cv2, np
+
+
+# Functions
+def laplacian_image(image: NDArray[Any], kernel_size: int = 3, ignore_dtype: bool = False) -> NDArray[Any]:
+	""" Apply Laplacian edge detection to an image.
+
+	Args:
+		image         (NDArray[Any]):  Image to apply Laplacian edge detection
+		kernel_size   (int):         Size of the kernel (must be odd)
+		ignore_dtype  (bool):        Ignore the dtype check
+	Returns:
+		NDArray[Any]: Image with Laplacian edge detection applied
+
+	>>> ## Basic tests
+	>>> image = np.array([[100, 150, 200], [50, 125, 175], [25, 75, 225]])
+	>>> edges = laplacian_image(image.astype(np.uint8))
+	>>> edges.shape == image.shape[:2]  # Laplacian returns single channel
+	True
+
+	>>> rgb = np.random.randint(0, 256, (3,3,3), dtype=np.uint8)
+	>>> edges_rgb = laplacian_image(rgb)
+	>>> edges_rgb.shape == rgb.shape[:2]  # Laplacian returns single channel
+	True
+
+	>>> ## Test invalid inputs
+	>>> laplacian_image("not an image")
+	Traceback (most recent call last):
+		...
+	AssertionError: Image must be a numpy array
+
+	>>> laplacian_image(image.astype(np.uint8), kernel_size=2)
+	Traceback (most recent call last):
+		...
+	AssertionError: kernel_size must be odd, got 2
+	"""
+	# Check input data
+	check_image(image, ignore_dtype=ignore_dtype)
+	assert kernel_size % 2 == 1, f"kernel_size must be odd, got {kernel_size}"
+
+	# Convert to grayscale if needed
+	if len(image.shape) > 2:
+		image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+
+	# Apply Laplacian edge detection
+	laplacian: NDArray[Any] = cv2.Laplacian(image, cv2.CV_64F, ksize=kernel_size)
+
+	# Convert back to uint8 and normalize to 0-255 range
+	normalized: NDArray[Any] = cv2.normalize(laplacian, None, alpha=0, beta=255, norm_type=cv2.NORM_MINMAX)
+	return normalized.astype(np.uint8)
+

stouputils/data_science/data_processing/image/median_blur.py

@@ -1,52 +1,52 @@
-
-# pyright: reportUnusedImport=false
-# ruff: noqa: F401
-
-# Imports
-from .common import Any, NDArray, check_image, cv2, np
-
-
-# Functions
-def median_blur_image(image: NDArray[Any], kernel_size: int = 7, iterations: int = 1) -> NDArray[Any]:
-	""" Apply median blur to an image.
-
-	Args:
-		image         (NDArray[Any]):  Image to apply median blur
-		kernel_size   (int):         Kernel size for the median blur
-		iterations    (int):         Number of iterations for the median blur
-	Returns:
-		NDArray[Any]: Image with median blur applied
-
-	>>> ## Basic tests
-	>>> image = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=np.uint8)
-	>>> adjusted = median_blur_image(image, kernel_size=7, iterations=1)
-	>>> adjusted.tolist()
-	[[3, 3, 3], [4, 5, 6], [7, 7, 7]]
-	>>> adjusted.shape == image.shape
-	True
-	>>> adjusted.dtype == image.dtype
-	True
-
-	>>> median_blur_image(image, kernel_size=3, iterations=1).tolist()
-	[[2, 3, 3], [4, 5, 6], [7, 7, 8]]
-	>>> median_blur_image(image, kernel_size=3, iterations=2).tolist()
-	[[3, 3, 3], [4, 5, 6], [7, 7, 7]]
-	>>> median_blur_image(image, kernel_size=3, iterations=5).tolist()
-	[[3, 3, 3], [4, 5, 6], [7, 7, 7]]
-
-	>>> ## Test invalid inputs
-	>>> median_blur_image("not an image")
-	Traceback (most recent call last):
-		...
-	AssertionError: Image must be a numpy array
-	"""
-	# Check input data
-	check_image(image, ignore_dtype=True)
-
-	# Apply median blur
-	for _ in range(iterations):
-		image = cv2.medianBlur(image, kernel_size)
-
-	# Return the image
-	return image
-
+
+# pyright: reportUnusedImport=false
+# ruff: noqa: F401
+
+# Imports
+from .common import Any, NDArray, check_image, cv2, np
+
+
+# Functions
+def median_blur_image(image: NDArray[Any], kernel_size: int = 7, iterations: int = 1) -> NDArray[Any]:
+	""" Apply median blur to an image.
+
+	Args:
+		image         (NDArray[Any]):  Image to apply median blur
+		kernel_size   (int):         Kernel size for the median blur
+		iterations    (int):         Number of iterations for the median blur
+	Returns:
+		NDArray[Any]: Image with median blur applied
+
+	>>> ## Basic tests
+	>>> image = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=np.uint8)
+	>>> adjusted = median_blur_image(image, kernel_size=7, iterations=1)
+	>>> adjusted.tolist()
+	[[3, 3, 3], [4, 5, 6], [7, 7, 7]]
+	>>> adjusted.shape == image.shape
+	True
+	>>> adjusted.dtype == image.dtype
+	True
+
+	>>> median_blur_image(image, kernel_size=3, iterations=1).tolist()
+	[[2, 3, 3], [4, 5, 6], [7, 7, 8]]
+	>>> median_blur_image(image, kernel_size=3, iterations=2).tolist()
+	[[3, 3, 3], [4, 5, 6], [7, 7, 7]]
+	>>> median_blur_image(image, kernel_size=3, iterations=5).tolist()
+	[[3, 3, 3], [4, 5, 6], [7, 7, 7]]
+
+	>>> ## Test invalid inputs
+	>>> median_blur_image("not an image")
+	Traceback (most recent call last):
+		...
+	AssertionError: Image must be a numpy array
+	"""
+	# Check input data
+	check_image(image, ignore_dtype=True)
+
+	# Apply median blur
+	for _ in range(iterations):
+		image = cv2.medianBlur(image, kernel_size)
+
+	# Return the image
+	return image
+