|
@@ -40,7 +40,7 @@ class DotsOCRProcessor:
|
|
|
min_pixels: int = MIN_PIXELS,
|
|
min_pixels: int = MIN_PIXELS,
|
|
|
max_pixels: int = MAX_PIXELS,
|
|
max_pixels: int = MAX_PIXELS,
|
|
|
normalize_numbers: bool = False,
|
|
normalize_numbers: bool = False,
|
|
|
- debug: bool = False):
|
|
|
|
|
|
|
+ log_level: str = "INFO"):
|
|
|
"""
|
|
"""
|
|
|
初始化处理器
|
|
初始化处理器
|
|
|
|
|
|
|
@@ -53,7 +53,7 @@ class DotsOCRProcessor:
|
|
|
min_pixels: 最小像素数
|
|
min_pixels: 最小像素数
|
|
|
max_pixels: 最大像素数
|
|
max_pixels: 最大像素数
|
|
|
normalize_numbers: 是否标准化数字
|
|
normalize_numbers: 是否标准化数字
|
|
|
- debug: 是否启用调试模式
|
|
|
|
|
|
|
+ log_level: 日志级别(DEBUG, INFO, WARNING, ERROR),当为 DEBUG 时会打印详细错误信息
|
|
|
"""
|
|
"""
|
|
|
self.ip = ip
|
|
self.ip = ip
|
|
|
self.port = port
|
|
self.port = port
|
|
@@ -63,7 +63,7 @@ class DotsOCRProcessor:
|
|
|
self.min_pixels = min_pixels
|
|
self.min_pixels = min_pixels
|
|
|
self.max_pixels = max_pixels
|
|
self.max_pixels = max_pixels
|
|
|
self.normalize_numbers = normalize_numbers
|
|
self.normalize_numbers = normalize_numbers
|
|
|
- self.debug = debug
|
|
|
|
|
|
|
+ self.log_level = log_level
|
|
|
|
|
|
|
|
# 初始化解析器
|
|
# 初始化解析器
|
|
|
self.parser = DotsOCRParser(
|
|
self.parser = DotsOCRParser(
|
|
@@ -81,7 +81,7 @@ class DotsOCRProcessor:
|
|
|
logger.info(f" - 提示模式: {prompt_mode}")
|
|
logger.info(f" - 提示模式: {prompt_mode}")
|
|
|
logger.info(f" - 像素范围: {min_pixels} - {max_pixels}")
|
|
logger.info(f" - 像素范围: {min_pixels} - {max_pixels}")
|
|
|
logger.info(f" - 数字标准化: {normalize_numbers}")
|
|
logger.info(f" - 数字标准化: {normalize_numbers}")
|
|
|
- logger.info(f" - 调试模式: {debug}")
|
|
|
|
|
|
|
+ logger.info(f" - 日志级别: {log_level}")
|
|
|
|
|
|
|
|
def create_temp_session_dir(self) -> tuple:
|
|
def create_temp_session_dir(self) -> tuple:
|
|
|
"""创建临时会话目录"""
|
|
"""创建临时会话目录"""
|
|
@@ -191,7 +191,7 @@ class DotsOCRProcessor:
|
|
|
|
|
|
|
|
except Exception as e:
|
|
except Exception as e:
|
|
|
logger.error(f"Error saving results for {image_name}: {e}")
|
|
logger.error(f"Error saving results for {image_name}: {e}")
|
|
|
- if self.debug:
|
|
|
|
|
|
|
+ if self.log_level == "DEBUG":
|
|
|
traceback.print_exc()
|
|
traceback.print_exc()
|
|
|
|
|
|
|
|
return saved_files
|
|
return saved_files
|
|
@@ -296,7 +296,7 @@ class DotsOCRProcessor:
|
|
|
result_info["error"] = str(e)
|
|
result_info["error"] = str(e)
|
|
|
result_info["success"] = False
|
|
result_info["success"] = False
|
|
|
logger.error(f"Error processing {image_name}: {e}")
|
|
logger.error(f"Error processing {image_name}: {e}")
|
|
|
- if self.debug:
|
|
|
|
|
|
|
+ if self.log_level == "DEBUG":
|
|
|
traceback.print_exc()
|
|
traceback.print_exc()
|
|
|
|
|
|
|
|
finally:
|
|
finally:
|