变电站绝缘套管红外检测数据集

包含以下4个数据文件： /train：训练集 /valid：验证集 /test：测试集 README.txt：数据说明 【数据说明】检测目标以Pascal VOC格式进行标注，对每个图像进行以下预处理，统一调整大小为640x640。数据集共包含9900+变压器绝缘套管红外图像样张，其中train包含8560+红外图像样张（带xml标注），valid包含1280+红外图像样张（带xml标注），test包含60+红外图像样张（带xml标注），数据集未做数据增广，也可以自行划分训练集、验证集和测试集。 以上数据均已进行脱敏处理，可用于相关红外检测研究，

变电站绝缘套管红外检测数据集

数据集描述

该数据集是一个专门用于变电站绝缘套管红外图像检测的数据集，旨在帮助研究人员和开发者训练和评估基于深度学习的目标检测模型。绝缘套管是变电站中重要的电气设备之一，其状态直接影响电力系统的安全稳定运行。通过红外成像技术，可以有效检测绝缘套管的温度异常，从而及时发现潜在故障。数据集提供了大量的高分辨率红外图像，并且所有标注都是通过Pascal VOC格式进行的，确保了高质量的标注信息。

数据规模

• 总样本数量：9,900+张红外图像
  • 训练集 (train)：8,560+张红外图像（带xml标注）
  • 验证集 (valid)：1,280+张红外图像（带xml标注）
  • 测试集 (test)：60+张红外图像（带xml标注）
• 数据量：具体大小未提供
• 标注格式：Pascal VOC格式XML标注
• 目标类别：
  • 绝缘套管 (Insulator Bushings)

图像特性

• 统一预处理：每张图像都经过预处理，统一调整大小为640x640像素。
• 多样化场景：覆盖了不同类型的绝缘套管、不同的光照条件以及各种背景环境下的红外图像情况。
• 高质量手工标注：每张图像都有详细的边界框标注，支持直接用于训练目标检测模型。
• 真实拍摄：所有图像均为实际拍摄的真实场景，增强了模型在实际应用中的鲁棒性。
• 脱敏处理：数据已进行脱敏处理，可用于相关红外检测研究。
• 单一目标标签：每张图像中的目标标签有且仅有1个为“Insulator Bushings”。

应用场景

• 电力系统维护：通过自动检测绝缘套管的状态，辅助电力部门及时发现和处理故障，提高电力系统的可靠性和安全性。
• 智能巡检：集成到无人机或机器人巡检系统中，实现自动化巡检，减少人工成本。
• 科研分析：用于研究目标检测算法在特定工业应用场景中的表现，特别是在复杂背景和光照条件下的鲁棒性。
• 故障预警：通过早期检测绝缘套管的温度异常，提前预警潜在故障，减少停电风险。

数据集结构

典型的数据集目录结构如下：

1insulator_bushing_infrared_dataset/
2├── train/
3│   ├── images/
4│   │   ├── img_00001.jpg
5│   │   ├── img_00002.jpg
6│   │   └── ...
7│   ├── annotations/
8│   │   ├── img_00001.xml
9│   │   ├── img_00002.xml
10│   │   └── ...
11├── valid/
12│   ├── images/
13│   │   ├── img_00001.jpg
14│   │   ├── img_00002.jpg
15│   │   └── ...
16│   ├── annotations/
17│   │   ├── img_00001.xml
18│   │   ├── img_00002.xml
19│   │   └── ...
20├── test/
21│   ├── images/
22│   │   ├── img_00001.jpg
23│   │   ├── img_00002.jpg
24│   │   └── ...
25│   ├── annotations/
26│   │   ├── img_00001.xml
27│   │   ├── img_00002.xml
28│   │   └── ...
29├── README.txt  # 数据说明文件

数据说明

• 检测目标：以Pascal VOC格式进行标注。
• 图像预处理：每张图像都经过预处理，统一调整大小为640x640像素。
• 数据集内容：
  • 训练集 (train)：包含8,560+张红外图像样张（带xml标注）。
  • 验证集 (valid)：包含1,280+张红外图像样张（带xml标注）。
  • 测试集 (test)：包含60+张红外图像样张（带xml标注）。
• 目标标签：每张图像中的目标标签有且仅有1个为“Insulator Bushings”。
• 数据增广：数据集未做数据增广，用户可以根据需要自行划分训练集、验证集和测试集。
• 脱敏处理：数据已进行脱敏处理，可用于相关红外检测研究。

示例代码

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

1import os
2import cv2
3import numpy as np
4from PIL import Image
5import xml.etree.ElementTree as ET
6
7# 数据集路径
8dataset_path = 'path/to/insulator_bushing_infrared_dataset/'
9
10# 加载图像和边界框标注
11def load_image_and_boxes(image_path, annotation_path):
12    # 读取图像
13    image = Image.open(image_path).convert('RGB')
14    
15    # 解析VOC标注文件
16    tree = ET.parse(annotation_path)
17    root = tree.getroot()
18    boxes = []
19    for obj in root.findall('object'):
20        class_name = obj.find('name').text
21        bbox = obj.find('bndbox')
22        xmin = int(bbox.find('xmin').text)
23        ymin = int(bbox.find('ymin').text)
24        xmax = int(bbox.find('xmax').text)
25        ymax = int(bbox.find('ymax').text)
26        boxes.append([class_name, xmin, ymin, xmax, ymax])
27    return image, boxes
28
29# 展示图像和边界框
30def show_image_with_boxes(image, boxes):
31    img = np.array(image)
32    for box in boxes:
33        class_name, xmin, ymin, xmax, ymax = box
34        cv2.rectangle(img, (xmin, ymin), (xmax, ymax), (0, 255, 0), 2)
35        label = f'{class_name}'
36        cv2.putText(img, label, (xmin, ymin - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
37    
38    cv2.imshow('Image with Boxes', img)
39    cv2.waitKey(0)
40    cv2.destroyAllWindows()
41
42# 主函数
43if __name__ == "__main__":
44    subset = 'train'  # 可以选择 'train', 'valid', 或 'test'
45    images_dir = os.path.join(dataset_path, subset, 'images')
46    annotations_dir = os.path.join(dataset_path, subset, 'annotations')
47    
48    # 获取图像列表
49    image_files = [f for f in os.listdir(images_dir) if f.endswith('.jpg')]
50    
51    # 随机选择一张图像
52    selected_image = np.random.choice(image_files)
53    image_path = os.path.join(images_dir, selected_image)
54    annotation_path = os.path.join(annotations_dir, selected_image.replace('.jpg', '.xml'))
55    
56    # 加载图像和边界框
57    image, boxes = load_image_and_boxes(image_path, annotation_path)
58    
59    # 展示带有边界框的图像
60    show_image_with_boxes(image, boxes)

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

示例代码：使用预训练模型进行推理

以下是使用预训练模型进行推理的示例代码。这里我们假设您使用的是基于YOLOv5的模型，但您可以根据需要选择其他支持目标检测的模型。

1import torch
2import cv2
3import numpy as np
4from PIL import Image
5import yolov5  # 请确保已安装yolov5库
6
7# 数据集路径
8dataset_path = 'path/to/insulator_bushing_infrared_dataset/'
9
10# 加载预训练模型
11model = yolov5.load('path/to/pretrained/yolov5_weights.pt')  # 替换成实际的预训练模型路径
12model.eval()
13
14# 主函数
15if __name__ == "__main__":
16    subset = 'train'  # 可以选择 'train', 'valid', 或 'test'
17    images_dir = os.path.join(dataset_path, subset, 'images')
18    
19    # 获取图像列表
20    image_files = [f for f in os.listdir(images_dir) if f.endswith('.jpg')]
21    
22    # 随机选择一张图像
23    selected_image = np.random.choice(image_files)
24    image_path = os.path.join(images_dir, selected_image)
25    
26    # 读取并预处理图像
27    image = Image.open(image_path).convert('RGB')
28    
29    # 使用预训练模型进行推理
30    results = model(image)
31    
32    # 处理预测结果
33    boxes = results.xyxy[0].cpu().numpy()
34    
35    # 在图像上绘制边界框
36    img = np.array(image)
37    for box in boxes:
38        xmin, ymin, xmax, ymax, conf, class_id = box
39        class_name = results.names[int(class_id)]
40        label = f'{class_name} {conf:.2f}'
41        cv2.rectangle(img, (int(xmin), int(ymin)), (int(xmax), int(ymax)), (0, 255, 0), 2)
42        cv2.putText(img, label, (int(xmin), int(ymin) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
43    
44    # 显示结果
45    cv2.imshow('Image with Boxes', img)
46    cv2.waitKey(0)
47    cv2.destroyAllWindows()

这段代码展示了如何使用预训练的YOLOv5模型进行推理，并显示和保存推理结果。您可以根据实际需求进一步扩展和修改这段代码，以适应您的具体应用场景。如果您需要使用其他模型进行更高级的功能，如模型微调或增量训练，可以参考相应模型的官方文档来进行相应的配置和操作。