license: agpl-3.0 pipeline_tag: zero-shot-object-detection tags:

• model_hub_mixin
• pytorch_model_hub_mixin

模型描述

YOLOE：实时视觉全能模型

• 论文链接: https://arxiv.org/abs/2503.07465
• GitHub仓库: https://github.com/THU-MIG/yoloe

安装指南

pip install git+https://github.com/THU-MIG/yoloe.git#subdirectory=third_party/CLIP
pip install git+https://github.com/THU-MIG/yoloe.git#subdirectory=third_party/ml-mobileclip
pip install git+https://github.com/THU-MIG/yoloe.git#subdirectory=third_party/lvis-api
pip install git+https://github.com/THU-MIG/yoloe.git

wget https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt

预测功能

详见 https://github.com/THU-MIG/yoloe?tab=readme-ov-file#prediction。

迁移学习

详见 https://github.com/THU-MIG/yoloe?tab=readme-ov-file#transferring。

验证流程

详见 https://github.com/THU-MIG/yoloe?tab=readme-ov-file#validation。

训练方法

详见 https://github.com/THU-MIG/yoloe?tab=readme-ov-file#training。

BibTeX引用信息

@misc{wang2025yoloerealtimeseeing,
      title={YOLOE：实时视觉全能模型}, 
      author={王傲、刘立浩、陈辉、林子佳、韩军功、丁贵广},
      year={2025},
      eprint={2503.07465},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2503.07465}, 
}