基于 DINO 与 Chinese-CLIP 的自动驾驶语义检索系统架构

整体架构

组件 原始模型 升级后模型 优势
开集检测 通用目标检测模型 DINO 更强的开放世界检测能力,无需预定义类别
CLIP 模型 英文 CLIP 达摩院 Chinese-CLIP 支持中文语义,适配本地化查询

🛠️ 二、关键组件详解

1. 📸 前视广角图片

  • 来源:车载摄像头(ADAS 系统)
  • 特点:视野宽、包含多个物体(车、人、路牌等),但无标注

2. 🔍 DINO 模型(用于开集检测)

是什么?
  • DINO(DETR with No Object Queries )是 Meta 提出的一种基于 Transformer 的 自监督学习框架
  • 实际应用中常指 DINOv2 或 DINOv2-Huge,具有强大的图像特征提取能力
  • 可用于:
    • 无监督目标分割
    • 零样本分类
    • 开放词汇检测
如何用在本项目中?

✅ 使用 DINO 的 视觉编码器(ViT-Huge) 提取图像特征,再配合 Open-Vocabulary Detection 方法实现"开集检测"

推荐方式:
  • 使用 DINOv2 + Open-Vocabulary Detector (如 Grounding DINO)

  • 输入一张广角图 → 输出所有检测框及其描述性标签(如:"红色SUV"、"骑电动车的人")

    from groundingdino.models import build_model
    from groundingdino.util.sliding_window import sliding_window_inference

    model = build_model(args)
    outputs = model(image, text_prompts=["car", "pedestrian", "traffic sign"])

⭐ 优点:可检测未见过的类别;输出带语义信息的目标区域


3. 🖼️ 目标物级别图片

  • 从 DINO 输出的每个检测框中裁剪出独立图像
  • 示例:每辆车、每个人、每个标志都变成单独的小图
  • 存储格式:(bbox, image_crop) 对

4. 🧠 达摩院 CLIP 模型(damo/chinese-clip)

官方名称:
  • damo/chinese-clip-vit-base-patch16
  • 在 ModelScope 上开源,支持中文文本和图像编码
功能:
  • 将图像或文本转换为 512 维 embedding 向量
  • 图像和文本共享同一语义空间,便于计算相似度
使用示例:
复制代码
from modelscope.pipelines import pipeline
clip_pipeline = pipeline(
    task='multi_modal_embedding',
    model='damo/chinese-clip-vit-base-patch16'
)

# 编码图像
img_emb = clip_pipeline({'image': 'car.jpg'})['image_embedding']

# 编码文本
text_emb = clip_pipeline({'text': '一辆黑色奥迪轿车'})['text_embedding']

5. 💾 向量数据库

  • 推荐工具:FAISS 、Milvus 、Pinecone 、Weaviate

  • 存储结构:

    复制代码
    {
      "id": "001",
      "embedding": [0.1, 0.2, ..., 0.9],
      "metadata": {
        "timestamp": "2025-04-05T10:00:00Z",
        "camera_id": "cam_01",
        "bbox": [x1,y1,x2,y2],
        "description": "黑色奥迪"
      }
    }

6. 🔎 数据向量搜索(双通道输入)

支持两种查询方式:

表格

查询类型 流程
文本查询 用户输入中文 → CLIP 编码 → 查找最相似 embedding
图片查询 用户上传图片 → CLIP 编码 → 查找匹配项
最终步骤:向量相似度 rerank
复制代码
# 1. 初步召回 Top-K 结果(ANN 搜索)
candidates = vector_db.search(query_embedding, k=100)

# 2. 对候选结果重新计算余弦相似度
scores = []
for cand in candidates:
    score = cosine_sim(query_embedding, cand.embedding)
    scores.append(score)

# 3. 按分数排序,返回 top-N
results = sorted(zip(candidates, scores), key=lambda x: x[1], reverse=True)[:10]

🚀 三、完整流程图更新版

复制代码
┌──────────────────────────────────────┐
│           数据向量化入库               │
├──────────────────────────────────────┤
│  前视广角图片 ──→ DINO 模型 ──→ 目标物级别图片 ──→ 达摩院 CLIP ──→ embedding ──→ 向量数据库
└──────────────────────────────────────┘
                          ↑
                          │
┌──────────────────────────────────────┐
│           数据向量搜索               │
├──────────────────────────────────────┤
│ 搜索目标文本 ──→ 达摩院 CLIP ──→ embedding ──┐
│ 搜索目标图片 ──→ 达摩院 CLIP ──→ embedding ──┤
│                                          ↓
│                                向量相似度rerank
│                                          ↓
│                                  匹配结果
└──────────────────────────────────────┘
相关推荐
许雪里3 分钟前
XXL-AI|AI应用开发平台(Agent编排、多供应商、「MCP + SKILL + RAG」扩展、工程化底座)
人工智能·agent·ai编程
Csvn5 小时前
第 28 章 案例四 多智能体协作系统
人工智能·aigc·agent
IT_陈寒5 小时前
Java中equals方法比了个寂寞?原来这才是正确的重写姿势
前端·人工智能·后端
吴佳浩5 小时前
Agent 怎么做自动化评测?构建端到端的 Agent Evaluation 体系
人工智能·agent·ai编程
火山引擎开发者社区5 小时前
火山引擎云数据库 TiDB 版公测开启,MySQL 架构升级的一站式选择
人工智能
代码方舟5 小时前
Java数据工程:利用天远全网运营商三要素优化线上实名认证合规体验
java·人工智能
Csvn5 小时前
第 27 章 案例三 自动化工作流 Agent
人工智能·aigc·agent
知几蜗牛5 小时前
AI眼镜把记忆放上云,怎样证明云端也看不见?
人工智能
知几蜗牛5 小时前
训练数据越多越好吗?用LeRobot讲清数据质量与版本化
人工智能
知几蜗牛5 小时前
PR合并慢,别再只看平均时长:GitHub把等待拆成了三段
人工智能