智慧城市新动力:5000张YOLO格式道路违规识别数据集| 人行横道违规穿越 闯红灯违规识别

随着智慧城市建设的推进,利用人工智能进行高效交通管理成为趋势。今天介绍一个专注于道路违规场景自动识别的YOLO格式数据集,它能为训练检测模型、开发智能监控系统提供基础支持。


数据集核心信息

根据原文,该数据集的核心要素如下:

项目 内容
类别数量 7 类
具体类别 横穿违规、人行横道违规、未戴头盔违规、正常状态、超员违规、行人违规、闯红灯违规
数据总量 5000 张图片
数据集格式 YOLO 格式

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

7大类别详解

该数据集覆盖了交通场景中常见的7类关键状态与违规行为,划分贴合实际管理需求:

  • 横穿违规:行人或非机动车未按规定横穿道路。
  • 人行横道违规:在人行横道区域发生的违规行为。
  • 未戴头盔违规:骑行人员未按规定佩戴安全头盔。
  • 闯红灯违规:车辆或行人违反交通信号灯指示。
  • 行人违规:行人的其他违反交通规则的行为。
  • 超员违规:车辆承载人数超过规定。
  • 正常状态:合规的交通场景,作为对照类别,可辅助模型区分合规与违规场景,提升检测准确性。

数据集核心价值与应用场景

原文明确指出,该数据集具备以下三重核心应用价值

  1. 训练交通违规行为检测模型:实现道路违规场景的自动识别。
  2. 支撑智能交通监控系统开发:通过自动识别多种违规行为,降低人工监控成本,提升交通管理效率。
  3. 为研究与管控提供数据支撑:为交通违规行为分析研究提供标准化数据,并辅助交通管理部门制定精准的管控策略。

快速上手:模型训练代码示例(基于YOLOv8)

以下提供基于 YOLOv8 的训练示例代码,帮助你快速使用该数据集。请注意:以下代码为通用示例,具体训练参数需根据实际硬件和数据调整。

1. 环境安装
pip install ultralytics
2. 数据集准备

将下载的数据集按以下结构组织(YOLO格式):

datasets/
  └── traffic_violation/
        ├── images/
        │   ├── train/  # 训练集图片
        │   └── val/    # 验证集图片
        └── labels/
            ├── train/  # 训练集标签(.txt文件)
            └── val/    # 验证集标签

每个标签文件与图片同名,内容格式为:class_id x_center y_center width height(归一化坐标)。

3. 创建数据配置文件

新建 traffic_violation.yaml,写入以下内容(请根据实际类别名称填写):

path: ./datasets/traffic_violation  # 数据集根目录
train: images/train  # 训练图片相对路径
val: images/val      # 验证图片相对路径

nc: 7  # 类别数量
names: ['crossing_violation', 'crosswalk_violation', 'no_helmet', 'red_light', 'pedestrian_violation', 'overcrowding', 'normal']
4. 模型训练
from ultralytics import YOLO

# 加载预训练模型
model = YOLO('yolov8n.pt')  # n/s/m/l/x 可选

# 开始训练
results = model.train(
    data='traffic_violation.yaml',
    epochs=100,
    imgsz=640,
    batch=16,
    device=0,  # GPU编号,CPU则设为'cpu'
    workers=8,
    patience=20,  # 早停轮数
    augment=True, # 启用数据增强
    project='traffic_violation_detector', # 项目保存路径
    name='exp1' # 实验名称
)
5. 模型验证与推理
# 在验证集上评估模型
metrics = model.val()
print(f"mAP50-95: {metrics.box.map:.4f}")
print(f"mAP50: {metrics.box.map50:.4f}")

# 对单张图片进行推理
results = model('test_image.jpg', save=True)  # 保存检测结果图片

标签聚合

#智慧城市 #城市治理 #道路违规场景自动识别 #智能交通监控系统 #YOLO格式数据集 #闯红灯识别 #斑马线违规识别 #未戴头盔检测 #计算机视觉 #目标检测


Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐