睡岗检测数据集(工作 课堂等) 3100张 增强 睡岗趴睡 带标注 voc yolo

睡岗检测是一项重要的安全监测任务,尤其是在需要高度警觉的工作环境中。该数据集旨在为研究人员和开发者提供一个高质量的、可用于训练和评估睡岗检测模型的数据集。

数据集特点
  • 类型:睡岗检测图像数据集。
  • 规模:包含3100张图像,涵盖了各种不同环境下的睡岗和趴睡场景。
  • 标注:提供详细的标注信息,支持 VOC 和 YOLO 两种标注格式。
  • 多样性:数据集包含了不同光线条件、不同角度、不同背景的睡岗图像。
  • 增强:通过对原始图像进行多种增强处理(如旋转、翻转、色彩调整等),增加了数据集的多样性和鲁棒性。

数据集组成

  • 训练集:用于训练模型,包含大多数图像。
  • 验证集:用于模型调参和验证,包含一定比例的图像。
  • 测试集:用于最终评估模型性能,包含一定比例的图像。

数据集目录结构

复制代码
1sleep_detection_dataset/
2├── images/
3│   ├── train/
4│   ├── val/
5│   └── test/
6├── annotations_voc/
7│   ├── train/
8│   ├── val/
9│   └── test/
10├── annotations_yolo/
11│   ├── train/
12│   ├── val/
13│   └── test/
14└── dataset.yaml  # YOLOv5 configuration file

标注信息

示例

每张图像都附带一个 XML 文件(VOC 格式)和一个 TXT 文件(YOLO 格式),包含睡岗对象的位置信息。

示例标注文件
VOC 格式
复制代码
1<annotation>
2    <folder>sleep_detection_dataset</folder>
3    <filename>000001.jpg</filename>
4    <object>
5        <name>Sleeping</name>
6        <pose>Unspecified</pose>
7        <truncated>0</truncated>
8        <difficult>0</difficult>
9        <bndbox>
10            <xmin>100</xmin>
11            <ymin>200</ymin>
12            <xmax>300</xmax>
13            <ymax>400</ymax>
14        </bndbox>
15    </object>
16</annotation>
YOLO 格式
10 0.45 0.55 0.2 0.3  # 类别索引0表示Sleeping,后续数字表示中心点坐标和宽高相对于图像尺寸的比例

使用指南

  1. 环境搭建:确保安装了YOLOv5或其他必要的软件包。
  2. 数据集准备:下载数据集并按照上述目录结构组织数据。
  3. 模型配置:设置YOLOv5或其他模型的训练参数。
  4. 模型训练:使用训练集对模型进行训练。
  5. 模型评估:使用验证集和测试集评估模型性能。

关键代码示例

1. 数据集准备
1import os
2import xml.etree.ElementTree as ET
3
4def parse_voc_annotation(xml_path):
5    tree = ET.parse(xml_path)
6    root = tree.getroot()
7    size = root.find('size')
8    width = int(size.find('width').text)
9    height = int(size.find('height').text)
10
11    objects = []
12    for obj in root.iter('object'):
13        difficult = obj.find('difficult').text
14        cls = obj.find('name').text
15        if cls not in ['Sleeping'] or int(difficult) == 1:
16            continue
17        xmlbox = obj.find('bndbox')
18        b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text), float(xmlbox.find('ymax').text))
19        objects.append([cls, b])
20    return objects, width, height
21
22def load_data(data_dir):
23    data = []
24    for split in ['train', 'val', 'test']:
25        images_dir = os.path.join(data_dir, 'images', split)
26        annotations_dir = os.path.join(data_dir, 'annotations_voc', split)
27        for img_name in os.listdir(images_dir):
28            img_path = os.path.join(images_dir, img_name)
29            ann_path = os.path.join(annotations_dir, img_name.replace('.jpg', '.xml'))
30            objects, width, height = parse_voc_annotation(ann_path)
31            data.append((img_path, objects, width, height))
32    return data
33
34data_dir = 'path/to/sleep_detection_dataset'
35data = load_data(data_dir)

报告和文档

  • 报告:报告应包括数据集背景、数据集组成、标注信息、使用指南等内容。
  • 文档:文档应包括数据集的组织方式、标注信息、模型训练及评估的具体步骤等。

注意事项

  • 确保数据集的质量,特别是标注信息的准确性。
  • 在训练过程中,注意监控模型的学习曲线,确保模型没有过拟合。
  • 考虑到睡岗检测的多样性和复杂性,建议使用较大的模型和较长的训练周期以获得更好的性能。
相关推荐
xrgs_shz9 分钟前
MATLAB读入不同类型图像并显示图像和相关信息
图像处理·计算机视觉·matlab
糖豆豆今天也要努力鸭29 分钟前
torch.__version__的torch版本和conda list的torch版本不一致
linux·pytorch·python·深度学习·conda·torch
何大春1 小时前
【弱监督语义分割】Self-supervised Image-specific Prototype Exploration for WSSS 论文阅读
论文阅读·人工智能·python·深度学习·论文笔记·原型模式
Suyuoa1 小时前
附录2-pytorch yolov5目标检测
python·深度学习·yolo
余生H2 小时前
transformer.js(三):底层架构及性能优化指南
javascript·深度学习·架构·transformer
罗小罗同学3 小时前
医工交叉入门书籍分享:Transformer模型在机器学习领域的应用|个人观点·24-11-22
深度学习·机器学习·transformer
孤独且没人爱的纸鹤3 小时前
【深度学习】:从人工神经网络的基础原理到循环神经网络的先进技术,跨越智能算法的关键发展阶段及其未来趋势,探索技术进步与应用挑战
人工智能·python·深度学习·机器学习·ai
阿_旭3 小时前
TensorFlow构建CNN卷积神经网络模型的基本步骤:数据处理、模型构建、模型训练
人工智能·深度学习·cnn·tensorflow
羊小猪~~3 小时前
tensorflow案例7--数据增强与测试集, 训练集, 验证集的构建
人工智能·python·深度学习·机器学习·cnn·tensorflow·neo4j
极客代码3 小时前
【Python TensorFlow】进阶指南(续篇三)
开发语言·人工智能·python·深度学习·tensorflow