From 3d3d0d31700b6c580ee1194dc6ba6feb245f1684 Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Thu, 18 Aug 2022 03:07:46 +0000
Subject: [PATCH] fix vis error

---
 ppstructure/table/predict_structure.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppstructure/table/predict_structure.py b/ppstructure/table/predict_structure.py
index 7198fb2b..a580947a 100755
--- a/ppstructure/table/predict_structure.py
+++ b/ppstructure/table/predict_structure.py
@@ -148,7 +148,7 @@ def main(args):
                                                 bbox_list_str))
 
             if len(bbox_list) > 0 and len(bbox_list[0]) == 4:
-                img = draw_rectangle(image_file, pred_res['cell_bbox'])
+                img = draw_rectangle(image_file, bbox_list)
             else:
                 img = utility.draw_boxes(img, bbox_list)
             img_save_path = os.path.join(args.output,
-- 
GitLab