feat: modify foot note bbox tmp

efed5faa · 许瑞 · 05161c6e · efed5faa
Commit efed5faa authored Mar 23, 2024 by 许瑞
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

convert_to_train_format.py magic_pdf/train_utils/convert_to_train_format.py +2 -2

No files found.
--- 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