From fa9c69fa577eb3f31c447e2c66349469804ebe70 Mon Sep 17 00:00:00 2001 From: Billy Cao Date: Tue, 29 Oct 2024 10:58:55 +0800 Subject: [PATCH] Default use paddle ocr to false to not break jupyter notebook example Add paddleocr to requirements --- requirements.txt | 2 ++ utils.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index cddf7fd..08bfc38 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,3 +14,5 @@ dill accelerate timm einops==0.8.0 +paddlepaddle +paddleocr \ No newline at end of file diff --git a/utils.py b/utils.py index 3af9c55..13b3c9e 100755 --- a/utils.py +++ b/utils.py @@ -380,7 +380,7 @@ def get_xywh_yolo(input): -def check_ocr_box(image_path, display_img = True, output_bb_format='xywh', goal_filtering=None, easyocr_args=None, use_paddleocr=True): +def check_ocr_box(image_path, display_img = True, output_bb_format='xywh', goal_filtering=None, easyocr_args=None, use_paddleocr=False): if use_paddleocr: result = paddle_ocr.ocr(image_path, cls=False)[0] coord = [item[0] for item in result]