From efed5faa53d9d80140a2add60261eda06e7cae5d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=AE=B8=E7=91=9E?= Date: Sat, 23 Mar 2024 14:34:25 +0800 Subject: [PATCH] feat: modify foot note bbox tmp --- magic_pdf/train_utils/convert_to_train_format.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/magic_pdf/train_utils/convert_to_train_format.py b/magic_pdf/train_utils/convert_to_train_format.py index be359c79..c50e6b3b 100644 --- a/magic_pdf/train_utils/convert_to_train_format.py +++ b/magic_pdf/train_utils/convert_to_train_format.py @@ -54,8 +54,8 @@ def convert_to_train_format(jso: dict) -> []: n_bbox = {"category_id": 10, "bbox": inter_equation["bbox"]} bboxes.append(n_bbox) - for footnote in v['bak_footer_note_bboxes']: - n_bbox = {"category_id": 5, "bbox": footnote["bbox"]} + for footnote_bbox in v["bak_footer_note_bboxes"]: + n_bbox = {"category_id": 5, "bbox": list(footnote_bbox)} bboxes.append(n_bbox) info["bboxes"] = bboxes