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

带标注的骑电动车是否佩戴头盔数据集,识别率77.1%,1345张图,支持yolo,coco json,voc xml,文末有模型训练代码


带标注的骑电动车是否佩戴头盔数据集,识别率77.1%,1345张图,支持yolo,coco json,voc xml,文末有模型训练代码

模型训练指标参数:

模型训练图:

数据集拆分

总图数:1345 张图数
训练集

1077 张图

验证集

135 张图

测试集

133 张图

预处理

自动定向:应用

调整大小:拉伸到640x640

增强

数据集标签:

[‘Without Helmet’, ‘elecar’, ‘With Helmet’]

标签解释
Without Helmet:未佩戴头盔
elecar:电动车(电动两轮车、电摩简写)
With Helmet:佩戴头盔

数据集图片和标注信息示例:

数据集下载:

yolo26,yolo v12,yolo v11,yolo v9,yolo v8,yolo v7,coco json,pascal voc xml:
https://download.csdn.net/download/pbymw8iwm/93058799

YOLO模型训练

下载数据集之后解压到当前文件夹,然后将 我的仓库 https://gitcode.com/pbymw8iwm/YOLOProject 里的训练模型脚本复制到文件夹下,假设你使用的是yolov8来训练你就用python train_yolov8.py
​​

模糊图片在模型训练中的优势分析

该数据集中包含了如果包含了一些模糊图片,这并非数据缺陷,而是一种精心设计的数据增强策略,能为模型训练带来以下显著优势:

  1. 提升模型鲁棒性:现实场景中,可能发生在运动模糊,对焦不准或光线不足的情况下。在训练集中引入模糊样本,可以迫使模型学习更本质的特征,而不是过度依赖清晰的边缘或纹理细节,从而提升模型在复杂、非理想成像条件下的识别能力。
  2. 模拟真实世界噪声:监控摄像头、手机等设备在低光照或快速移动时极易产生模糊。包含此类图片的数据集能让模型提前“见识”并适应这种噪声,减少在实际部署时因图像质量下降而导致的性能骤降。
  3. 防止模型过拟合:如果训练集全是高清、摆拍的完美图片,模型容易记住这些特定场景下的“干净”特征,而对新的、稍有模糊的图片泛化能力差。模糊图片作为一种有效的正则化手段,可以增加数据分布的多样性,防止模型过拟合到有限的清晰样本上。
    总结:因此,数据集中包含的模糊图片,与你看到的“水平翻转”、“剪切形变”、“随机遮挡”等增强操作一样,都是为了构建一个更接近真实世界复杂分布的数据环境,从而训练出更健壮、更泛化、更实用的模型。

模型验证测试情况:

验证测试代码:

#需要安装pip install ultralyticsfromultralyticsimportYOLOimportcv2# ===================== 1. 加载YOLO模型 =====================# 自动下载预训练模型(yolov8n最轻量快速),也可换 yolov8s/m/l/xmodel=YOLO("best.pt")# ===================== 2. 推理配置 =====================image_path="326_jpg.rf.8ad64cc0668df32c4e5f59b50e899e9c.jpg"# 你的图片路径save_result=True# 是否保存标注后的图# ===================== 3. 执行推理 =====================results=model.predict(source=image_path,conf=0.01,# 置信度阈值(低于该值忽略)save=False,# 关闭默认保存,自定义保存verbose=False# 关闭冗余日志)# ===================== 4. 解析结果:目标区域 + 标注信息 =====================print("="*50)print("YOLO 推理结果(目标区域 + 标注信息)")print("="*50)# 获取图片(用于绘制框)img=cv2.imread(image_path)# 遍历所有检测目标foridx,resultinenumerate(results):boxes=result.boxes# 所有检测框forboxinboxes:# ========== 提取目标区域(坐标) ==========# xyxy: 左上角x, 左上角y, 右下角x, 右下角yx1,y1,x2,y2=box.xyxy[0].cpu().numpy()# 宽高w=x2-x1 h=y2-y1# ========== 提取标注信息 ==========cls_id=int(box.cls[0])# 类别IDcls_name=model.names[cls_id]# 类别名称conf=float(box.conf[0])# 置信度# ========== 打印信息 ==========print(f"目标{idx+1}:")print(f" 标注类别:{cls_name}")print(f" 置信度:{conf:.2f}")print(f" 目标区域坐标:")print(f" 左上角 ({x1:.1f},{y1:.1f})")print(f" 右下角 ({x2:.1f},{y2:.1f})")print(f" 宽×高:{w:.1f}×{h:.1f}")print("-"*30)# ========== 在图片上绘制检测框 ==========cv2.rectangle(img,(int(x1),int(y1)),(int(x2),int(y2)),(0,255,0),2)cv2.putText(img,f"{cls_name}{conf:.2f}",(int(x1),int(y1)-10),cv2.FONT_HERSHEY_SIMPLEX,0.5,(0,255,0),2)# ===================== 5. 保存/显示结果 =====================ifsave_result:cv2.imwrite("yolo_result.jpg",img)print("✅ 标注图片已保存为:yolo_result.jpg")# 显示图片(可选)cv2.imshow("YOLO Result",img)#cv2.waitKey(0)#cv2.destroyAllWindows()
http://www.jsqmd.com/news/1119189/

相关文章:

  • 如何通过HsMod插件实现炉石传说游戏体验的全面优化
  • 国际化办公首选!全域多语言切换会议录音APP
  • PHP安全编码实战:从SQL注入到XSS攻击的全面防护指南
  • 基于Hermes Agent与Harness Engineering构建生产级AI智能体实战指南
  • 打通智能体的“知识供应链”:OKF 重构 Agent 时代的知识基建
  • 工业视觉质检延迟,核心瓶颈该如何定位?
  • Windows 端 OpenClaw 完整安装流程|全程可视化操作 + 安装包获取
  • GPT-4o真实效能评估:何时该用,何时该弃
  • 锐捷RG-N18000-X 交换机一对多端口镜像(RSPAN)保姆级实战指南
  • 记住窗口位置大小一键恢复免费工具
  • SAM 3 视频分割实战教程:用文本提示分割并跟踪视频中的目标
  • 全驱数字人API实战教程:一张图片即可生成AI数字人(附完整API文档)
  • CAD画图时如何快速地进行图层的设置?-CAD画图基础
  • Triton 编译器在 ROCm 下的应用,自定义 Kernel 开发的桥梁
  • 如何科学评估大语言模型性能:避开虚假版本与误导性跑分
  • ComfyUI v0.27.0更新:Int8模型正式落地,卷积模型加速、Turing显卡支持、视频与多分辨率能力全面增强
  • 【Java毕业设计】中小型汽配企业销售台账管理系统的设计与实现 基于 SpringBoot 的汽车配件供应商与采购销售系统(源码+文档+远程调试,全bao定制等)
  • CTF 基础密码学:模素数二次剩余解题 Writeup
  • 融数筑基联产链·同源贯通兴煤化——孪生空间数据融通 打通煤化工矿生产管理数据链路技术白皮书
  • 让用户选择而不是重新填写
  • 中欧班列物流系统的多线路管理架构
  • 3个核心功能解决你的Windows日志分析困境:为什么LogExpert能成为开发运维的终极利器?
  • STM32学习笔记【30.SPI总线】
  • Excel 的质量管控文档设计
  • zkGolf 竞赛:构建成本最低零知识电路,电路越紧凑得分越优!
  • 如何一键实现8个平台同步直播:OBS多RTMP插件完全指南
  • Python1
  • CSRF攻击原理与防御策略全解析:从Samesite Cookie到Token验证实战
  • AllenAI:终端智能体强化学习训练配方
  • 怎么量化一个 AI Agent 的好坏?面试官问「Agent 评测」时真正想听什么