This commit is contained in:
HuangHai
2026-01-14 07:05:03 +08:00
parent e25aa5b756
commit a3347b9340
9 changed files with 14 additions and 66 deletions

View File

@@ -152,9 +152,10 @@ class ReadImageKit:
每个对象包含:
- "name": 场站名称 (仅提取名称文字)
- "address": 场站地址 (通常在名称下方,请完整提取地址文字)
- "is_valid": true/false (是否为真实的场站卡片)
注意:仅提取名称,不要包含距离或价格。
注意:仅提取名称和地址,不要包含距离或价格。
"""
else:
# 如果 CV 没检测到,退回到纯 VLM 模式
@@ -163,6 +164,7 @@ class ReadImageKit:
忽略顶部的筛选栏,仅提取下方重复出现的场站卡片。
输出格式为 JSON 数组,每个对象包含:
- "name": 场站名称
- "address": 场站地址 (请仔细识别并提取场站的详细地址信息)
- "point": 场站卡片的中心点击坐标 [x, y]
- "bbox": 场站卡片的边界框 [x1, y1, x2, y2]
"""
@@ -184,6 +186,7 @@ class ReadImageKit:
if res and (res.get("is_valid") is True or (res.get("name") and res.get("is_valid") is not False)):
final_stations.append({
"name": res.get("name"),
"address": res.get("address"),
"point": [(bbox[0] + bbox[2]) // 2, (bbox[1] + bbox[3]) // 2],
"bbox": bbox
})