【目标识别检测】Windows10下yolov5环境搭建

环境：Windows10
时间：2020.12.17
下一篇：yolov5s人脸检测模型训练实例

一、安装Anaconda3

anaconda(开源版)
可以安装在D盘

二、打开Anaconda Prompt，设置conda为清华源

Anaconda Prompt（在开始里，作用等效于cmd）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

三、创建虚拟环境

conda create -n yolov5 python=3.7
activate yolov5

“yolov5”是环境名，可以自定义。python版本强烈建议3.7，须指定版本；如果不指定，默认安装3.8的。
之后所有关于yolov5的操作都在这个环境里进行，打开Anaconda Prompt后进入环境的方法为：activate yolov5

四、安装cuda和cudnn

conda install cudatoolkit==10.2.89
conda install cudnn==7.6.5

如果选择其他版本cuda，cuda版本必须和cudnn对应。由于最新的官方yolov5对各类库的版本要求，不建议使用10.0以下的cuda。

五、安装各类库

这是官方给出的requirements.txt

这里给出我自己的库版本：
numpy：1.19.2
opencv-python：4.4.0.46
Cython：0.29.21
pillow：8.0.1
matplotlib：3.3.2
pyyaml：5.3.1
tensorboard：2.3.0
scipy：1.5.2
tqdm：4.54.0
pandas：1.1.3
seaborn：0.11.0

安装时，先conda install 库名，看看版本是否符合要求，若默认版本不符合要求：
方法一：去anaconda网站上找相应版本的库，然后安装。
以opencv为例：https://anaconda.org/conda-forge/opencv

conda install -c conda-forge opencv

方法二：去清华源或库的官网上下载.tar.bz2的压缩包
在anaconda Prompt中，进入yolov5环境，cd到压缩包所在目录，输入：

conda install --offline XXX.tar.bz2

这里给一个最快安装cv4.4的方法：pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python

六、安装pytorch和torchvision

（两者版本也要匹配）
我用的是pytorch1.6.0和torchvision0.7.0。别的版本我也试过，强烈建议按照此版本。
注意: pytorch和torchvision在安装完所有其它库后最后安装！否则会报错！

七、下载源码

yolov5源码

八、下载模型

yolov5模型（.pt文件）
把模型放到源码文件的./weights文件下

九、运行

图片：python detect.py --source ./inference/images/ --weights ./weights/yolov5s.pt --conf 0.4
摄像头：python detect.py --source 0 --weights ./weights/yolov5s.pt --conf 0.4
自己的模型：python detect.py --source 0 --weights ./模型地址 --conf 阈值

十、其它疑难杂症

详询此文
https://blog.csdn.net/qq_45389690/article/details/111338525

下一篇：yolov5s人脸检测模型训练实例