From 9e4ebea939084e0e1f4af2439b117be76e41a4e1 Mon Sep 17 00:00:00 2001 From: myhloli Date: Mon, 16 Dec 2024 17:27:19 +0800 Subject: [PATCH] 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 --- magic_pdf/model/doc_analyze_by_custom_model.py | 1 - magic_pdf/model/pdf_extract_kit.py | 1 - magic_pdf/model/sub_modules/mfd/yolov8/YOLOv8.py | 4 ++-- magic_pdf/pdf_parse_union_core_v2.py | 1 - 4 files changed, 2 insertions(+), 5 deletions(-) diff --git a/magic_pdf/model/doc_analyze_by_custom_model.py b/magic_pdf/model/doc_analyze_by_custom_model.py index ec46d867..f3e69843 100644 --- a/magic_pdf/model/doc_analyze_by_custom_model.py +++ b/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 diff --git a/magic_pdf/model/pdf_extract_kit.py b/magic_pdf/model/pdf_extract_kit.py index 0ff01670..ff3faead 100644 --- a/magic_pdf/model/pdf_extract_kit.py +++ b/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 diff --git a/magic_pdf/model/sub_modules/mfd/yolov8/YOLOv8.py b/magic_pdf/model/sub_modules/mfd/yolov8/YOLOv8.py index 6e4505ed..78ffc725 100644 --- a/magic_pdf/model/sub_modules/mfd/yolov8/YOLOv8.py +++ b/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, ) ] diff --git a/magic_pdf/pdf_parse_union_core_v2.py b/magic_pdf/pdf_parse_union_core_v2.py index aa6db3c3..29ca1351 100644 --- a/magic_pdf/pdf_parse_union_core_v2.py +++ b/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):