소스 검색

refactor: optimize OCR batch processing and enhance image cropping logic

myhloli 5 달 전
부모
커밋
6b1df41947
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      mineru/backend/pipeline/batch_analyze.py

+ 2 - 2
mineru/backend/pipeline/batch_analyze.py

@@ -127,7 +127,7 @@ class BatchAnalyze:
                 if not lang_crop_list:
                     continue
 
-                logger.info(f"Processing OCR detection for language {lang} with {len(lang_crop_list)} images")
+                # logger.info(f"Processing OCR detection for language {lang} with {len(lang_crop_list)} images")
 
                 # 获取OCR模型
                 ocr_model = atom_model_manager.get_atom_model(
@@ -171,7 +171,7 @@ class BatchAnalyze:
 
                     # 批处理检测
                     batch_size = min(len(batch_images), self.batch_ratio * 16)  # 增加批处理大小
-                    logger.debug(f"OCR-det batch: {batch_size} images, target size: {target_h}x{target_w}")
+                    # logger.debug(f"OCR-det batch: {batch_size} images, target size: {target_h}x{target_w}")
                     batch_results = ocr_model.text_detector.batch_predict(batch_images, batch_size)
 
                     # 处理批处理结果