Update visuals.py

src/koyo/visuals.py

@@ -10,6 +10,7 @@
 import matplotlib.pyplot as plt
 import numba as nb
 import numpy as np
+from koyo.typing import PathLike
 
 from koyo.image import clip_hotspots
 from koyo.utilities import is_above_version
@@ -44,6 +45,20 @@ def close_pil_image(fig: Image) -> None:
         del fig
 
 
+def save_rgb(path: PathLike, image: np.ndarray) -> None:
+    """Save image a RGB using PIL."""
+    from PIL import Image
+
+    Image.fromarray(image).save(path)
+
+
+def save_gray(path: PathLike, image: np.ndarray, multiplier: int = 255) -> None:
+    """Save image as grayscale using PIL."""
+    from PIL import Image
+
+    Image.fromarray(image * multiplier).convert("L").save(path)
+
+
 def set_tick_fmt(ax: plt.Axes, use_offset: bool = False, axis: str = "both") -> plt.Axes:
     """Set tick format to control whether scientific notation is shown."""
     ax.ticklabel_format(axis=axis, style="scientific" if use_offset else "plain", useOffset=use_offset)