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]