仓库:https://github.com/AILab-CVC/YOLO-World
下载权重:
仓库下载和环境设置
- 下载仓库:使用以下命令从 GitHub 上克隆仓库:
git clone --recursive https://github.com/AILab-CVC/YOLO-World.git
-
创建并激活环境:首先,创建一个名为
yoloword1
的 Conda 环境,并激活它:
conda create -n yoloword1 python=3.8 -y
conda activate yoloword1
安装依赖项
- 安装 PyTorch 和 Gradio:使用以下命令安装 PyTorch 和 Gradio:
pip install torch wheel gradio==4.16.0
-
安装 YOLO-World 包:进入项目目录并使用
-e
参数来安装可编辑模式下的 YOLO-World 包:
cd YOLO-World/
pip install -e .
运行示例
- 运行示例代码:使用以下命令运行示例代码。确保你提供了正确的路径和参数:
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