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):