Function load_image_from_input

ddddocr/utils/image_io.py:82–119

从多种输入格式加载图片 Args: img_input: 图片输入，支持bytes、base64字符串、文件路径、PIL Image对象或numpy数组 Returns: PIL Image对象 Raises: ImageProcessError: 当图片加载失败时

(img_input: Union[bytes, str, pathlib.PurePath, Image.Image, np.ndarray])

def load_image_from_input(img_input: Union[bytes, str, pathlib.PurePath, Image.Image, np.ndarray]) -> Image.Image:
    """
    从多种输入格式加载图片

    Args:
        img_input: 图片输入，支持bytes、base64字符串、文件路径、PIL Image对象或numpy数组

    Returns:
        PIL Image对象

    Raises:
        ImageProcessError: 当图片加载失败时
    """
    try:
        if isinstance(img_input, bytes):
            return Image.open(io.BytesIO(img_input))
        elif isinstance(img_input, Image.Image):
            return img_input.copy()
        elif isinstance(img_input, np.ndarray):
            return _numpy_to_pil_image(img_input)
        elif isinstance(img_input, str):
            # 先尝试作为文件路径，如果失败则作为base64
            if os.path.exists(img_input):
                return Image.open(img_input)
            else:
                return base64_to_image(img_input)
        elif isinstance(img_input, pathlib.PurePath):
            return Image.open(img_input)
        else:
            supported_types = (bytes, str, pathlib.PurePath, Image.Image, np.ndarray)
            raise ImageProcessError(
                f"不支持的图片输入类型: {type(img_input)}。"
                f"支持的类型: {supported_types}"
            )
    except ImageProcessError:
        raise
    except Exception as e:
        raise ImageProcessError(f"图片加载失败: {str(e)}") from e


def _numpy_to_pil_image(array: np.ndarray) -> Image.Image: