Kevin Hu
commited on
Commit
·
2820402
1
Parent(s):
d256cb6
Fix param error. (#4355)
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
|
@@ -38,7 +38,7 @@ def main(args):
|
|
| 38 |
detr = TableStructureRecognizer()
|
| 39 |
ocr = OCR()
|
| 40 |
|
| 41 |
-
layouts = detr(images, 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"]
|
|
|
|
| 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"]
|