centersl/MinerU
1
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-03-27 11:08:32 +07:00
Code Issues Packages Projects Releases Wiki Activity

feat: sort blocks by index in rebuilt page blocks for consistent ordering

Browse Source
This commit is contained in:
myhloli
2026-03-21 04:18:21 +08:00
parent 28eeebb95d
commit 6562d440db
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
mineru/backend/pipeline/pipeline_magic_model.py
View File

@@ -426,15 +426,16 @@ class MagicModel:
else:
self.chart_groups.append(group_info)
rebuilt_page_blocks.append(
{
"type": original_block_type,
"bbox": block["bbox"],
"blocks": [body_block, *captions, *footnotes],
"index": block["index"],
"score": block.get("score"),
}
)
two_layer_block = {
"type": original_block_type,
"bbox": block["bbox"],
"blocks": [body_block, *captions, *footnotes],
"index": block["index"],
"score": block.get("score"),
}
# 对blocks按index排序
two_layer_block["blocks"].sort(key=lambda x: x["index"])
rebuilt_page_blocks.append(two_layer_block)
self.page_blocks = rebuilt_page_blocks