- 创建虚拟环境
conda create --name yolox python=3.9.0
2.安装torch=1.11.0
#如果下载过慢,可以加上百度镜像源:
pip install -i http://pip.baidu-int.com/simple/ torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
安装完torch后,需要安装mmengine等OpenMMLab中的库,为了方便安装,可以从pip转为mim进行安装
#安装openmim,之后可以通过mim安装mmopenlab下的安装包
pip install -U openmim
mim install mmcv==2.0.0
mim install mmdet==3.0.0
mim install mengine==0.10.3
mim install mmyolo==0.6.0
安装mmyolo
git clone https://github.com/open-mmlab/mmyolo.git
cd mmyolo
# 安装依赖
pip install -r requirements/albu.txt
# 安装mmyolo
mim install -v -e .
# "-v" 指详细说明,或更多的输出
# "-e" 表示在可编辑模式下安装项目,因此对代码所做的任何本地修改都会生效,从而无需重新安装。
际运行时还需要一些包:supervision和transformers
#这两个包直接在pip上安装即可:
pip install -i http://pip.baidu-int.com/simple/ supervision==0.18.0
pip install -i http://pip.baidu-int.com/simple/ transformers==4.38.1
所有需要的依赖环境就已经安装完啦,现在你可以直接clone项目然后开始尝试了!
git clone https://github.com/AILab-CVC/YOLO-World.git
cd /YOLO-World
#修改以下相关文件目录即可
python image_demo.py path/to/config path/to/weights image/path/directory 'person,dog,cat' --topk 100 --threshold 0.005 --output-dir demo_outputs