Skip to content

GitLab

Commit efed5faa authored by 许瑞's avatar 许瑞
Browse files

feat: modify foot note bbox tmp

parent 05161c6e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
magic_pdf/train_utils/convert_to_train_format.py
View file @ efed5faa
...@@ -54,8 +54,8 @@ def convert_to_train_format(jso: dict) -> []: ...@@ -54,8 +54,8 @@ def convert_to_train_format(jso: dict) -> []:
n_bbox = {"category_id": 10, "bbox": inter_equation["bbox"]} n_bbox = {"category_id": 10, "bbox": inter_equation["bbox"]}
bboxes.append(n_bbox) bboxes.append(n_bbox)
for footnote in v['bak_footer_note_bboxes']: for footnote_bbox in v["bak_footer_note_bboxes"]:
n_bbox = {"category_id": 5, "bbox": footnote["bbox"]} n_bbox = {"category_id": 5, "bbox": list(footnote_bbox)}
bboxes.append(n_bbox) bboxes.append(n_bbox)
info["bboxes"] = bboxes info["bboxes"] = bboxes
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment