Kevin Hu
commited on
Commit
·
957cf28
1
Parent(s):
551c783
Fix t_recognizer issue. (#4387)
Browse files### 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
CHANGED
@@ -23,7 +23,7 @@ sys.path.insert(
|
|
23 |
'../../')))
|
24 |
|
25 |
from deepdoc.vision.seeit import draw_box
|
26 |
-
from deepdoc.vision import LayoutRecognizer, TableStructureRecognizer, OCR, init_in_out
|
27 |
import argparse
|
28 |
import re
|
29 |
import numpy as np
|
@@ -33,12 +33,12 @@ def main(args):
|
|
33 |
images, outputs = init_in_out(args)
|
34 |
if args.mode.lower() == "layout":
|
35 |
detr = LayoutRecognizer("layout")
|
|
|
36 |
if args.mode.lower() == "tsr":
|
37 |
labels = TableStructureRecognizer.labels
|
38 |
detr = TableStructureRecognizer()
|
39 |
ocr = OCR()
|
40 |
-
|
41 |
-
layouts = detr(images, thr=float(args.threshold))
|
42 |
for i, lyt in enumerate(layouts):
|
43 |
if args.mode.lower() == "tsr":
|
44 |
#lyt = [t for t in lyt if t["type"] == "table column"]
|
|
|
23 |
'../../')))
|
24 |
|
25 |
from deepdoc.vision.seeit import draw_box
|
26 |
+
from deepdoc.vision import LayoutRecognizer, TableStructureRecognizer, OCR, init_in_out, Recognizer
|
27 |
import argparse
|
28 |
import re
|
29 |
import numpy as np
|
|
|
33 |
images, outputs = init_in_out(args)
|
34 |
if args.mode.lower() == "layout":
|
35 |
detr = LayoutRecognizer("layout")
|
36 |
+
layouts = super(Recognizer, detr)(images, thr=float(args.threshold))
|
37 |
if args.mode.lower() == "tsr":
|
38 |
labels = TableStructureRecognizer.labels
|
39 |
detr = TableStructureRecognizer()
|
40 |
ocr = OCR()
|
41 |
+
layouts = detr(images, thr=float(args.threshold))
|
|
|
42 |
for i, lyt in enumerate(layouts):
|
43 |
if args.mode.lower() == "tsr":
|
44 |
#lyt = [t for t in lyt if t["type"] == "table column"]
|