OWL-ViT（Open-World Localization Vision Transformer）是由Google Research提出的开放词汇目标检测模型，结合CLIP的视觉-语言预训练能力与目标检测框架，实现无需微调即可检测任意文本描述的新类别物体。其核心将CLIP的图像编码器与文本编码器扩展为检测架构，通过对比学习对齐图像区域与文本描述，动态生成候选框并预测匹配分数。模型基于Vision Transformer（ViT）提取全局特征，利用轻量级检测头定位目标，支持零样本（Zero-Shot）迁移至未知类别（如“紫色独角兽玩偶”或“带logo的背包”），在LVIS等开放数据集上展现强泛化能力。适用于动态场景下的新兴物体检测（如零售商品识别、机器人交互）。

源模型

• 输入尺寸: [[1,3,768,768]], [[1,16],[1,16]],[[1,24,24,768],[1,512],[1,16]]
• 参数量: 84.92M, 60.46M, --
• 模型大小: 339.91M, 242.06M, 1.51M
• 输出尺寸：[[1,24,24,768],[1,576,4]], [[1,512]], [[1,576,1]]

源模型工程请访问：OWL-ViT

在右侧性能参考板块中点击模型转换参考查看模型转换步骤

Model Farm 所提供的模型性能基准测试以及示例代码皆基于阿加犀AidLite SDK实现

SDK安装

详情请参考AidLite开发者文档

• 安装AidLite SDK

# install aidlite sdk c++ api
sudo aid-pkg -i aidlite-sdk

# install aidlite sdk python api
python3 -m pip install pyaidlite -i https://mirrors.aidlux.com --trusted-host mirrors.aidlux.com

• 验证AidLite SDK

# aidlite sdk c++ check
python3 -c "import aidlite; print(aidlite.get_library_version())"

# aidlite sdk python check
python3 -c "import aidlite; print(aidlite.get_py_library_version())"

推理示例代码

• 点击模型 & 代码下载模型文件和推理代码包，文件结构如下

/model_farm_{model_name}_aidlite
    
    |__ models # folder where model files are stored

    |__ python # aidlite python model inference example

    |__ cpp # aidlite cpp model inference example

    |__ README.md