'commit'
This commit is contained in:
@@ -137,6 +137,43 @@ def is_background_dimmed(image_path, threshold=80):
|
||||
return avg_brightness < threshold
|
||||
|
||||
|
||||
from Util.EasyOcrKit import get_easyocr_reader
|
||||
|
||||
# 预加载 EasyOCR Reader (单例模式)
|
||||
def get_ocr_reader():
|
||||
return get_easyocr_reader(gpu=True)
|
||||
|
||||
|
||||
def detect_price_info_container_cv(image_path):
|
||||
"""
|
||||
使用 OCR 精准定位详情页中的价格入口文本(“全部时段”)。
|
||||
返回: [x1, y1, x2, y2] 归一化坐标 (0-1000),如果未找到则返回 None
|
||||
"""
|
||||
img = read_image(image_path)
|
||||
if img is None:
|
||||
return None
|
||||
h, w = img.shape[:2]
|
||||
|
||||
keywords = ['全部时段']
|
||||
|
||||
try:
|
||||
reader = get_ocr_reader()
|
||||
# 获取所有识别结果
|
||||
results = reader.read_text(img)
|
||||
|
||||
for (quad, text, prob) in results:
|
||||
# 检查是否包含关键字
|
||||
if any(kw in text for kw in keywords) and prob >= 0.5:
|
||||
# 使用封装后的方法计算归一化矩形
|
||||
res = reader.get_normalized_rect(quad, w, h)
|
||||
logger.info(f"[OCR识别] 找到文本: '{text}', 置信度: {prob:.4f}, 归一化坐标: {res}")
|
||||
return res
|
||||
except Exception as e:
|
||||
logger.error(f"OCR 识别发生异常: {e}")
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def detect_rabbit_ad_close(image_path, debug_dir=None):
|
||||
"""
|
||||
通过图形学算法检测“新电兔AI”广告的关闭按钮
|
||||
|
||||
Reference in New Issue
Block a user