Browse Source

refactor(magic_pdf): remove YOLO_VERBOSE setting and update YOLOv8 prediction verbosity

- Remove YOLO_VERBOSE environment variable from multiple files
- Set verbose=False in YOLOv8 prediction method to suppress logger output
myhloli 11 tháng trước cách đây
mục cha
commit
9e4ebea939

+ 0 - 1
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -10,7 +10,6 @@ import paddle
 paddle.disable_signal_handler()
 
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
-os.environ['YOLO_VERBOSE'] = 'False'  # disable yolo logger
 
 try:
     import torchtext

+ 0 - 1
magic_pdf/model/pdf_extract_kit.py

@@ -10,7 +10,6 @@ from loguru import logger
 from PIL import Image
 
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
-os.environ['YOLO_VERBOSE'] = 'False'  # disable yolo logger
 
 try:
     import torchtext

+ 2 - 2
magic_pdf/model/sub_modules/mfd/yolov8/YOLOv8.py

@@ -8,7 +8,7 @@ class YOLOv8MFDModel(object):
 
     def predict(self, image):
         mfd_res = self.mfd_model.predict(
-            image, imgsz=1888, conf=0.25, iou=0.45, verbose=True, device=self.device
+            image, imgsz=1888, conf=0.25, iou=0.45, verbose=False, device=self.device
         )[0]
         return mfd_res
 
@@ -22,7 +22,7 @@ class YOLOv8MFDModel(object):
                     imgsz=1888,
                     conf=0.25,
                     iou=0.45,
-                    verbose=True,
+                    verbose=False,
                     device=self.device,
                 )
             ]

+ 0 - 1
magic_pdf/pdf_parse_union_core_v2.py

@@ -37,7 +37,6 @@ from magic_pdf.pre_proc.ocr_dict_merge import fill_spans_in_blocks, fix_block_sp
 from magic_pdf.pre_proc.ocr_span_list_modify import get_qa_need_list_v2, remove_overlaps_low_confidence_spans, remove_overlaps_min_spans
 
 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1'  # 禁止albumentations检查更新
-os.environ['YOLO_VERBOSE'] = 'False'  # disable yolo logger
 
 
 def __replace_STX_ETX(text_str: str):