five

SafetyHelmetWearing-Dataset

收藏
github2024-05-20 更新2024-05-31 收录
下载链接:
https://github.com/njvisionpower/SafetyHelmetWearing-Dataset
下载链接
链接失效反馈
官方服务:
资源简介:
安全帽佩戴检测数据集,包含7581张图像,其中有9044个佩戴安全帽的正样本和111514个未佩戴安全帽的负样本。数据集中的正样本来自谷歌或百度,负样本部分来自SCUT-HEAD数据集,并进行了手动标注。数据集格式遵循Pascal VOC标准。

The safety helmet wearing detection dataset comprises 7,581 images, including 9,044 positive samples of individuals wearing safety helmets and 111,514 negative samples of individuals not wearing safety helmets. The positive samples in the dataset are sourced from Google or Baidu, while the negative samples are partially derived from the SCUT-HEAD dataset and have been manually annotated. The dataset format adheres to the Pascal VOC standard.
创建时间:
2019-08-15
原始信息汇总

数据集概述

数据集名称

  • 安全帽佩戴检测数据集 (SafetyHelmetWearing-Dataset, SHWD)

数据集内容

  • 图像数量:7581张
  • 标注对象
    • 安全帽佩戴对象(正例):9044个
    • 正常头部对象(未佩戴或负例):111514个
  • 数据来源
    • 正例对象来自Google或Baidu,手动标注使用LabelImg。
    • 部分负例对象来自SCUT-HEAD数据集,经过修正以适应Pascal VOC格式。

数据集格式

  • 标注格式:Pascal VOC格式
  • 类别名称
    • 正例:"hat"
    • 负例:"person"

数据集下载

预训练模型

  • 模型下载链接
  • 模型基准
    • 模型 | darknet | mobile1.0 | mobile0.25
    • map | 88.5 | 86.3 | 75.0

使用指南

  • 数据集结构

    ---VOC2028
    ---Annotations
    ---ImageSets
    ---JPEGImages

  • 模型测试

    • 使用预训练模型进行推理,支持两种方式:
      • 通过命令行运行python test_yolo.py,可配置网络类型、阈值、GPU使用等。
      • 使用MXNet符号模型进行推理,需下载符号模型后运行python test_symbol.py

训练指南

  • 数据集路径设置:在train_yolo.py中设置数据集路径。

  • 训练示例

    python train_yolo.py --batch-size 4 -j 4 --warmup-epochs 3

注意事项

  • 训练时可能遇到梯度爆炸问题,建议增加预热周期或降低学习率。
  • 数据集加载可能占用大量CPU资源,增加-j参数值可利用多核CPU提升训练速度。
搜集汇总
数据集介绍
main_image_url
构建方式
SafetyHelmetWearing-Dataset(安全帽佩戴检测数据集)的构建基于对安全帽佩戴和未佩戴的头部图像的广泛收集与标注。数据集包含7581张图像,其中9044个对象为佩戴安全帽的正样本,111514个对象为未佩戴安全帽的负样本。正样本图像主要来源于谷歌和百度,并通过手动标注工具LabelImg进行标注。负样本则部分来源于SCUT-HEAD数据集,并对其进行了修正,确保数据能够以标准的Pascal VOC格式直接加载。
特点
该数据集的显著特点在于其丰富的样本数量和多样性,涵盖了不同场景下的安全帽佩戴情况。正负样本的比例经过精心设计,确保模型在训练过程中能够有效区分佩戴与未佩戴安全帽的头部。此外,数据集采用Pascal VOC格式,便于直接应用于现有的计算机视觉框架,并提供了预训练模型,进一步简化了模型的开发与测试过程。
使用方法
使用SafetyHelmetWearing-Dataset时,用户需将数据集解压至指定路径,并按照Pascal VOC格式进行配置。数据集包含两个类别:‘hat’表示佩戴安全帽的正样本,‘person’表示未佩戴安全帽的负样本。用户可通过提供的预训练模型进行推理,或根据需求自行训练模型。训练过程中,可通过调整超参数如学习率、批量大小等优化模型性能。
背景与挑战
背景概述
SafetyHelmetWearing-Dataset(安全帽佩戴检测数据集,SHWD)是由研究人员创建的一个专注于安全帽佩戴检测的数据集。该数据集包含了7581张图像,其中包含9044个佩戴安全帽的正样本和111514个未佩戴安全帽的负样本。数据集的正样本主要来源于谷歌和百度,并通过手动标注工具LabelImg进行标注。负样本则部分来源于SCUT-HEAD数据集,并对原始数据进行了修复,使其能够直接加载为标准的Pascal VOC格式。该数据集的创建旨在推动安全帽佩戴检测技术的发展,特别是在工业安全领域,具有重要的应用价值。
当前挑战
SafetyHelmetWearing-Dataset在构建过程中面临了多个挑战。首先,数据集的标注工作需要高度精确,以确保模型能够准确识别佩戴和未佩戴安全帽的样本。其次,数据集的多样性也是一个挑战,因为需要涵盖不同场景、光照条件和背景下的安全帽佩戴情况。此外,数据集的规模较大,处理和存储这些数据对计算资源提出了较高的要求。最后,模型训练过程中可能遇到的梯度爆炸问题和数据加载速度问题也需要通过优化算法和硬件配置来解决。
常用场景
经典使用场景
SafetyHelmetWearing-Dataset(安全帽佩戴检测数据集)在工业安全领域中具有广泛的应用前景。该数据集主要用于训练和评估安全帽佩戴检测模型,旨在识别图像中工人是否正确佩戴安全帽。通过提供7581张标注图像,其中包括9044个佩戴安全帽的正样本和111514个未佩戴安全帽的负样本,该数据集为研究人员和工程师提供了一个标准化的基准,用于开发和测试基于深度学习的安全帽检测算法。
衍生相关工作
SafetyHelmetWearing-Dataset的发布催生了一系列相关的经典工作。研究人员基于该数据集开发了多种先进的深度学习模型,如YOLO、MobileNet等,用于安全帽佩戴检测。这些模型不仅在学术界引起了广泛关注,还在工业界得到了实际应用。此外,该数据集还激发了对多目标检测、实时监控系统等领域的深入研究,推动了整个计算机视觉领域的技术进步。
数据集最近研究
最新研究方向
在工业安全和智能监控领域,SafetyHelmetWearing-Dataset(安全帽佩戴检测数据集)的研究正聚焦于提升检测算法的精度和实时性。随着深度学习技术的不断进步,研究人员正致力于开发更高效的模型,以应对复杂场景下的安全帽佩戴检测任务。此外,数据集的多样性和标注质量也成为研究的重点,旨在通过增强数据集的覆盖范围和标注准确性,进一步提升模型的泛化能力。这些研究不仅推动了工业安全监控系统的智能化发展,也为相关领域的法规执行和技术应用提供了坚实的基础。
以上内容由遇见数据集搜集并总结生成
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

面向社区/商业的数据集话题

二维码
科研交流群

面向高校/科研机构的开源数据集话题

数据驱动未来

携手共赢发展

商业合作