Function validate_image_input

ddddocr/utils/validators.py:15–31

验证图片输入是否为支持的类型 Args: img_input: 图片输入 Returns: bool: 是否为有效的图片输入 Raises: DDDDOCRError: 当输入类型不支持时

(img_input: Any)

def validate_image_input(img_input: Any) -> bool:
    """
    验证图片输入是否为支持的类型

    Args:
        img_input: 图片输入

    Returns:
        bool: 是否为有效的图片输入

    Raises:
        DDDDOCRError: 当输入类型不支持时
    """
    valid_types = (bytes, str, pathlib.PurePath, Image.Image, np.ndarray)
    if not isinstance(img_input, valid_types):
        raise DDDDOCRError(f"不支持的图片输入类型: {type(img_input)}。支持的类型: {valid_types}")
    return True


def validate_model_config(ocr: bool = True, det: bool = False, old: bool = False,