From 0bf30fea675388add8a8dc0ddc5c3e1bf713a090 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Thu, 3 Jun 2021 16:47:38 +0800 Subject: [PATCH] fix eval parse_args import error --- ppstructure/table/eval_table.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/ppstructure/table/eval_table.py b/ppstructure/table/eval_table.py index baa70177..46df68df 100755 --- a/ppstructure/table/eval_table.py +++ b/ppstructure/table/eval_table.py @@ -21,7 +21,8 @@ import cv2 import json from tqdm import tqdm from ppstructure.table.table_metric import TEDS -from ppstructure.table.predict_table import TableSystem, utility +from ppstructure.table.predict_table import TableSystem +from ppstructure.predict_system import parse_args def main(gt_path, img_root, args): @@ -32,8 +33,6 @@ def main(gt_path, img_root, args): pred_htmls = [] gt_htmls = [] for img_name in tqdm(jsons_gt): - if img_name != 'PMC1064865_002_00.png': - continue # 读取信息 img = cv2.imread(os.path.join(img_root,img_name)) pred_html = text_sys(img) @@ -61,7 +60,7 @@ def get_gt_html(gt_structures, contents_with_block): if __name__ == '__main__': - args = utility.parse_args() + args = parse_args() gt_path = 'table/match_code/f_gt_bbox.json' img_root = 'table/imgs' main(gt_path,img_root, args) -- GitLab