yolov8训练自己的数据集

前文中完成了简单的配置,现在学习一下怎么使用模型进行训练。

训练:yolo train data=... model=yolov8n.pt epochs=... lr0=0.03

试一下怎么用:

执行过程长这样:

执行后在对应文件夹看结果,啥都有。包括训练/验证时的图像,各种评估方法等,并且会保存最优权重。


使用coco训练集可以完成模型的训练,那么要训练自己的数据集,要怎么做呢?--那当然是模仿!

准备数据集时,要模仿coco数据集的存储格式并且进行配置。

coco8

  • images
    • train
    • val
  • labels
    • train
    • val

images里面存放图片,labels里面存放信息---打开一个.txt文件看看长啥样

里面存放了五个参数,第一个是类别,第二、三个参数是归一化后的中心坐标,第四、五个参数是归一化后的长&宽

  • 什么是归一化?为什么要归一化?
    • 就是读入图片时,画出的boundingbox的各参数与原图片的宽和高之比
    • 归一化是为了数据便于计算,统一存储格式,以及避免因原图片尺寸不同而导致计算量等有所影响。

接下来简单看一下.yaml中数据的存储格式

path:存储绝对路径

train/val/test:都是存储在path下的路径。就是path指定了一个大文件夹,而这三个是大文件夹下的子文件夹。


知道大概流程就开始打标了- -,我打算在多分辨率下进行标注,是个不小的工程量。用labelimg标注时一直按w绘制方框就完了。

(下图中只是部分框选,是未完成打标的图像,因为也可以看到漏了好几个的)

打标打了160张图,五六千个框框,人麻了- -打标打了两天


完成标注后,下一步是对yolo网络的调用。前面已经提到文件夹的排列方式。

训练集和验证集相应存放图片/标签

完成文件存放后,配置.yaml文件


下一步,就是切换路径,调用模型训练自己的数据集了。

激活环境---切换路径(主要是懒得打绝对路径才这样的)---调用模型,设置参数

这里手快打漏了cd 顾着截图去了- 在这里,设置了迭代次数为10,图像大小为640* 640,其他参数比如初始学习率等也可以设置,详见其他博客。

yolo train model=yolov8n.pt data=mydata.yaml epochs=10 imgsz=640

首先会显示模型的架构以及参数量,

跑起来长这样,然后等训练完成就好了。后续工作就是进行参数的调整或者针对具体任务改变网络架构(修改.yaml文件,改model,再设置pretrained=xxx.pt)等。模型训练完成后,在相应的runs文件夹查看混淆矩阵以及各曲线,且会自动存储最优权重。


小结:其实yolo网络上手是很简单的,零基础的都能轻松使用,也不需要什么编程基础...会打标,会放文件,会配置环境就好了。之后的项目中也要经常用到yolo的,这次算是开了个头吧!

相关推荐
枫叶林FYL5 分钟前
【群体智能集群控制工程实践】第3章 一致性协同算法
算法
禹凕23 分钟前
滑动窗口算法实战指南
python·算法
DisonTangor30 分钟前
【腾讯混元雪耻归来】 Hy4 preview:770B 参数 MoE 旗舰模型,1M 上下文全面开源
人工智能·算法·开源·aigc·腾讯云·腾讯云ai代码助手
明月_清风2 小时前
从二叉树到 B+ 树:一文搞懂工程中「树」的演化之道
数据结构·算法·go
渡我白衣2 小时前
并查集:基础认识与模拟实现
android·java·javascript·数据结构·c++·算法·并查集
hetao17338372 小时前
2026-09-01~09-04 hetao1733837 的刷题记录
c++·算法
Evand J2 小时前
【MATLAB例程,图像滤波5】 反谐波均值滑动窗口滤波(CHMF)图像降噪与质量评价,附代码下载链接
图像处理·算法·计算机视觉·matlab·均值算法·滑动窗口滤波·均值滑动
血小板要健康3 小时前
链表 阶段算法总结
java·数据结构·笔记·算法·leetcode·链表
a187927218313 小时前
【算法】回溯算法(三):三记重锤与 N 皇后——记忆化、状态设计与三层漏斗
算法·leetcode·go·剪枝·回溯·n皇后·算法讲解
HugoStudio_SWAN5 小时前
洛谷 P1420 / P1179 / B4262 最长连号、数字统计与词频统计——统计的三种面孔
c++·学习·程序人生·算法