当前位置：首页 > article >正文

电气设备施工现场风险状态判断ai模型训练数据集

article 2024/9/23 1:20:47

电气设备施工现场风险状态判断ai模型训练数据集
id:18 电气设备施工现场工人人工智能学习数据和工作环境安全数据，建立系统化管理体系，改变全球EHS范式，预防工业事故。数据集记录了387709例子电力设施建设以及施工现场相关的灾害安全环境数据，格式为jpg；同时以边界框、关键点作为标签，主要以json记录。

数据集名称

电气设备施工现场风险状态判断AI模型训练数据集

数据集描述

该数据集旨在用于训练人工智能模型，以识别和判断电气设备施工现场的风险状态，从而帮助建立系统化的管理体系，预防工业事故。数据集包含了大量与电力设施建设和施工现场相关的灾害安全环境数据，可用于训练模型识别工人在施工过程中的不安全行为和环境风险。

数据规模

数据集共记录了387,709个样本，每个样本都是一个与电气设备施工现场相关的图像，格式为JPG。

标签格式

数据集中的标签采用了边界框（bounding boxes）和关键点（keypoints）的方式进行标注，主要以JSON格式记录。这种标注方式使得模型不仅可以识别出图像中的对象，还可以精确到对象的具体位置和姿态。

标注类别

数据集中的标注类别可能包括但不限于：

工人未佩戴安全帽或安全带。
高空作业时无安全措施。
施工现场的其他不安全行为或环境因素。
关键点标注（例如，工人头部、肩膀、手臂等的位置）。

数据集结构

典型的JSON标签文件结构如下：

1{
2    "image_id": "000001",
3    "width": 1920,
4    "height": 1080,
5    "objects": [
6        {
7            "category": "未佩戴安全帽",
8            "bbox": [100, 150, 200, 300],
9            "keypoints": [150, 200, 1, 200, 250, 2, ...]
10        },
11        {
12            "category": "高空作业无人监护",
13            "bbox": [300, 200, 500, 400],
14            "keypoints": [350, 250, 1, 400, 350, 2, ...]
15        }
16    ]
17}

数据来源

数据来源于真实的电气设备施工现场，涵盖了不同的时间段、天气条件和地理环境，以确保数据集的多样性和实用性。

应用场景

该数据集可以用于训练和评估机器学习模型，在电力施工和工业安全领域有着广泛的应用前景，包括但不限于：

自动识别施工现场的安全隐患。
辅助现场管理人员及时发现并纠正不安全行为。
提升施工人员的安全意识。
实现智能化的风险管理和预防机制。

示例代码

下面是一个使用Python和相关库（如OpenCV、PyTorch等）来加载和展示数据集的简单示例代码：

1import os
2import json
3import cv2
4import numpy as np
5from PIL import Image
6
7# 数据集路径
8dataset_path = 'path/to/dataset/'
9
10# 加载图像和标签
11def load_image_and_labels(image_path, label_path):
12    # 读取图像
13    image = Image.open(image_path).convert('RGB')
14    # 加载JSON标签文件
15    with open(label_path, 'r') as f:
16        data = json.load(f)
17    return image, data
18
19# 展示图像
20def show_image_with_boxes(image, data):
21    img = np.array(image)
22    for obj in data['objects']:
23        category = obj['category']
24        bbox = obj['bbox']
25        keypoints = obj['keypoints']
26        
27        # 绘制边界框
28        cv2.rectangle(img, (bbox[0], bbox[1]), (bbox[2], bbox[3]), (0, 255, 0), 2)
29        
30        # 绘制关键点
31        for i in range(0, len(keypoints), 3):
32            if keypoints[i + 2] == 1:  # 可见的关键点
33                cv2.circle(img, (int(keypoints[i]), int(keypoints[i + 1])), 5, (0, 0, 255), -1)
34        
35        # 添加类别标签
36        cv2.putText(img, category, (bbox[0], bbox[1] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
37    
38    cv2.imshow('Image with Boxes and Keypoints', img)
39    cv2.waitKey(0)
40    cv2.destroyAllWindows()
41
42# 主函数
43if __name__ == "__main__":
44    images_dir = os.path.join(dataset_path, 'JPEGImages')
45    labels_dir = os.path.join(dataset_path, 'Annotations')
46    
47    # 获取图像列表
48    image_files = [f for f in os.listdir(images_dir) if f.endswith('.jpg')]
49    
50    # 随机选择一张图像
51    selected_image = np.random.choice(image_files)
52    image_path = os.path.join(images_dir, selected_image)
53    label_path = os.path.join(labels_dir, selected_image.replace('.jpg', '.json'))
54    
55    # 加载图像和标签
56    image, labels = load_image_and_labels(image_path, label_path)
57    
58    # 展示带有标注框和关键点的图像
59    show_image_with_boxes(image, labels)

这段代码展示了如何加载图像和其对应的JSON标签文件，并在图像上绘制边界框和关键点。您可以根据实际需求进一步扩展和修改这段代码，以适应您的具体应用场景。