机器学习AI算法工程 | YOLO-World开集目标检测，可直接上手

本文来源公众号“机器学习AI算法工程”，仅用于学术分享，侵权删，干货满满。

原文链接：YOLO-World开集目标检测，可直接上手

1. 前言

        关于Demo：尽管YOLO-World官方给出了在线试用的Demo：https://huggingface.co/spaces/stevengrove/YOLO-World ，但还是不如在自己电脑上离线运行一遍来得直接。恰好，近期Ultralytics（YOLOv8）也新增了对YOLO-World的支持，我们不如直接体验一把。

        关于YOLO-World：

        其是一个使用开放词汇进行目标检测的新框架，且是以YOLOv8框架为detector，所以其特点就继承了YOLO系列，也即轻量、快速、性能好。另外，既然是文本和图片一起作为输入，那么就需要有一个文本embedding的模块，这里用的是CLIP，其将用户输入的词汇列表转换为特征向量，与输入图像一起进行推理。

2. 安装（更新）

        由于YOLO-World主要包括两