当前位置: 首页 > article >正文

【Datawhale AI夏令营】从零上手CV竞赛Task3

文章目录

  • 前言
  • 一、数据集增强
  • 二、设置 YOLO 模型训练参数
  • 三、模型微调
  • 总结


前言

本文的Task3对Task1的baseline代码继续进行优化的过程。

在这里插入图片描述


一、数据集增强

数据增强是机器学习和深度学习中常用的技术,用于通过从现有数据集中生成新的训练样本来提高模型的泛化能力。

常见的增强技术包括翻转、旋转、缩放和颜色调整。例如 Albumentations、Imgaug 和 TensorFlow的 ImageDataGenerator等多个库可以生成这些增强。

数据增强方法描述
Mosaic Augmentation将四张训练图像组合成一张,增加物体尺度和位置的多样性。
Copy-Paste Augmentation复制一个图像的随机区域并粘贴到另一个图像上,生成新的训练样本。
Random Affine Transformations包括图像的随机旋转、缩放、平移和剪切,增加对几何变换的鲁棒性。
MixUp Augmentation通过线性组合两张图像及其标签创造合成图像,增加特征空间的泛化。
Albumentations一个支持多种增强技术的图像增强库,提供灵活的增强管道定义。
HSV Augmentation对图像的色相、饱和度和亮度进行随机变化,改变颜色属性。
Random Horizontal Flip沿水平轴随机翻转图像,增加对镜像变化的不变性。

如下图所示就是数据增强的例子:

在这里插入图片描述

参数名类型默认值取值范围描述
hsv_hfloat0.0150.0 - 1.0调整图像色调,引入颜色变异性,提高不同光照下的泛化能力。
hsv_sfloat0.70.0 - 1.0调整图像饱和度,改变颜色强度,模拟不同环境条件。
hsv_vfloat0.40.0 - 1.0调整图像亮度,帮助模型在不同光照下表现良好。
degreesfloat0-180 - +180随机旋转图像,提高识别不同方向物体的能力。
translatefloat0.10.0 - 1.0平移图像,帮助模型学习检测部分可见物体。
scalefloat0.5>=0.0缩放图像,模拟物体与相机之间的不同距离。
shearfloat0-180 - +180剪切图像,模拟从不同角度观察物体的效果。
perspectivefloat00.0 - 0.001应用随机透视变换,增强模型对3D空间物体的理解能力。
flipudfloat00.0 - 1.0上下翻转图像,增加数据变异性,不影响物体特征。
fliplrfloat0.50.0 - 1.0左右翻转图像,有助于学习对称物体和增加数据集多样性。
bgrfloat00.0 - 1.0翻转图像通道从RGB到BGR,提高对通道顺序错误的鲁棒性。
mosaicfloat10.0 - 1.0合成四张图像,模拟不同场景组合和物体交互,增强复杂场景理解。
mixupfloat00.0 - 1.0混合两张图像及标签,创建合成图像,增强泛化能力。
copy_pastefloat00.0 - 1.0复制物体并粘贴到另一图像,增加实例和学习遮挡。
auto_augmentstrrandaugment-自动应用预定义增强策略,优化分类任务。
erasingfloat0.40.0 - 0.9随机擦除图像部分,鼓励模型关注不明显特征。

二、设置 YOLO 模型训练参数

YOLO 模型的训练设置包括多种超参数和配置,这些设置会影响模型的性能、速度和准确性。

微调涉及采用预先训练的模型并调整其参数以提高特定任务或数据集的性能。该过程也称为模型再训练,使模型能够更好地理解和预测在实际应用中将遇到的特定数据的结果。

通常,在初始训练时期,学习率从低开始,逐渐增加以稳定训练过程。但是,由于预训练模型已经从以前的数据集中学习了一些特征,因此立即从更高的学习率开始可能更有益。在 YOLO 中绝大部分参数都可以使用默认值。

参数描述
imgsz训练时的目标图像尺寸,所有图像在此尺寸下缩放。
save_period保存模型检查点的频率(周期数),-1 表示禁用。
device用于训练的计算设备,可以是单个或多个 GPU,CPU 或苹果硅的 MPS。
optimizer训练中使用的优化器,如 SGD、Adam 等,或 ‘auto’ 以根据模型配置自动选择。
momentumSGD 的动量因子或 Adam 优化器的 beta1。
weight_decayL2 正则化项。
warmup_epochs学习率预热的周期数。
warmup_momentum预热阶段的初始动量。
warmup_bias_lr预热阶段偏置参数的学习率。
box边界框损失在损失函数中的权重。
cls分类损失在总损失函数中的权重。
dfl分布焦点损失的权重。

三、模型微调

在Task2的基础上,我仍然使用8张4090显卡进行yolov8x更多epoch的训练,同时修改了一些超参数。

在这里插入图片描述

results = model.train(data="/mnt/sdb1/yolo-dataset/yolo.yaml", epochs=10, imgsz=1280, batch=32, device="0,1,2,3,4,5,6,7", mixup=0.4, copy_paste=0.5, mosaic=0)


results = model(path, conf=0.25, imgsz=1280, verbose=False, augment=True)

但是令人遗憾的是,可能mosaic不能设置成0,因为可能需要增加物体位置和尺度的多样性。导致最后的提升比较微小,从0.132提升到了0.137。

在这里插入图片描述

类别目标数量检测数量精度(Precision)召回率(Recall)F1 分数mAP@0.5
all2595196580.7070.2370.330.233
非机动车违停2591126660.8290.6350.7440.447
机动车违停259157370.9990.3140.5640.484
垃圾桶满溢7917911.00.00.00.0
违法经营2324640.00.00.01340.00136

在这里插入图片描述
在这里插入图片描述

而且从训练来看,训练10个epoch过程中差别不是很大,说明可能存在过拟合的问题。

在这里插入图片描述

最后我查看群友们的建议,发现比赛的数据存在严重的数据集不均衡的问题,所以我重新使用一张4090,换用了另一种训练集:验证集为4:1比例的数据划分。

在这里插入图片描述

类别目标数量检测数量精度(Precision)召回率(Recall)F1 分数mAP@0.5
all9307355610.6340.6440.6510.485
非机动车违停9306241280.9040.9310.9570.681
机动车违停9211112221.01.00.9950.774
垃圾桶满溢2112110.00.00.00.0

在这里插入图片描述

在这里插入图片描述

此时的成绩提升到了0.175,提升比较大,说明数据集的标签均衡是非常重要的影响因素。


总结

本次城市治理的违规行为识别的比赛到此就完成所有步骤了,以后有机会还会在此基础上继续深入研究,挑战更多可能!


http://www.kler.cn/a/285065.html

相关文章:

  • 43.第二阶段x86游戏实战2-提取游戏里面的lua
  • 关于sass在Vue3中编写bem框架报错以及警告问题记录
  • 期权懂|期权新手入门教学:期权合约有哪些要素?
  • 综合案例铁锅炖(CSS项目大杂烩)
  • sqlserver删除最近2个月的记录
  • Kubebot:一款Google云平台下的Slackbot安全测试工具
  • 惠中科技PV-Wiper全自动光伏清洁系统,根治污染难题
  • 2024最详细Maven配置教程
  • Java算法之归并排序(Merge Sort)
  • 【Godot4.3】MarkDown解析和生成类 - MDdoc
  • 仿华为车机功能之--修改Launcher3,实现横向滑动桌面空白处切换壁纸
  • 在Ubuntu 20.04上安装MySQL的方法
  • 神经网络搭建实战与Sequential的使用
  • 南京观海微电子----VCC、 VDD、VSS、VEE 电压符号解释
  • <Rust>egui学习之小部件(八):如何在窗口中添加滑动条slider部件?
  • Vue.js入门系列(十九):深入理解和应用组件自定义事件
  • C++宏展开
  • 2024.08.28 C++初学
  • Notepad++回车不自动补全
  • Python算法工程师面试整理-概率与统计
  • 数学基础 -- 线性代数之矩阵因式分解
  • 计算多图的等价无向图的邻接链表表示
  • MySQL中日期和时间戳的转换:字符到DATE和TIMESTAMP的相互转换
  • OpenHarmony 实战开发——一文总结ACE代码框架
  • 在多云生态下,如何实现跨云的自动化身份管理?
  • 【React】从零开始搭建 react 项目(初始化+路由)