保姆级官方yolov7训练自己的数据集及项目部署详解
作者：xiao9wo8tian 发布时间：2023-09-16 14:19:45
前言
首先，先说明我只是初步接触yolov7,写这篇文章的主要目的是可以让大家快速应用自己的数据集进行训练。没有接触过yolov5也没有关系，该篇文章会逐步进行演示如何训练。
第一步 数据集准备
首先确保你有labelimg标图软件，若无，需要自行去下一个并看一下标图教程。
当你已经标注完成，获得了img以及相对应的xml之后（如图）
接下来就是可选择项：是否需要图像增强来获取更多样本，如需要点击这里下载 ，直接用enhance_img.py去增强。之后我的习惯是每有一个项目需要训练，则会新建一个文件夹，项目存放的文件如图：
下面ImageSets\Main用于存放后续脚本文件划分训练集测试集的相对应的train.txt,test.txt。hyp.scratch.p5.yaml为yolov7的超参设置，可以直接从yolov7\data下面去拷过来放到你项目里。icon.yaml为你要所要训练的类别和相应的类，同时也会写上实际训练时训练数据和测试数据。该文件如下图。
Main.py就是划分你的数据为训练集和测试集.txt的脚本。整体新建的项目目录就是这样。下面说一下操作流程：1 首先确保是该项目目录方式 2 运行main.py脚本文件得到了ImageSets\Main下面的train.txt,test.txt。3 在yolov7的根目录下运行xml2txt脚本文件。main.py xml2txt.py文件点击这里下载 注意要将该文件的类和项目名改成自己的。如图：
此时，我们的数据准备阶段已经完成，项目目录如图：
第二步 train.py载入自己的数据集并训练
红色箭头和红框就是常见改动的地方，改动后就可以训练了。
第三步 将训练好的pt文件做成接口调用
在等待训练完成之后，就会在runs/train下面获得训练的best.pt,你可以拿着这个pt去做接口使用了。首先，在自己的项目里使用必须要确保yolov7根目录下的models和utils文件夹放到了你的项目根目录。然后下载model_import.py 点击这里下载 嵌入你的任何项目路径下调用predict函数就可以输出检出结果了。
来源：https://blog.csdn.net/weixin_39849397/article/details/125688383
0
投稿猜你喜欢
乱码问题破解压缩包时候会存在中文乱码问题！1：直接使用Everything搜索出要修改的库文件 zipfile.py ，并用notepad+- 本文实例讲述了微信小程序实现图片上传、删除和预览功能的方法。分享给大家供大家参考，具体如下：这里主要介绍一下微信小程序的图片上传图片删除和图
- 作为一个合格的开发者，在本地环境下完成开发还远远不够，我们需要把Web App部署到远程服务器上，这样，广大用户才能访问到网站。很多做开发的
- 一、下载软件1. 进入MySQL官网，登陆自己的Oracle账号(没有账号的自己注册一个),下载Mysql-5.7.17，下载地址：http
- 设置表名为中文1.设置Models.py文件class Post(models.Model): name = models.CharFiel
- 启动服务发生1067错误：1.删除datadir 下的用户数据库文件 和 日志文件 （ib_logfile0，ib_logfile1）。2.
- 1.锦短情长为什么选择这个标题，借鉴了一封情书里面的情长纸短，还吻你万千。锦短情长都只谓人走茶凉，怎感觉锦短情长？一提起眼泪汪汪，是明月人心
- 1.数据源2.数据整体排名1）普通排名从1开始，按照顺序一次往下排(相同的值也是不同的排名)。set @rank =0;select cit
- 一、CSS HACK以下两种方法几乎能解决现今所有HACK.1, !important随着IE7对!important的支持, !impor
- 通过购物车的一个案列，把vuex学习了一篇。vuex概念浅谈Vuex 是一个专为 Vue.js 应用程序开发的状态管理模式。它采用集中式存储
- 栗子：计算斐波那契数列（任一个数都是前两个数之和的数字序列）Python2.7实现代码如下：<strong><span s
- BootstrapValidator 是一款专门针对Boostrap v3的表单检验jQuery插件，能够实现众多常用的检验功能，并且易于扩
- 在django中,ORM(对象关系映射器—object-relational mapper)任务是：模型化数据库，创建数
- 前言硬要说这篇文章怎么来的，那得先从那几个吃野味的人开始说起…… 前天睡醒：假期还有几天；昨天睡醒：假期还有十几天；今天睡醒：假期还有一个月
- 操作Excel通常是用如下三个扩展体：import xlrdimport xlwtimport openpyxlwb1 = openpyxl
- view.pyfrom django.views.generic import ListView,DetailViewfrom xxxx.m
- 一、写在前面我从未想过自己会写python系列的自动化文章，有些同学会问，那你现在为什么又开始写了？不止一个人找过我，问我可以写一些Pyth
- title: Python 装饰器装饰类中的方法comments: truedate: 2017-04-17 20:44:31tags: [
- tkinter库：Python的标准Tk GUI工具包的接口示例：from tkinter import *root = Tk()#你的ui
- 自己的小Python项目好几天没有写了，今天打开PyCharm准备继续写，突然发现之前的激活码被取消不能用了，本来激情满满的准备干活啦！之前
