Commit 9e4ebea9 authored by myhloli's avatar myhloli
Browse files

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
parent 379cf150
...@@ -10,7 +10,6 @@ import paddle ...@@ -10,7 +10,6 @@ import paddle
paddle.disable_signal_handler() paddle.disable_signal_handler()
os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1' # 禁止albumentations检查更新 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1' # 禁止albumentations检查更新
os.environ['YOLO_VERBOSE'] = 'False' # disable yolo logger
try: try:
import torchtext import torchtext
......
...@@ -10,7 +10,6 @@ from loguru import logger ...@@ -10,7 +10,6 @@ from loguru import logger
from PIL import Image from PIL import Image
os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1' # 禁止albumentations检查更新 os.environ['NO_ALBUMENTATIONS_UPDATE'] = '1' # 禁止albumentations检查更新
os.environ['YOLO_VERBOSE'] = 'False' # disable yolo logger
try: try:
import torchtext import torchtext
......
...@@ -8,7 +8,7 @@ class YOLOv8MFDModel(object): ...@@ -8,7 +8,7 @@ class YOLOv8MFDModel(object):
def predict(self, image): def predict(self, image):
mfd_res = self.mfd_model.predict( 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] )[0]
return mfd_res return mfd_res
...@@ -22,7 +22,7 @@ class YOLOv8MFDModel(object): ...@@ -22,7 +22,7 @@ class YOLOv8MFDModel(object):
imgsz=1888, imgsz=1888,
conf=0.25, conf=0.25,
iou=0.45, iou=0.45,
verbose=True, verbose=False,
device=self.device, device=self.device,
) )
] ]
......
...@@ -37,7 +37,6 @@ from magic_pdf.pre_proc.ocr_dict_merge import fill_spans_in_blocks, fix_block_sp ...@@ -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 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['NO_ALBUMENTATIONS_UPDATE'] = '1' # 禁止albumentations检查更新
os.environ['YOLO_VERBOSE'] = 'False' # disable yolo logger
def __replace_STX_ETX(text_str: str): def __replace_STX_ETX(text_str: str):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment