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

Merge pull request #3557 from opendatalab/dev

Browse Source 
Dev
This commit is contained in:
Xiaomeng Zhao
2025-09-20 15:30:40 +08:00
committed by GitHub
parent 4f8c00cc34 c7bde0ab39
commit e96e4a0ce4
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mineru/backend/pipeline/batch_analyze.py
View File

@@ -116,16 +116,14 @@ class BatchAnalyze:
atom_model_name=AtomicModel.ImgOrientationCls,
)
try:
import torch
from packaging import version
if version.parse(torch.__version__) >= version.parse("2.8.0"):
for table_res in table_res_list_all_page:
rotate_label = img_orientation_cls_model.predict(table_res['table_img'])
img_orientation_cls_model.img_rotate(table_res, rotate_label)
else:
if self.enable_ocr_det_batch:
img_orientation_cls_model.batch_predict(table_res_list_all_page,
det_batch_size=self.batch_ratio * OCR_DET_BASE_BATCH_SIZE,
batch_size=TABLE_ORI_CLS_BATCH_SIZE)
else:
for table_res in table_res_list_all_page:
rotate_label = img_orientation_cls_model.predict(table_res['table_img'])
img_orientation_cls_model.img_rotate(table_res, rotate_label)
except Exception as e:
logger.warning(
f"Image orientation classification failed: {e}, using original image"