Ultralytics YOLO27:

Construction-PPE 数据集#

Construction-PPE Dataset In Colab

Ultralytics Construction-PPE 数据集是一个目标检测数据集,包含 1,416 张图像(1,132 张用于训练、143 张用于验证、141 张用于测试),覆盖 11 个类别,用于检测个人防护装备(PPE)——安全帽、手套、背心、靴子和护目镜——并标记施工现场缺失的装备。该数据集整理自真实施工环境,既包含合规案例,也包含不合规案例,是训练用于监控工作场所安全的计算机视觉模型的实用资源。



Watch: How to train Ultralytics YOLO on Personal Protective Equipment Dataset | VisionAI in Construction 👷

数据集结构#

Construction-PPE 数据集包含 1,416 张图像,分为三个预定义子集,由 construction-ppe.yaml 配置定义:

划分图像标注
训练1,132
验证143
测试141

每张图像都采用 Ultralytics YOLO 格式进行标注,确保与先进的目标检测跟踪流程兼容。

该数据集提供 11 个类别,涵盖已佩戴装备、缺失装备和人员:

  • 已佩戴的 PPE(5)helmetglovesvestbootsgoggles
  • 缺失的 PPE(4)no_helmetno_glovesno_bootsno_goggle
  • 其他(2)Personnone

将已佩戴和缺失的标签配对后,模型既能检测正确佩戴的装备,也能标记安全违规行为。请注意,vest 没有专门的缺失背心标签。

业务价值#

建筑施工是危险性最高的行业之一,而其面临的挑战通常在于执行,而不是缺乏法规。健康与安全团队人手紧张,无法实时观察繁忙且不断变化的施工现场的每个角落。

基于计算机视觉的 PPE 检测有助于弥合这一差距。通过自动检查工人是否佩戴规定的安全帽、背心和其他装备,它可以在不同施工现场一致地执行安全规则,并发现风险的先行指标——在事故发生前揭示合规趋势。PPE 检测还可以标记未经授权进入现场的人员,因为他们通常最先出现未佩戴适当安全装备的情况。

应用#

Construction-PPE 可支持各种以安全为重点的计算机视觉应用:

  • 自动合规监控:训练 AI 模型,立即检查工人是否佩戴安全帽、背心或手套等规定的安全装备,降低现场风险。
  • 工作场所安全分析:持续跟踪 PPE 使用情况,发现频繁违规行为,并生成有助于改善安全文化的洞察。
  • 智能监控系统:将检测模型与摄像头连接,在 PPE 缺失时发送实时警报,防止事故发生。
  • 机器人和自主系统:让无人机或机器人能够在大型施工现场执行 PPE 检查,从而支持更快速、更安全的巡检。
  • 研究与教育:为探索工作场所安全和人机交互的学生与研究人员提供真实世界数据集。

无需管理本地基础设施,即可在浏览器中通过 Ultralytics Platform 完成 PPE 检测模型的标注、训练和部署全流程。

数据集 YAML#

Construction-PPE 数据集包含一个 YAML 配置文件,其中定义了训练、验证和测试图像路径以及完整的目标类别列表。你可以在此处直接访问 Ultralytics 仓库中的 construction-ppe.yaml 文件:https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/construction-ppe.yaml

ultralytics/cfg/datasets/construction-ppe.yaml
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# Construction-PPE dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/detect/construction-ppe
# Example usage: yolo train data=construction-ppe.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── construction-ppe ← downloads here (178.4 MB)

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: construction-ppe # dataset root dir
train: images/train # train images (relative to 'path') 1132 images
val: images/val # val images (relative to 'path') 143 images
test: images/test # test images (relative to 'path') 141 images

# Classes
names:
  0: helmet
  1: gloves
  2: vest
  3: boots
  4: goggles
  5: none
  6: Person
  7: no_helmet
  8: no_goggle
  9: no_gloves
  10: no_boots

# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/construction-ppe.zip

使用方法#

你可以使用 Construction-PPE 数据集训练 YOLO26n 模型 100 个周期,图像尺寸设为 640。以下示例展示了如何快速开始。如需了解更多选项和高级配置,请参阅训练指南

训练示例
from ultralytics import YOLO

# Load pretrained model
model = YOLO("yolo26n.pt")

# Train the model on Construction-PPE dataset
model.train(data="construction-ppe.yaml", epochs=100, imgsz=640)

示例图像和标注#

该数据集涵盖不同环境、光照条件和姿态下的建筑工人,同时包含合规不合规案例。

带有安全装备检测的 Construction-PPE 数据集示例

许可证与署名#

Construction-PPE 基于 AGPL-3.0 许可证开发和发布,在正确署名的前提下支持开源研究和商业应用。

如果你在研究中使用此数据集,请引用:

引用
@dataset{Dalvi_Construction_PPE_Dataset_2025,
    author = {Mrunmayee Dalvi and Niyati Singh and Sahil Bhingarde and Ketaki Chalke},
    title = {Construction-PPE: Personal Protective Equipment Detection Dataset},
    month = {January},
    year = {2025},
    version = {1.0.0},
    license = {AGPL-3.0},
    url = {https://docs.ultralytics.com/datasets/detect/construction-ppe},
    publisher = {Ultralytics}
}

常见问题#

  • 与通用建筑施工数据集不同,Construction-PPE 明确包含缺失装备类别no_helmetno_glovesno_bootsno_goggle)。这种双重标注方法不仅让模型能够检测已佩戴的 PPE,还能实时标记违规行为。

  • Construction-PPE 数据集包含 11 个类别:五个已佩戴 PPE 项目(helmetglovesvestbootsgoggles)、四个缺失 PPE 标签(no_helmetno_glovesno_bootsno_goggle),以及 Person 和通用的 none 类别。请注意,vest 没有专门的缺失背心标签。

  • Construction-PPE 数据集包含 1,416 张图像,涵盖 11 个类别——1,132 张用于训练、143 张用于验证、141 张用于测试。有关完整的数据集划分和类别细分,请参阅数据集结构部分。

  • 该数据集(178.4 MB)会在你首次使用 data="construction-ppe.yaml" 进行训练时自动下载,无需手动操作。Ultralytics 会将其获取并解压到本地数据集目录。你可以在目标检测数据集概览中浏览相关数据集。

  • 要使用 Construction-PPE 数据集训练 YOLO26 模型,你可以使用以下代码片段:

    训练示例
    from ultralytics import YOLO
    
    # Load a model
    model = YOLO("yolo26n.pt")  # load a pretrained model (recommended for training)
    
    # Train the model
    results = model.train(data="construction-ppe.yaml", epochs=100, imgsz=640)
  • 适合。图像整理自不同条件下的真实施工现场,因此该数据集非常适合用于构建可部署的工作场所安全监控系统。

  • 该数据集支持对个人防护装备进行实时检测,有助于监控施工现场的工人安全。由于同时包含已佩戴和缺失装备类别,它支持 AI 系统自动标记安全违规行为、生成合规洞察并降低风险。此外,它还为开发工作场所安全、机器人和学术研究领域的计算机视觉解决方案提供了实用资源。

评论