准备工作
在此之前,先把编译器选一下还有anaconda下载好了
深度学习环境------anaconda,anaconda的下载,安装与使用(无废话)_anaconda 机器学习环境下载-CSDN博客
还有就是大家都用的pycharm或者vscode,这俩都可以,pycharm下载社区版的就够用了。
下载好之后,为了使用方便,可以安装一下中文插件,这在csdn上搜索一下就有很多教程的。
在开始之前,先看一下自己电脑的配置,看看有没有显卡。
看自己的设备
打开任务管理器,点击性能,列表里就有
我的是RTX 3050的(具体如何看,我的这里的GPU0 下面写的是Inter这种的,表示的是这是一块核显,只是cpu带的,这种和独立显卡还是不一样的,我们主要看的是独立显卡,就是下面的写有NVIDIA的这种的,在深度学习中主要用的就是这种显卡)
关于看显存的大小,看这里,GPU这里往下拉
我的显卡的显存就是4G大小的
有显卡的看这
有显卡的,继续往下看,没有显卡的跳过这一步。
打开终端命令行,如果下载好了anaconda的话,打开这个
这俩随便一个都行,
打开之后,输入
nvidia-smi
输入后按回车,会显示
这个我画圈的地方-说明的意思是当前的驱动下所能支持的最高cuda版本,我的是11.7
就是说我安装的cuda版本,下载的时候我最大只能选择11.7版本的,不能再大了
这里说一下CUDA是啥
CUDA就是在跑深度学习代码的时候,这个东西是专门去让显卡工作的东西,给显卡分配任务的。
具体安装看这个深度学习环境----CUDA+cudnn(无废话) 安装cuda,cudnn_安装cudnn-CSDN博客
项目代码下载,查看pytorch的版本
进入GitHub网站,搜索yolov7,第一个就是了
点进去
下载压缩包,解压了就可以了
GitHub网站是外国网站,可能你的网进不去,我这里给一个百度云链接
链接:https://pan.baidu.com/s/1aAHhYn3f6S6mI2LJrnHFiQ?pwd=peyt
提取码:peyt
下载之后,解压,
这个文件放的话,最好都是英文名字。
解压后,打开文件,里面有
这就是下载的yolov7项目所需要的环境配置要求,打开看一哈
这里说的就是pytorch版本要大于等于1.7.0,不能等于1.12.0
环境配置
配置环境的时候,由于直接使用下载指令的话是从外网下载的,速度会慢,这里建议就是加个镜像,看这里
在此之前,先创一个环境
建议先看一下这个深度学习环境------anaconda,anaconda的下载,安装与使用(无废话)_anaconda 机器学习环境下载-CSDN博客
下面我再以yolov7为参考写一下具体操作
我先创一个名为yolov7的环境,选择的python版本为3.8
conda create -n yolov7 python=3.8
具体操作如下:
打开
输入
conda create -n yolov7 python=3.8
按回车
会出现
输入Y,按回车
等会
出现了
进入环境里面
输入
conda activate yolov7
如果显示了画圈的名字,就进入了现在的环境了。
有显卡的看这
由于pytorch有版本要求,我们就在自己cuda版本能选的范围内去选择合适的pytorch版本就可以了,我的cuda版本是11.6,
打开pytorch网站, 直接搜torch就可以了
具体操作看这个
深度学习环境-------pytorch安装------超详细_pytorch==1.13.1 支持版本-CSDN博客
我在这选择的是1.13.1版本的,我在环境中输入
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
回车就可以了。前提是,现在要进入自己创建的环境里!!!!!!
等一段时间安装
出现这个就代表着安装成功了。
然后就是安装其他的东西了
这里的话,先确定一下这个requirements.txt的位置,
右击属性
这个复制下来
我的是:C:\Users\Lenovo\Desktop\yolov7\yolov7-main
先把斜杠换成/的,然后就是在后面加上/requirements.txt
我的就是:C:/Users/Lenovo/Desktop/yolov7/yolov7-main/requirements.txt
然后终端进入到自己创建的环境中,输入
pip install -r C:/Users/Lenovo/Desktop/yolov7/yolov7-main/requirements.txt
这里我的路径是这样的,具体看自己的路径改一下就可以了
等一会
现在就是在安装了
出现这个,就代表着安装好了
无显卡的看这
进入到直接的环境里,直接按照这个requirements.txt就可以了,自动安装的就是cpu版本的
看一下上面的安装操作就可以了。
训练自己的数据集
对于数据集的问题,先看这个
至于训练操作的话,就和yolov5没啥差别,train.py就是训练的,把参数调一下,看我其他的文章