Windows10下的yolov5环境搭建
- 一、安装Anaconda3
- 二、打开Anaconda Prompt,设置conda为清华源
- 三、创建虚拟环境
- 四、安装cuda和cudnn
- 五、安装各类库
- 六、安装pytorch和torchvision
- 七、下载源码
- 八、下载模型
- 九、运行
- 十、其它疑难杂症
环境:Windows10
时间:2020.12.17
下一篇:yolov5s人脸检测模型训练实例
一、安装Anaconda3
anaconda(开源版)
可以安装在D盘
二、打开Anaconda Prompt,设置conda为清华源
Anaconda Prompt(在开始里,作用等效于cmd)
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu/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/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/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/qq_45389690/article/details/111338525
下一篇:yolov5s人脸检测模型训练实例
更多推荐
【目标识别检测】Windows10下yolov5环境搭建
发布评论