From eca75bc6af0f03ddfc8bc10d1fb89a47ffbf6299 Mon Sep 17 00:00:00 2001
From: "yuanzhi.zyz" <yuanzhi.zyz@alibaba-inc.com>
Date: Fri, 24 Feb 2023 17:02:24 +0800
Subject: [PATCH] add ocr_recognition bugfix

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11785977
---
 modelscope/models/cv/ocr_recognition/preprocessor.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modelscope/models/cv/ocr_recognition/preprocessor.py b/modelscope/models/cv/ocr_recognition/preprocessor.py
index f47dcaef..6405e3e4 100644
--- a/modelscope/models/cv/ocr_recognition/preprocessor.py
+++ b/modelscope/models/cv/ocr_recognition/preprocessor.py
@@ -1,5 +1,6 @@
 # Copyright (c) Alibaba, Inc. and its affiliates.
 
+import math
 import os
 
 import cv2
@@ -71,6 +72,8 @@ class OCRRecognitionPreprocessor(Preprocessor):
             )
 
         if self.do_chunking:
+            PRED_LENTH = 75
+            PRED_PAD = 6
             data = []
             img_h, img_w = img.shape
             wh_ratio = img_w / img_h