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

refactor(draw_bbox): clear cuda cache and update bbox sorting

Browse Source 
- Added CUDA cache clearing after layoutreader prediction to free up GPU memory.
- Modified the bbox sorting logic to sort text and title blocks separately.
- Adjusted drawing colors for better distinction in debug visualizations.
This commit is contained in:
myhloli
2024-09-27 16:06:15 +08:00
parent 00cda7a6f4
commit 36220d69d4
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
magic_pdf/libs/draw_bbox.py
+4 -1
View File
@@ -341,10 +341,13 @@ def draw_layout_sort_bbox(pdf_info, pdf_bytes, out_path, filename):
for page in pdf_info:
page_line_list = []
for block in page['preproc_blocks']:
if block['type'] == 'text' or block['type'] == 'title':
if block['type'] == 'text' or block['type'] == 'title' or block['type'] == 'interline_equation':
for line in block['lines']:
bbox = line['bbox']
page_line_list.append(bbox)
if block['type'] == 'table' or block['type'] == 'image':
bbox = block['bbox']
page_line_list.append(bbox)
# 使用layoutreader排序
page_size = page['page_size']