ocr1.0
This commit is contained in:
@ -7,6 +7,11 @@ def setup_logger():
|
||||
配置一个全局日志记录器,支持输出到控制台和文件。
|
||||
"""
|
||||
# 创建一个日志记录器
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
logger = logging.getLogger("ViolationDetectorLogger")
|
||||
logger.setLevel(logging.DEBUG) # 设置最低级别为DEBUG
|
||||
|
||||
|
@ -99,7 +99,7 @@ class OCRViolationDetector:
|
||||
"""初始化RapidOCR引擎"""
|
||||
self.logger.info("正在初始化RapidOCR引擎...")
|
||||
|
||||
config_path = r"../ocr/config/1.yaml"
|
||||
config_path = r"D:\Git\bin\video\ocr\config\1.yaml"
|
||||
try:
|
||||
# 检查配置文件是否存在
|
||||
if not os.path.exists(config_path):
|
||||
@ -157,4 +157,3 @@ class OCRViolationDetector:
|
||||
|
||||
# 返回检测结果(是否有违禁词、所有违禁词列表、对应置信度列表)
|
||||
return len(all_prohibited) > 0, all_prohibited, all_confidences
|
||||
|
||||
|
Reference in New Issue
Block a user