SafetyHelmetWearing-Dataset
收藏github2019-12-23 更新2024-05-31 收录
下载链接:
https://github.com/CUIcaicai/Safety-Helmet-Wearing-Dataset
下载链接
链接失效反馈官方服务:
资源简介:
安全帽佩戴检测数据集,包含7581张图像,其中9044个安全帽佩戴对象(正例)和111514个普通头部对象(未佩戴或负例)。数据集中的正例对象来自谷歌或百度,并使用LabelImg手动标注。部分负例对象来自SCUT-HEAD数据集,经过修复以符合Pascal VOC格式。
The safety helmet wearing detection dataset comprises 7,581 images, including 9,044 instances of helmet-wearing objects (positive examples) and 111,514 instances of ordinary head objects (non-wearing or negative examples). The positive examples in the dataset are sourced from Google or Baidu and are manually annotated using LabelImg. Some of the negative examples are derived from the SCUT-HEAD dataset, which has been modified to conform to the Pascal VOC format.
创建时间:
2019-12-23
原始信息汇总
安全帽佩戴检测数据集概述
数据集介绍
- 名称: SafetyHelmetWearing-Dataset(安全帽佩戴检测数据集)
- 用途: 用于安全帽佩戴和人类头部检测
- 数据规模: 包含7581张图像,其中9044个安全帽佩戴对象(正例)和111514个未佩戴安全帽的头部对象(负例)
- 数据来源: 正例对象来自Google或Baidu,负例对象部分来自SCUT-HEAD数据集,并进行了手动标注和格式修正
- 标注工具: 使用LabelImg进行手动标注
- 格式: 数据集遵循Pascal VOC格式
数据集与模型下载
- 数据集下载链接:
- 模型下载链接:
数据集使用
- 数据集结构: 遵循Pascal VOC格式,包含Annotations、ImageSets和JPEGImages三个主要目录
- 对象类别: 数据集定义了两个对象类别,"hat"(正例,佩戴安全帽)和"person"(负例,未佩戴安全帽)
模型使用
- 依赖: 需要安装MXNet, GluonCV, OpenCV
- 测试: 提供两种测试方式,可通过命令行参数调整测试设置
- 训练: 可通过设置数据集路径和调整训练参数进行模型训练
模型性能
- 基准测试:
- 模型 | darknet | mobile1.0 | mobile0.25
- map | 88.5 | 86.3 | 75.0
注意事项
- 训练时注意防止梯度爆炸,可通过增加预热周期或降低学习率来解决
- 数据集加载可能较慢,可通过增加CPU核心数来提升训练速度
- 在Windows系统上训练时可能遇到程序阻塞问题,需注意系统资源分配
搜集汇总
数据集介绍

构建方式
SafetyHelmetWearing-Dataset是一组专注于安全帽佩戴检测的图像数据集。该数据集的构建采取了从Google和百度搜索引擎中获取正样本(即佩戴安全帽的对象),并使用LabelImg工具进行手动标注。负样本(即未佩戴安全帽的对象)部分来源于SCUT-HEAD数据集,并对原始SCUT-HEAD数据集中的部分错误进行了修正,使其能够以标准的Pascal VOC格式直接加载。同时,该数据集还提供了基于MXNet GluonCV的预训练模型。
特点
该数据集的主要特点在于其专注于安全帽佩戴检测任务,包含7581张图像,其中有9044个佩戴安全帽的正面样本和111514个正常头部(未佩戴安全帽)的负面样本。数据集以Pascal VOC格式组织,包含两个对象类别名称,'hat'代表正面样本,'person'代表负面样本。此外,数据集还提供了不同大小的yolo模型,以适应不同的检测需求。
使用方法
使用该数据集时,用户需要确保已安装MXNet、GluonCV和OpenCV依赖。数据集的使用包括两种方式:一种是直接下载预训练模型进行测试,另一种是通过mxnet symbol进行推断。训练时,用户可以通过调整参数来设置数据集路径,并根据需要设置训练和验证数据集。训练过程中,建议注意梯度爆炸问题,并适当调整训练参数以优化模型性能。
背景与挑战
背景概述
安全帽佩戴检测数据集(SafetyHelmetWearing-Dataset,简称SHWD)是一项专注于安全头盔佩戴识别的研究成果,由相关研究人员于近年开发。该数据集由9044个佩戴安全帽的对象(阳性样本)和111514个未佩戴安全帽的普通头部对象(阴性样本)组成,旨在为安全帽佩戴识别以及人头检测提供高质量的数据支持。SHWD数据集的创建,不仅丰富了计算机视觉领域的数据资源,也为建筑、制造等行业的安全监管提供了技术支持,其主要研究人员来自我国相关科研机构。
当前挑战
在研究过程中,SHWD数据集的构建面临着多项挑战。首先,如何确保数据集的多样性和均衡性,以覆盖不同场景、不同光照条件下安全帽的佩戴情况,是一大难题。其次,数据标注的准确性直接影响到模型训练的效果,而大规模数据的手动标注工作既耗时又易出错。此外,在数据集构建过程中,研究人员还需克服数据格式转换、数据增强等技术难题。在解决领域问题方面,安全帽佩戴检测面临着如何在复杂环境下准确识别佩戴状态、降低误报率等挑战。
常用场景
经典使用场景
在计算机视觉领域,安全帽佩戴检测是工业安全监控的重要组成部分。SafetyHelmetWearing-Dataset作为一个专门针对安全帽佩戴检测的数据集,其经典使用场景在于训练深度学习模型以识别和区分工人是否佩戴安全帽。该数据集提供了丰富的标注图像,包含佩戴安全帽的正面样本和不佩戴安全帽的负面样本,为模型训练和评估提供了坚实基础。
实际应用
在实际应用中,SafetyHelmetWearing-Dataset已被用于构建工业现场的安全监控系统中。通过该数据集训练出的模型能够在 Construction、Mining 等高风险工作环境中实时监测工人安全帽的佩戴情况,对未佩戴安全帽的工人进行实时警告,有效预防安全事故的发生。
衍生相关工作
基于SafetyHelmetWearing-Dataset,学术界和工业界已衍生出多项相关工作。研究者们利用此数据集开发了多种安全帽检测算法,并在不同的应用场景中进行验证。此外,该数据集还促进了相关技术如目标检测、图像分割等在工业安全监控领域的应用研究。
以上内容由遇见数据集搜集并总结生成



