discard division warning in HSV transform

color_conversions.py

@@ -1,44 +1,54 @@
 import numpy as np
+import warnings
 
 
 def rgb_to_hsv(rgb):
-    # Ensure the input is a float numpy array and in the range [0, 1]
-    rgb = np.asarray(rgb)
+    with warnings.catch_warnings():
+        # Suppress the specific RuntimeWarning
+        warnings.filterwarnings(
+            "ignore", category=RuntimeWarning, message="invalid value encountered in divide")
+
+        # Ensure the input is a float numpy array and in the range [0, 1]
+        rgb = np.asarray(rgb)
+
+        # Separate the R, G, B channels
+        r, g, b = rgb[..., 0], rgb[..., 1], rgb[..., 2]
 
-    # Separate the R, G, B channels
-    r, g, b = rgb[..., 0], rgb[..., 1], rgb[..., 2]
+        # Max and min of RGB values
+        max_rgb = np.max(rgb, axis=-1)
+        min_rgb = np.min(rgb, axis=-1)
+        delta = max_rgb - min_rgb
 
-    # Max and min of RGB values
-    max_rgb = np.max(rgb, axis=-1)
-    min_rgb = np.min(rgb, axis=-1)
-    delta = max_rgb - min_rgb
+        # Hue calculation
+        hue = np.zeros_like(max_rgb)
 
-    # Hue calculation
-    hue = np.zeros_like(max_rgb)
-    mask = delta != 0
+        # Avoid division by zero: Only calculate where delta is non-zero
+        mask = delta != 0
 
-    # Red is max
-    idx = (max_rgb == r) & mask
-    hue[idx] = (60 * ((g - b) / delta % 6))[idx]
+        # Red is max
+        idx = (max_rgb == r) & mask
+        hue[idx] = (60 * ((g - b) / delta % 6))[idx]
 
-    # Green is max
-    idx = (max_rgb == g) & mask
-    hue[idx] = (60 * ((b - r) / delta + 2))[idx]
+        # Green is max
+        idx = (max_rgb == g) & mask
+        hue[idx] = (60 * ((b - r) / delta + 2))[idx]
 
-    # Blue is max
-    idx = (max_rgb == b) & mask
-    hue[idx] = (60 * ((r - g) / delta + 4))[idx]
+        # Blue is max
+        idx = (max_rgb == b) & mask
+        hue[idx] = (60 * ((r - g) / delta + 4))[idx]
 
-    # Saturation calculation
-    saturation = np.zeros_like(max_rgb)
-    saturation[mask] = delta[mask] / max_rgb[mask]
+        # Saturation calculation
+        saturation = np.zeros_like(max_rgb)
+        saturation[mask] = delta[mask] / max_rgb[mask]
 
-    # Value calculation
-    value = max_rgb
+        # Handle the edge case where max_rgb is 0 (black pixels)
+        saturation[max_rgb == 0] = 0
 
-    # Stack the HSV components together
-    hsv = np.stack((hue, saturation, value), axis=-1)
+        # Value calculation
+        value = max_rgb
 
+        # Stack the HSV components together
+        hsv = np.stack((hue, saturation, value), axis=-1)
     return hsv