opensource/MinerU
3
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-10-30 19:33:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: update return result

Browse Source
This commit is contained in:
icecraft
2024-10-25 15:58:09 +08:00
parent 92579040c4
commit 2c60172b83
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
magic_pdf/model/magic_model.py
+1 -4
View File
@@ -782,8 +782,7 @@ class MagicModel:
for i in sub_obj_map_h.keys():
ret.append(
{
'sub_bbox': subjects[i]['bbox'],
'score': subjects[i]['score'],
'sub_bbox': {'bbox': subjects[i]['bbox'], 'score': subjects[i]['score']},
'obj_bboxes': [
{'score': objects[j]['score'], 'bbox': objects[j]['bbox']}
for j in sub_obj_map_h[i]
@@ -803,7 +802,6 @@ class MagicModel:
record = {
'image_body': v['sub_bbox'],
'image_caption_list': v['obj_bboxes'],
'score': v['score'],
}
filter_idx = v['sub_idx']
d = next(filter(lambda x: x['sub_idx'] == filter_idx, with_footnotes))
@@ -819,7 +817,6 @@ class MagicModel:
record = {
'table_body': v['sub_bbox'],
'table_caption_list': v['obj_bboxes'],
'score': v['score']
}
filter_idx = v['sub_idx']
d = next(filter(lambda x: x['sub_idx'] == filter_idx, with_footnotes))