Ubuntu16.04LTS Yolov5 仅CPU版本安装

【安装时间】2021年4月,使用的yoloV5版本是github上最新的版本,如果后面隔了很久用这个方法可能会失败。
【系统环境】Ubuntu16.04LTS+python3.8.5+pencv-python==4.1.2.30+anaconda3-2020.11-Linux-x86_64.sh(这个版本)
1. 安装准备工作
1.1. anaconda安装、使用看这个链接
https://blog.csdn.net/qq_43769747/article/details/115546639?spm=1001.2014.3001.5501
因为部分需要,不能使用GPU,只能使用CPU运行。
因此这个安装不需要使用cuda,所以pytorch安装有部分差异。GPU版本安装不能用!!!!
1.2 下载yolov5模型
GitHub链接:https://github.com/ultralytics/yolov5
1.3 下载yolov5s权重文件
GitHub链接:https://github.com/ultralytics/yolov5/releases
这个权重文件要放到yolov5模型的文件夹下
1.3 创建虚拟环境yolov5
见1.1的链接

2. 准备安装
2.1 查看安装要求
安装要求在下载链接的requirements.txt 文档中,直接使用这个安装可能会出错。
2.2 准备安装
【CPU版本主要是第四步的差异】
(1) matplotlib>=3.2.2 pillow
输入命令:pip install matplotlib
在这里插入图片描述
(2)PyYAML.>=5.3.1
输入命令:pip install pyyaml
Ubuntu16.04LTS Yolov5 仅CPU版本安装_第1张图片
(3) scipy>=1.4.1 numpy >= 1.18.5
输入命令:pip install scipy
Ubuntu16.04LTS Yolov5 仅CPU版本安装_第2张图片
(4)torch>=1.7.0 torchvision>=0.8.1
这一步有人使用命令:【现在不能用】
pip install torch1.5.1+cpu torchvision0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
在实际使用这个命令安装不能正确运行现在的yolo模型,版本太低了
正确使用方法:
进入pytorch官网
参考官网链接:https://pytorch.org/get-started/previous-versions/
找到对应自己python要求的命令。
【正确使用命令】
pip install torch1.8.0+cpu torchvision0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
在这里插入图片描述
(5)tensorboard>=2.4.1
输入命令:pip install tensorboard
在这里插入图片描述
(6)tqdm>=4.41.0
输入命令:pip install tqdm
Ubuntu16.04LTS Yolov5 仅CPU版本安装_第3张图片
(7)安装requests
输入命令:pip install requests
在这里插入图片描述
(8)安装seaborn
输入命令:pip install seaborn
在这里插入图片描述

3. 测试yolo模型
输入命令:python detect.py --source data/images --weights yolov5s.pt --conf 0.25
Ubuntu16.04LTS Yolov5 仅CPU版本安装_第4张图片
这个加载时间明显比yolov3快很多,v3的权重文件加载太慢了而且很大。

你可能感兴趣的:(CPU版本,YoloV5,python,pytorch,神经网络,深度学习)