Yolo-wrold安装

最新推荐文章于 2024-05-14 18:32:26 发布

沐雨清风 จุ๊บ

最新推荐文章于 2024-05-14 18:32:26 发布

阅读量404

点赞数 3

文章标签： YOLO

本文链接：https://blog.csdn.net/jiayou2021/article/details/136935184

版权

本文介绍了如何在Python环境中创建一个用于深度学习项目，特别是YOLO-v3开发的虚拟环境，包括安装Torch、OpenMMLab库（如mmcv、mmdet和mmyolo），以及添加额外依赖如supervision和transformers。通过gitclone获取YOLO-World项目并配置参数进行图像识别演示。

摘要由CSDN通过智能技术生成

创建虚拟环境

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