opensource/MinerU
3
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-10-24 01:23:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: modify foot note bbox tmp

Browse Source
This commit is contained in:
许瑞
2024-03-23 14:34:25 +08:00
parent 05161c6e62
commit efed5faa53
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
magic_pdf/train_utils/convert_to_train_format.py
+2 -2
View File
@@ -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