Function main

examples/basic_ocr.py:50–78

()

def main() -> int:
    parser = build_parser()
    args = parser.parse_args()
    if not args.image.exists():
        parser.error(f"文件 {args.image} 不存在")

    ocr = DdddOcr(ocr=True, det=False, beta=args.beta, old=args.old, show_ad=False)

    try:
        data = args.image.read_bytes()
        if args.probability:
            result = ocr.classification(
                data,
                probability=True,
                colors=list(args.colors),
            )
        else:
            result = ocr.classification(
                data,
                probability=False,
                colors=list(args.colors),
            )
    except (DdddOcrInputError, InvalidImageError) as exc:
        parser.error(str(exc))
        return 1

    print("识别结果:")
    print(result)
    return 0


if __name__ == "__main__":