Yolov11的空标注负样本技术在模型训练中的应用

空标注负样本技术就是对图像进行先标注一个类型,保存成xml文件后,然后再对标注的类型进行删除,形成一个没有目标的xml的文件,告诉系统这张图像就是一个纯背景,这对消除检测误报具有重大意义。对于模型的训练中负样本包含没有目标的图像和没有目标的的XML文件。

1、具体的处理方式包括如下内容:

1)图像处理 :负样本在训练时仍然可以使用。这些图像可以帮助模型学习不产生误检,即在没有目标时应该输出什么(通常是无检测结果)。

2)标注文件处理 :对于负样本中不包含任何目标对象(即没有object节点)。在训练时,解析标注文件时如果发现没有目标,则会生成一个空的标签数据。

3)YOLOv11训练过程:在训练时,每个批次中可能包含正样本(有目标)和负样本(无目标)。对于负样本,损失函数计算时,定位损失和分类损失都不考虑(因为没有真实框),但是会计算置信度损失,因为模型应该预测这些区域没有目标(即置信度接近0)。

4)数据配置 :在YOLOv11的数据配置文件中(如data.yaml),你需要指定训练和验证图像所在的目录以及标注文件所在的目录。YOLOv11会读取指定目录下的所有图像和对应的标注文件(标注文件与图像同名,但扩展名为.txt.xml,具体取决于标注格式)。

5)标注格式转换 :YOLOv11通常使用YOLO格式的标签(每个图像对应一个.txt文件,每行一个目标,格式为:class_id x_center y_center width_height,坐标是归一化的)。如果你使用的是XML格式(如Pascal VOC格式),你需要先将XML文件转换为YOLO格式。在转换过程中,如果XML中没有目标,那么生成的.txt文件将是空的(0字节)。

6)负样本的处理 :在转换后的YOLO格式中,负样本对应的.txt文件是空的。在训练时,YOLOv8会读取这些空的标签文件,并将其视为没有真实框。

7)数据加载:在数据加载时,YOLOv11会检查标签文件是否为空。如果为空,则不会分配任何真实框给该图像。在训练过程中,这样的图像将只用于降低置信度(即让模型学会在没有目标时不要预测出目标)。

2、注意事项

虽然负样本有助于减少误检,但过多的负样本可能会导致模型偏向于预测为背景,从而漏检真正的目标。因此,负样本的数量需要适当控制,通常建议正负样本比例要均衡。

具体的注意事项包括如下内容

  • 在数据集中包含一定比例的负样本(保持正负样本平衡),建议:5%-20%。对于小目标检测,可以减少负样本比例.

  • 验证集必须包含负样本以评估实际性能

  • 使用数据增强确保模型不过拟合特定背景

  • 监控误报率作为重要评估指标

  • 使用数据增强技术,如 mosaic 增强,这可能会将负样本与其他图像混合,从而生成更复杂的训练样本。

3.验证方法

包含正负样本的训练集和验证集训练得到模型pt,并分别对部分正负样本测试,可对正负样本的效果进行验证。

相关推荐
神奇霸王龙4 小时前
GB/T 46886 闭环屠夫:5 旗舰多模态 LLM 工业质检实测
人工智能·计算机视觉·ai·开源·ai编程·本地部署
南讯股份Nascent5 小时前
洽洽全域会员项目启动会圆满召开
大数据·人工智能
大郭鹏宇5 小时前
基于 LangGraph 构建智能分诊系统(一):项目概述与环境搭建
大数据·人工智能·microsoft·langchain
小林ixn5 小时前
大模型的“高考成绩单”:读懂Benchmark,选对真·生产力模型
人工智能·llm·测试
冬奇Lab5 小时前
AI 评测系列(04):RAG 评测——RAGAS 四指标实战与一个反直觉发现
人工智能·llm·agent
三声三视5 小时前
uni-app 鸿蒙端传参变成 [object Object]?顺着源码追到 ArkTS router 底层才搞明白
人工智能·ai·uni-app·aigc·ai编程·harmonyos
fthux5 小时前
“装闭”,让装修套路“装”不下去
人工智能·ai·开源·github·open source
andxe6 小时前
安科士 AndXe 技术博客:400G QSFP112 SR4 光模块|AI 算力与超算短距互联最优方案
网络·人工智能·光模块·光通信
计算机魔术师7 小时前
Karpathy:用语音与LLM长谈可提升理解效率
人工智能·ai编程
甲维斯7 小时前
我要开始吹牛逼了!Kimi K3 “宇宙无敌”!
前端·人工智能