YOLO-World部署

01 前置操作

默认Anaconda、pytorch、pycharm均安装配置成功

1.Anaconda环境

首先创建Anaconda环境，打开Anaconda Prompt，输入

conda create -n your_env_name   # your_env_name 为你虚拟环境名

若要指定python版本号，则输入

conda create -n your_env_name python=3.x # your_env_name 为你虚拟环境名

查看环境列表，输入

conda env list

环境已存在

conda activate your_env_name

激活虚拟环境

2.YOLO-World下载

yolo-world下载：https://github.com/AILab-CVC/YOLO-Worldhttps://github.com/AILab-CVC/YOLO-World

权重下载：https://docs.ultralytics.com/zh/models/yolo-world/#available-models-supported-tasks-and-operating-modeshttps://docs.ultralytics.com/zh/models/yolo-world/#available-models-supported-tasks-and-operating-modes

下以YOLOv8l-worldv2为例

02 环境配置

使用pycharm打开下载好的YOLO-World项目，将下载好的yolov8l-worldv2.pt放入

在终端输入

pip install ultralytics

03 运行

创建一个python文件，同时在data文件夹下放入需检测的图片

from ultralytics import YOLOWorld

model = YOLOWorld('yolov8l-worldv2.pt')

model.set_classes(['person', 'helmet'])

results = model.predict('data/img_1.png')

results[0].show()

成功