【深度学习】yolo-World,数据标注,zeroshot,目标检测

时间:2024-04-28 07:17:47

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

下载权重:
在这里插入图片描述

仓库下载和环境设置

  1. 下载仓库:使用以下命令从 GitHub 上克隆仓库:
git clone --recursive https://github.com/AILab-CVC/YOLO-World.git
  1. 创建并激活环境:首先,创建一个名为 yoloword1 的 Conda 环境,并激活它:
conda create -n yoloword1 python=3.8 -y
conda activate yoloword1

安装依赖项

  1. 安装 PyTorch 和 Gradio:使用以下命令安装 PyTorch 和 Gradio:
pip install torch wheel gradio==4.16.0
  1. 安装 YOLO-World 包:进入项目目录并使用 -e 参数来安装可编辑模式下的 YOLO-World 包:
cd YOLO-World/
pip install -e .

运行示例

  1. 运行示例代码:使用以下命令运行示例代码。确保你提供了正确的路径和参数:
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