"sgl-kernel/python/vscode:/vscode.git/clone" did not exist on "8db3ac55a975ecd849dc8a0e8a306eddd4ae0be5"
Commit 12caa784 authored by myhloli's avatar myhloli
Browse files

fix(table): handle empty OCR result in rapidtable

- Add check for empty OCR result when using PaddleOCR model
- Assign None to ocr_result if no text is detected, preventing further errors
parent 4d110d31
...@@ -27,8 +27,11 @@ class RapidTableModel(object): ...@@ -27,8 +27,11 @@ class RapidTableModel(object):
elif self.ocr_model_name == "PaddleOCR": elif self.ocr_model_name == "PaddleOCR":
bgr_image = cv2.cvtColor(np.asarray(image), cv2.COLOR_RGB2BGR) bgr_image = cv2.cvtColor(np.asarray(image), cv2.COLOR_RGB2BGR)
ocr_result = self.ocr_engine.ocr(bgr_image)[0] ocr_result = self.ocr_engine.ocr(bgr_image)[0]
ocr_result = [[item[0], item[1][0], item[1][1]] for item in ocr_result if if ocr_result:
ocr_result = [[item[0], item[1][0], item[1][1]] for item in ocr_result if
len(item) == 2 and isinstance(item[1], tuple)] len(item) == 2 and isinstance(item[1], tuple)]
else:
ocr_result = None
else: else:
logger.error("OCR model not supported") logger.error("OCR model not supported")
ocr_result = None ocr_result = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment