第一天,yolov5 dome 安装及使用

340 阅读1分钟

目标: 实现人脸头像捕获项目的部署

image.png

老手1天左右,新手可以2天。

Anaconda安装

参考文章: blog.csdn.net/qq_42257666… blog.csdn.net/weixin_4285… 群网盘里面有,要求 :Anaconda3 安装完成之后

在cmd如输入 conda activate base #base 是conda 的基础环境 如果像我这样报错了 image.png

那么以后启动conda 先输入 activate 再输入命令, 出现下面这个即为安装成功

image.png

之后创建虚拟环境 conda create -n yolov5 python=3.8 #yolov5 虚拟环境名字 使用3.8版本python

进入虚拟环境 conda activate yolov5

image.png

yolov5下载

github 链接 github.com/ultralytics…

image.png

下载zip解压即可

依赖环境的安装

cuda和cudnn的安装

参考文章: blog.csdn.net/m0_45447650…

pytorch安装

在 conda activate yolov5 命令之后使用

pip install torch==1.6.0 -f download.pytorch.org/whl/torch_s… pip install torchvision==0.7.0 -f download.pytorch.org/whl/torch_s… #最好逐条执行

在虚拟环境中启动python,输入指令测试是否安装成功 python import torch torch.cuda.is_available() #返回True则说明已经安装了cuda from torch.backends import cudnn cudnn.is_available() #返回True则说明已经安装了cuDNN

image.png 我的是False 但是不影响使用,如果影响你使用了。建议重装驱动

其余环境的安装

在解压后的yolov目录中

image.png

image.png

torch在其requirements.txt是有pytorch安装,但是pip在无法直接安装这个版本,上个步骤是直接从官网中下载

此处需要将其注掉

conda中来到项目文件 pip install -i pypi.tuna.tsinghua.edu.cn/simple -U -r requirements.txt

image.png

等待安装完成即可

测试

在项目目录下 python detect.py --source 0 #使用官方训练好的模型 0 为本机摄像头 官方文档有更详细说明

完整版本:

image.png

如果报错为:

问题:FileNotFoundError: [Errno 2] No such file or directory: ‘yolov5s.pt’

网速太慢,建议直接下载对应文件放在根文件夹下。

英文下 按 q 退出,或者直接叉掉

最后结果:

image.png