位置: IT常识 - 正文

YOLO_V8训练自己的数据集(yolov8训练自己的数据)

编辑:rootadmin
YOLO_V8训练自己的数据集

推荐整理分享YOLO_V8训练自己的数据集(yolov8训练自己的数据),希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:yolov8训练自己的数据,yolov8训练自己的模型,yolov8训练自己的模型,yolov8训练自己的数据,yolov8训练自己的数据集,yolov8训练自己的模型,yolov8训练自己的数据,yolov8训练自己的数据,内容如对您有帮助,希望把文章链接给更多的朋友!

YOLO_V8在2023年开年横空出世,在春节前还得卷一下。

由于YOLO_V8和YOLO_V5是同一个作者,所以很多操作都是一样的,下面主要描述一下如何用自己的数据集进行训练和测试(非命令行的方式)。

1、训练数据和模型的目录结构

这里以口罩数据集为例,该数据集分为两类,戴口罩和不戴口罩:['mask', 'no-mask'],由于本文是进行目标检测任务,直接将数据集放到“detect”目录下,实际使用时可将“MaskDataSet”文件夹放到工程中的任何位置。

MaskDataSet├─ test│ ├─ images│ │ └─ ······│ └─ labels│ └─ ······├─ train│ ├─ images│ │ └─ ······│ └─ labels│ └─ ······├─ valid│ ├─ images│ │ └─ ······│ └─ labels│ └─ ······├─ data.yaml└─ yolov8s.yaml

2、训练数据集的说明

train文件夹下主要包含训练图片images和标签labels,其中图片名和标签文件(.txt)名一一对应,且标签文件中保存的是对应图片中各个目标的类别和坐标(和YOLO_V5一样),例如:

类别+坐标0 0.4669 0.2392 0.1822 0.3123

test和valid存储的东西和train完全一致。

3、数据集参数文件data.yaml

data.yaml文件保存训练数据集的目录,类别数,类别名,如图所示:

我这里由于数据集的路径太深,直接使用了绝对路径,可根据需求自行修改。

YOLO_V8训练自己的数据集(yolov8训练自己的数据)

4、模型参数文件yolov8s.yaml

YOLO_V8包含5种模型,主要差异还是在:depth_multiple和width_multiple

depth_multiple: 0.33 # scales module repeatswidth_multiple: 0.50 # scales convolution channels

本文以YOLOv8s为例进行说明,将yolov8s.yaml中的nc改为2即可。

5、训练前的参数配置

之前的YOLO_V5实在train.py文件中配置模型路径,数据路径,epochs等参数的,但是YOLO_V8做了较大的改变,将所有的参数整合到一个文件中集中配置(.\YOLOv8\ultralytics\yolo\configs\default.yaml),并且划分了检测、分类、分割任务,便于后续集成多个项目,特别容易维护。

task: "detect" # choices=['detect', 'segment', 'classify', 'init'] # init is a special case. Specify task to run.mode: "predict" # choices=['train', 'val', 'predict', 'export'] # mode to run task in.

特别注意:mode应该加一个‘export’,在模型格式转换时就选择的是export。

6、开始训练

运行:YOLOv8\ultralytics\yolo\v8\detect\train.py,即可开始训练。

7、可能遇到的问题

1> OSError: [WinError 1455]页面文件太小,无法完成操作。

解决方法: 修改train.py中配置项workers的默认值,从初始值8修改为0(也可以试试1或2等较小的数)。

错误原因: workers=2, # 每个gpu分配的线程数,给每一个GPU喂数据的进程,GPU性能越强,取值越大,这样才能充分利用GPU的算力。如果用自己的电脑训练,这个值需要取小一些。

我在使用过程中直接将workers改为0

参考:https://blog.csdn.net/q839039228/article/details/124514664

本文链接地址:https://www.jiuchutong.com/zhishi/288761.html 转载请保留说明!

上一篇:【vue】vue中下载文件的方法(vue下载安装)

下一篇:塞里雅兰瀑布后的午夜太阳,冰岛南海岸附近 (© Tom Mackie/plainpicture)(塞里雅兰瀑布图片高清)

  • 所得税费用调增分录
  • 税金及附加是负债类吗
  • 建筑发票开具与土增税扣有什么关系?
  • 预缴增值税附加税费申报明细表
  • 公司购买汽车的折旧年限
  • 市场营销策划开题报告
  • 公司内控制度由谁制定
  • 企业临时外出经营如何使用发票
  • 免抵退税的计算数据
  • 股东溢价转让股份交易市场会计分录
  • 红字撤销需要带什么去税务局
  • 以旧换新的金子划算吗
  • 费用进项税额转出怎么做账务处理分录
  • 渔业专业合作社简介
  • 小规模纳税人纳税申报流程
  • 分公司是否可以贷款
  • 所得税网上申报表
  • 运费发票是含税还是不含税
  • 发票收款人和开票人
  • 其他应付款需要做预算会计吗
  • 公司经营情况说明怎么写模板
  • 资产负债表应交税费计算公式
  • 总分类账的账簿启用表怎么填
  • 捐赠计提管理费的账务处理?
  • mac应用程序安装权限在哪找
  • 如何解决win7系统搜不到蓝牙耳机
  • typecho插件开发教程
  • 电脑CPU性能天梯
  • wordpress怎么上传网站
  • qttask.exe是什么进程?qttask.exe是不是病毒?
  • mmc.exe是什么进程
  • php实现搜索的方式
  • 废品损失科目的期末余额在借方表示什么?
  • PHP:imagepalettetotruecolor()的用法_GD库图像处理函数
  • 大群的芒基蝠鲼跃出水面,墨西哥加利福尼亚湾 (© Mark Carwardine/Minden Pictures)
  • hypergraph learning
  • laravel orm使用
  • 花卉苗木栽培技术
  • 公司账户转法人账户
  • 前端schema
  • php简单检测网址是什么
  • cpu power management设置
  • 购买方红字信息表已开销售方系统查不到
  • dubbo 实现原理
  • 走工资支付福利怎么算
  • 如何用python给excel每一行添加序号
  • java删除类
  • 收到的销项负数发票如何申报
  • 其他应收应付如何平账
  • 累计扣除包括哪些
  • 向投资者支付的股息
  • 购买材料发生商损怎么算
  • 自来水差额征税申报表填写
  • 其他应付款是什么科目代码
  • 机票的电子发票可以报销吗
  • 出租土地使用权计入什么科目
  • 税金及附加如何记账
  • xp附件游戏
  • MacBook怎么恢复出厂设置
  • 超级本可以玩游戏吗
  • win8.1系统安装
  • win8系统磁盘清理在哪里
  • 笔记本win8.1怎么重装系统
  • win7内存检测工具
  • windows 查看补丁
  • mac安装dw
  • linux下history命令显示历史指令记录的使用方法
  • hyper虚拟机连接外网
  • 微软商城会员
  • win10系统如何关闭
  • 升级win10系统错误代码0x80072F8F
  • 电脑window8系统怎么样
  • js 正则验证
  • cocos2d-js游戏开发
  • 利用Matplotlib对一组数据进行分析
  • qq聊天背景闪烁
  • python标准模板库
  • 雷爵电动车官网
  • django批量上传图片
  • javascript的主要功能
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设