Function image_to_numpy

ddddocr/utils/image_io.py:170–189

将PIL Image转换为numpy数组 Args: image: PIL Image对象 target_mode: 目标颜色模式，默认为'RGB' Returns: numpy数组 Raises: ImageProcessError: 当转换失败时

(image: Image.Image, target_mode: str = 'RGB')

def image_to_numpy(image: Image.Image, target_mode: str = 'RGB') -> np.ndarray:
    """
    将PIL Image转换为numpy数组
    
    Args:
        image: PIL Image对象
        target_mode: 目标颜色模式，默认为'RGB'
        
    Returns:
        numpy数组
        
    Raises:
        ImageProcessError: 当转换失败时
    """
    try:
        if image.mode != target_mode:
            image = image.convert(target_mode)
        return np.array(image)
    except Exception as e:
        raise ImageProcessError(f"图片转numpy数组失败: {str(e)}") from e


def numpy_to_image(array: np.ndarray, mode: str = 'RGB') -> Image.Image: