def pil_to_numpy()

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

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

    def pil_to_numpy(images: Union[List[PIL.Image.Image], PIL.Image.Image], normalize: bool = True) -> np.ndarray:
        """
        Convert a PIL image or a list of PIL images to NumPy arrays.

        Args:
            images (Union[List[PIL.Image.Image], PIL.Image.Image]): The input image(s) as PIL image(s).

        Returns:
            np.ndarray: The converted image(s) as a NumPy array.
        """
        if not isinstance(images, list):
            images = [images]

        if normalize:
            images = [np.array(image).astype(np.float32) / 255.0 for image in images]
        else:
            images = [np.array(image).astype(np.float32) for image in images]

        images = np.stack(images, axis=0)

        return images