仓库：https://github.com/AILab-CVC/YOLO-World

下载权重：

仓库下载和环境设置

1. 下载仓库：使用以下命令从 GitHub 上克隆仓库：

git clone --recursive https://github.com/AILab-CVC/YOLO-World.git

2. 创建并激活环境：首先，创建一个名为 yoloword1 的 Conda 环境，并激活它：

conda create -n yoloword1 python=3.8 -y
conda activate yoloword1

安装依赖项

3. 安装 PyTorch 和 Gradio：使用以下命令安装 PyTorch 和 Gradio：

pip install torch wheel gradio==4.16.0

4. 安装 YOLO-World 包：进入项目目录并使用 -e 参数来安装可编辑模式下的 YOLO-World 包：

cd YOLO-World/
pip install -e .

运行示例

5. 运行示例代码：使用以下命令运行示例代码。确保你提供了正确的路径和参数：

python image_demo.py ./configs/pretrain/yolo_world_v2_l_clip_large_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_800ft_lvis_minival.py /data/xiedong/yolo_world/yolo_world_v2_l_clip_large_o365v1_goldg_pretrain_800ft-9df82e55.pth /data/xiedong/yolo_world/testtmp 'fire,smoke' --topk 2 --threshold 0.01 --output-dir /data/xiedong/yolo_world/testtmpdemo_outputs

标注效果展示

标注效果：

下载请看这里：

https://docs.qq.com/sheet/DUEdqZ2lmbmR6UVdU?tab=BB08J2