def numpy_to_pil()

in src/image_gen_aux/image_processor.py [0:0]

• 9 lines of code
• 6 McCabe index (conditional complexity)

    def numpy_to_pil(images: np.ndarray) -> List[PIL.Image.Image]:
        """
        Convert a NumPy image or a batch of images to a PIL image.

        Args:
            images (np.ndarray): The input image(s) as a NumPy array.

        Returns:
            List[PIL.Image.Image]: The converted image(s) as PIL images.
        """
        if images.ndim == 3:
            images = images[None, ...]
        images = (images * 255).round().astype("uint8")
        if images.shape[-1] == 1:
            # special case for grayscale (single channel) images
            pil_images = [Image.fromarray(image.squeeze(), mode="L") for image in images]
        else:
            pil_images = [Image.fromarray(image) for image in images]

        return pil_images