yolov7训练自己的数据集(代码片段)

我把把C 我把把C     2022-12-08     540

关键词:

yolov7训练自己的数据集

一.安装yolov7并配置

1.安装Anaconda

miniconda清华源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

Miniconda3-py38_4.9.2-Windows-x86_64.exe:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_4.9.2-Windows-x86_64.exe

2.创建虚拟环境

在开始中找到Anaconda并打开

1.创建

conda create -n yolov7 python=3.7 -y

2.激活

conda activate yolov7


3.退出环境

conda deactivate

3.下载yolov7源码

方法一:yolov7官网:https://github.com/WongKinYiu/yolov7

方法二:打开cmd输入以下命令

git clone https://gitee.com/YFwinston/yolov7.git
cd yolov7

4.下载权重

方法一:百度网盘下载https://pan.baidu.com/s/1DbG5bI_Sm1U62poE9yb5dQ
提取码:cvss
方法二:官网下载(速度很慢不推荐)
打开官网后往下划找到Testing
yolov7官网:https://github.com/WongKinYiu/yolov7

权重下载完成后将其拖入下载好的yolov7文件夹中

5.安装依赖

打开yolov7文件夹后确认文件夹中是否有 requirements.txt 文档
在文件夹上方输入cmd按下回车打开如图

在打开的cmd中输入以下命令(可能需要安装一会)

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

二.运行目标检测

打开yolov7所在文件夹
拍摄一张自己的帅照放入 inference文件夹中的images文件夹中

在yolov7文件夹中打开cmd并激活yolov7环境然后运行运行detect.py进行检测

conda activate yolov7
python detect.py --weights yolov7x.pt --conf 0.25 --img-size 640 --source inference/images/horses.jpg 

horses.jpg可更改为自己的图片名

如果报错:ModuleNotFoundError: No module named 'cv2’在cmd中输入

pip install opencv-contrib-python


运行的结果在 runs/detect 文件夹中

三.训练自的数据集

1.创建所需文件


如图我们在yolov7下创建datasets文件夹然后创建images 和labels文件夹,一个是拿来放图片的,一个是拿来存images文件夹中处理jpg图片之后的txt数据

将自己所需要训练的图片放进 images/train和val下两个文件夹中放入差不多数量的图片

2.下载labelImg标注工具

参考::::labelImg使用教程
先打开Anaconda

输入以下命令安装

pip install labelImg

安装成功后cmd中输入labelImg即可打开

labelImg




按下w选中你要标注的物体即可

3.配置训练的相关文件

总共有两个文件需要配置,一个是/yolov7/cfg/training/yolov7.yaml,这个文件是有关模型的配置文件;一个是/yolov7/data/coco.yaml,这个是数据集的配置文件。
1.第一步,复制yolov7.yaml文件到相同的路径下,然后重命名,我们重命名为yolov7-xxx.yaml(起一个你想要的名字)然后打开

2.复制coco.yaml文件到相同的路径下,然后重命名,我们命名为xxx.yaml。(起一个你喜欢的名字)打开后进行更改如下图

ok

四.正式训练

此时我们在yolov7文件夹路径下cmd,并且激活虚拟环境,输入指令

conda activate yolov7

激活环境后输入

python train.py --weights yolov7x.pt --cfg cfg/training/yolov7-xxx.yaml --data data/xxx.yaml --batch-size 8 --epoch 300 --device 0

注意一定要将xxx改为自己的文件名称
这里我们使用gpu进行训练若用cpu去掉 --device 0 即可

对参数解释

–cfg 接受模型配置的参数
–data 接收数据配置的参数
–device 0 训练类型,我是一块GPU 所以用0
–batch-size 8 GPU内存大小决定
–epoch 训练次数,建议300
–weights 训练的权重

五.使用自己的权重

跟detect一样,进入虚拟环境输入权重路径和图片路径就ok了,指令如下

我是把best.pt直接拉到了yolov7文件夹路径下面,你们刚刚训练出来的在runs/train/circle/weights/best.pt
命令如下

python detect.py --weights best.pt --source xxx --device 0

注意将xxx换成自己的图片路径
ok

win10安装yolov7训练自己的数据集(代码片段)

1.安装https://www.pudn.com/news/62ddf431864d5c73acfb1a09.html下载yolov7源码yolov7修改requirements.txt修改torch以及torchvision版本,这里直接指定好版本让其版本对应安装pipinstall-rrequirements.txt-fhttps://download.pytorch.org/w 查看详情

win10安装yolov7训练自己的数据集(代码片段)

1.安装https://www.pudn.com/news/62ddf431864d5c73acfb1a09.html下载yolov7源码yolov7修改requirements.txt修改torch以及torchvision版本,这里直接指定好版本让其版本对应安装pipinstall-rrequirements.txt-fhttps://download.pytorch.org/w 查看详情

yolov7pytorch(代码片段)

yolov7主干部分结构图:yolov7主干yolov7数据集处理代码:yolov7数据集处理代码yolov7训练参数解释:yolov7训练参数【与本文代码有区别】yolov7训练代码详解:yolov7训练代码详解目录训练自己的训练集训练自己的训练集... 查看详情

yolov7训练自己的数据集及报错处理(代码片段)

...#################D:\\Anaconda3\\envs\\py38torch_gpu\\python.exeD:\\needed\\yolov7-main\\train.py--weightsweights/yolov7.pt--cfgcfg/training/yolov7.yaml--datadata/datasets.yaml--device0--batch-size8--epoch5 YOLOR 2022-9-16torch1.9.0+cu111CUDA:0(NVIDIAGeForceRTX3060Ti,8191.5MB)Namespace(adam=... 查看详情

yolov7:从搭配环境到训练自己的数据集(代码片段)

一.下载源码在github中下载Yolov7源码并解压缩YOLOv7源码数据标注可以参考这篇文章https://blog.csdn.net/qq_39779449/article/details/110172948?spm=1001.2014.3001.5502标签格式二.配置环境整体配置树结构安装Yolov7gpu环境在Anconda中创造Yolov7环境安装... 查看详情

yolov7训练自己的数据集(超详细)(代码片段)

介绍2022年7月,YOLOv7来临,论文链接:https://arxiv.org/abs/2207.02696代码链接:GitHub-WongKinYiu/yolov7:Implementationofpaper-YOLOv7:Trainablebag-of-freebiessetsnewstate-of-the-artforreal-timeobjectdetectors废话不多说,赶紧上车!文件配置... 查看详情

yolov7训练自己的数据集(超详细教程)(代码片段)

...录一,准备数据集1.1挑选照片1.2做标注二,下载YoloV7三,划分数据集四,模型训练4.1创建yaml文件4.2修改默认参数4.3开始训练五,训练模型遇到的问题5.1UnicodeDecodeError:'gbk'codeccan'tdecodetype0xafinposition525:il... 查看详情

[课程][原创]yolov7训练自己数据集实例分割模型

课程地址:搞定系列:yolov7-segmention训练实例分割模型win版--深度学习视频教程-人工智能-CSDN程序员研修院课程介绍课程目录讨论留言你将收获学会如何搭建自己cuda环境学会如何搭建自己yolov7环境学会如何使用roboflow或者l... 查看详情

[软件工具]yolov3yolov4yolov7训练自己的数据集windows上超简单全程不写代码

darknet训练自己的数据集一般要写代码自己转化数据集,还要修改配置文件,数据集也不保证都是正确标注的,这个流程一般对初学者或者不熟悉的人经常弄错,只要一个配置参数改错就会导致训练出现问题,... 查看详情

visdrone数据集visdrone+yolov7结果记录(代码片段)

VisDrone+YOLOv7结果YOLOv7训练验证集测试集YOLOv7训练命令python-mtorch.distributed.launch--nproc_per_node4--master_port9527train.py--workers4--device0,1,2,3--sync-bn--batch-size32--datadata/VisDrone.yaml--img640640--cfgcfg/training/yolov7.yaml--nameyolov7_--hypdata/hyp.scratch.p5.yam... 查看详情

yolov7(目标检测)入门教程详解---检测,推理,训练(代码片段)

目录一.前言二.yolov7源码下载三.detect(检测)四.Train(训练)数据准备:labellmg:配置训练的相关文件 配置数据集文件正式训练:推理:推理效果:五.总结一.前言  上篇文章:YOLOv7(目标... 查看详情

yolo|用yolov7训练自己的数据集(超详细版)

...程)二.克隆项目配置库gitclonehttps://github.com/WongKinYiu/yolov7打开项目下载要求的库:cdyolov7 查看详情

睿智的目标检测——pytorch搭建yolov7-obb旋转目标检测平台(代码片段)

睿智的目标检测——Pytorch搭建[YoloV7-OBB]旋转目标检测平台学习前言源码下载YoloV7-OBB改进的部分(不完全)YoloV7-OBB实现思路一、整体结构解析二、网络结构解析1、主干网络Backbone介绍2、构建FPN特征金字塔进行加强特征提... 查看详情

darknet训练yolov7-tiny(alexeyab版本)(代码片段)

darknet框架训练yolov7Yolov7在darknet框架下的训练过程配置darknet环境官方数据集下载模型和配置文件训练之前必须看参数修改模型训练模型评估Yolov7在darknet框架下的训练过程配置darknet环境Darknet环境可以看我之前写的博文,这里... 查看详情

yolov7使用cadcd数据集实验记录(代码片段)

...后的数据集:值得注意的是博主开始时训练时使用了yolov7的初始化权重,由于cadc数据集中有几个分类在数据集中出现很少,以及博主在处理数据集时对一些目标种类进行了合并,因此学习效果并不好,因此博... 查看详情

mmdetection训练自己的coco数据集(代码片段)

训练前的准备训练数据的导入将自己的COCO数据集放入到下载的mmdetection的源码里,(如果没有data文件夹需要新建)data->COCO->你的COCO数据集文件格式如下:mmdetection├──mmdet├──tools├──configs├──data│... 查看详情

detr训练自己的数据集-实践笔记(代码片段)

DETR(DetectionwithTRansformers)训练自己的数据集-实践笔记&问题总结DETR(DetectionwithTRansformers)是基于transformer的端对端目标检测,无NMS后处理步骤,无anchor。实现使用NWPUVHR10数据集训练DETR.NWPU数据集总共包含十种类别目标,... 查看详情

基于yolov7的闸片厚度检测方法(代码片段)

基于yolov7的闸片厚度检测方法图像采集YOLOV7训练制作训练集训练样本生成权重文件修改YOLO预测类imag与cv2转换方法欧氏距离像素坐标与世界坐标映射测量方法边缘提取滤波方法检测结果图像采集利用基于3D结构光的深度相机采集... 查看详情