From 2e40c2a6f6cd16effc662ff36364c86ec4335138 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Tue, 7 Jan 2025 13:17:46 +0800 Subject: [PATCH] Fix t_recognizer issue. (#4387) ### What problem does this PR solve? #4230 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- deepdoc/vision/t_recognizer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepdoc/vision/t_recognizer.py b/deepdoc/vision/t_recognizer.py index 024428f70..55a86b988 100644 --- a/deepdoc/vision/t_recognizer.py +++ b/deepdoc/vision/t_recognizer.py @@ -23,7 +23,7 @@ sys.path.insert( '../../'))) from deepdoc.vision.seeit import draw_box -from deepdoc.vision import LayoutRecognizer, TableStructureRecognizer, OCR, init_in_out +from deepdoc.vision import LayoutRecognizer, TableStructureRecognizer, OCR, init_in_out, Recognizer import argparse import re import numpy as np @@ -33,12 +33,12 @@ def main(args): images, outputs = init_in_out(args) if args.mode.lower() == "layout": detr = LayoutRecognizer("layout") + layouts = super(Recognizer, detr)(images, thr=float(args.threshold)) if args.mode.lower() == "tsr": labels = TableStructureRecognizer.labels detr = TableStructureRecognizer() ocr = OCR() - - layouts = detr(images, thr=float(args.threshold)) + layouts = detr(images, thr=float(args.threshold)) for i, lyt in enumerate(layouts): if args.mode.lower() == "tsr": #lyt = [t for t in lyt if t["type"] == "table column"]