雨雾天气交通目标检测数据集：8类别、5,200张图像 | 目标检测

源码数据分享

链接: https://pan.baidu.com/s/1H8eSFTu0lkrHPxVrObjRJg?pwd=djjq

提取码: djjq

一、引言：恶劣天气——自动驾驶

自动驾驶技术正在快速改变人类的出行方式，从L2级辅助驾驶到L4级无人驾驶，技术的每一次突破都令人振奋。然而，当大雨倾盆、浓雾弥漫时，即便是目前最先进的自动驾驶系统，其感知能力也会急剧下降。据美国国家公路交通安全管理局（NHTSA）统计，约21%的交通事故与恶劣天气相关，而雨雾天气下的交通事故致死率是晴天的2.5倍。

恶劣天气对视觉感知系统的影响是多方面的：雨水在摄像头镜片上形成水膜，严重降低图像清晰度；雾气散射光线，导致远处目标几乎不可见；路面反射造成强光干扰，影响特征提取。这些因素叠加在一起，使得雨雾天气下的目标检测成为一个极具挑战性的课题。

本8类雨天雾天交通目标检测数据集，正是在这一行业痛点驱动下构建的专业数据资源。5200张高质量雨雾天气实景标注图像，覆盖城市道路、高速公路、十字路口等多种交通环境，为全天候目标检测算法研发提供了关键数据支撑。

二、数据集全景剖析

2.1 基本信息

2.2 类别体系

数据集共包含8类道路交通核心目标：

类别设计覆盖自动驾驶感知与智慧交通监测中的主要交通参与者及交通基础设施，可满足多目标检测、交通态势分析和环境感知等任务需求。

2.3 目录结构

database/恶劣天气交通目标检测数据集

├── train
│   └── images
│
├── valid
│   └── images
│
└── test
    └── images

三、恶劣天气目标检测的技术挑战

3.1 雨天检测挑战

雨水对视觉系统的影响主要表现在三个方面：

镜头水膜效应：雨滴附着在摄像头表面形成水膜，导致图像整体模糊、对比度下降。这是最直接的干扰因素，物理上可通过疏水涂层和雨刷缓解，算法上需要图像去雾和增强技术。

背景干扰增强：雨丝在图像中形成大量细长条纹，容易与细长目标（如行人、电线杆）混淆。地面积水反射形成虚像，可能导致误检。

光照条件恶化：雨天光线暗淡，目标与背景的对比度降低，特别是行人穿着深色雨衣时，几乎与背景融为一体。

3.2 雾天检测挑战

雾天对视觉系统的影响更为深远：

能见度降低：浓雾天气下，能见度可能降至50米以下，远距离目标完全不可见。这意味着模型需要从极少的信息量中做出判断。

颜色失真：雾气散射导致远处目标颜色偏白偏灰，丢失原有的颜色信息，影响基于颜色的分类。

边缘模糊：雾气使目标边缘变得模糊，边界框的精确定位变得困难，直接影响检测精度。

对比度衰减：随着距离增加，目标对比度呈指数衰减，这是雾天检测的核心难点。

3.3 低能见度目标特征

数据集中大量目标存在以下特征：

• 雾气遮挡：目标部分或大部分被雾气遮蔽
• 雨滴干扰：图像中存在明显的雨丝或雨滴
• 远距离模糊目标：远处目标仅呈现模糊轮廓
• 低对比度目标：目标与背景对比度极低
• 光线衰减目标：远处目标亮度明显降低

四、恶劣天气检测算法优化

4.1 图像预处理增强

在模型推理前进行图像增强，是提升恶劣天气检测效果的直接手段：

去雾算法：

import cv2
import numpy as np

def dehaze(image, omega=0.95, t0=0.1, radius=15):
    """基于暗通道先验的去雾算法"""
    # 计算暗通道
    dark = np.min(image, axis=2)
    # 估计大气光
    atmospheric_light = estimate_atmospheric_light(dark, image)
    # 估计透射率
    transmission = estimate_transmission(dark, atmospheric_light, omega, radius)
    # 透射率下界约束
    transmission = np.clip(transmission, t0, 1.0)
    # 恢复无雾图像
    result = np.zeros_like(image, dtype=np.float64)
    for i in range(3):
        result[:,:,i] = (image[:,:,i] - atmospheric_light[i]) / transmission + atmospheric_light[i]
    return np.clip(result, 0, 255).astype(np.uint8)

雨滴去除：

• 基于物理模型的雨滴去除
• 基于深度学习的雨滴检测与修复
• 时序滤波消除雨丝干扰

对比度增强：

• CLAHE（限制对比度自适应直方图均衡化）
• Retinex增强算法
• 多尺度细节增强

4.2 域适应训练

将晴天预训练模型迁移到雨雾天气域，需要解决域偏移问题：

域随机化：在训练时随机叠加雨雾效果，增强模型对恶劣天气的适应能力。

def add_fog(image, fog_intensity=0.5):
    """模拟雾天效果"""
    h, w = image.shape[:2]
    fog_layer = np.ones((h, w, 3), dtype=np.float64) * 200
    blend = cv2.addWeighted(image.astype(np.float64), 1 - fog_intensity, 
                            fog_layer, fog_intensity, 0)
    return blend.astype(np.uint8)

def add_rain(image, rain_intensity=0.3):
    """模拟雨天效果"""
    h, w = image.shape[:2]
    rain_layer = np.zeros_like(image)
    for _ in range(int(h * w * rain_intensity * 0.01)):
        x, y = np.random.randint(0, w), np.random.randint(0, h)
        length = np.random.randint(10, 30)
        cv2.line(rain_layer, (x, y), (x + 2, y + length), (200, 200, 200), 1)
    return cv2.addWeighted(image, 1.0, rain_layer, 0.5, 0)

风格迁移：使用CycleGAN将晴天图像转换为雨雾风格，扩充恶劣天气训练数据。

混合域训练：同时使用晴天和雨雾天气数据训练，通过域分类器和梯度反转层实现域不变特征学习。

4.3 注意力机制增强

在恶劣天气下，目标的关键特征往往被干扰信息掩盖，注意力机制可以引导模型聚焦于有效信息：

• 通道注意力：SE-Net、ECA-Net，关注信息量最大的特征通道
• 空间注意力：CBAM空间注意力模块，关注目标所在的空间位置
• 自注意力：Transformer自注意力机制，建立长距离依赖关系
• 可变形注意力：DAT等，根据目标形状自适应调整注意力区域

4.4 多尺度特征融合优化

恶劣天气下，不同距离的目标清晰度差异极大，需要更强的多尺度特征融合能力：

• BiFPN：加权双向特征金字塔网络，实现高效的多尺度特征融合
• PANet：路径聚合网络，增强自底向上的信息传递
• AFPN：渐进式特征金字塔，逐步融合相邻尺度特征

五、模型训练与评估

5.1 训练配置

# data.yaml
path: database/恶劣天气交通目标检测数据集
train: train/images
val: valid/images
test: test/images

nc: 8
names:
  0: person
  1: bicycle
  2: car
  3: motorcycle
  4: bus
  5: train
  6: truck
  7: traffic_light

# 训练命令（增加数据增强）
yolo detect train \
  data=data.yaml \
  model=yolov8m.pt \
  epochs=200 \
  imgsz=640 \
  batch=16 \
  lr0=0.01 \
  hsv_h=0.02 \
  hsv_s=0.7 \
  hsv_v=0.4 \
  degrees=5 \
  translate=0.1 \
  scale=0.5 \
  fliplr=0.5 \
  mosaic=1.0 \
  mixup=0.1 \
  copy_paste=0.1

5.2 性能基准

在不同天气条件下的检测性能对比：

可以明显看到，雾天对检测性能的影响最为严重，尤其是行人和交通信号灯这类小目标。

5.3 增强策略对比

渐进式叠加优化策略，全天气mAP50从76.1%提升至82.7%，提升6.6个百分点。

六、工程部署方案

6.1 车载端部署

自动驾驶场景中，模型需要部署在车载计算平台上：

硬件选型：

• NVIDIA Orin X（254 TOPS）：高端方案
• NVIDIA Orin Nano（40 TOPS）：性价比方案
• 地平线J5（128 TOPS）：国产方案

部署流程：

PyTorch模型 → ONNX → TensorRT Engine → 车载推理

实时性要求：

• 检测延迟 < 50ms
• 吞吐量 > 20 FPS
• 行人检测召回率 > 95%

6.2 智能辅助驾驶（ADAS）

在ADAS系统中，恶劣天气检测模型可用于：

• 前向碰撞预警（FCW）：雨天制动距离加长，提前预警更为关键
• 行人检测与保护（PCD）：低能见度下行人识别，减少人车事故
• 交通信号识别（TSR）：雾天信号灯识别，辅助驾驶员决策
• 车道偏离预警（LDW）：雨天标线模糊，需要更强的识别能力

6.3 智慧交通监控

在交通监控场景中：

• 恶劣天气交通态势感知
• 低能见度下的事故检测
• 雨雾天气车速预警
• 能见度等级自动评估

七、未来发展方向

7.1 多传感器融合

视觉传感器在恶劣天气下的局限性，需要通过传感器融合来弥补：

• 视觉+毫米波雷达：雷达不受雨雾影响，提供距离和速度信息
• 视觉+激光雷达：LiDAR在雾天性能也会下降，但中近距离仍有效
• 视觉+红外：红外成像在夜间和雾天具有优势

7.2 自适应推理

根据天气条件自动调整推理策略：

• 晴天：轻量模型，低延迟
• 小雨/轻雾：标准模型，平衡精度和速度
• 大雨/浓雾：高精度模型，可能牺牲部分速度

7.3 端到端感知规划

将检测、跟踪、预测和规划统一到端到端框架中，让感知系统在恶劣天气下也能做出合理的驾驶决策。

八、总结

本8类恶劣天气交通目标检测数据集共包含5200张高质量雨天、雾天实景标注图像，覆盖复杂天气条件下的核心交通目标。数据集具有恶劣天气场景丰富、标注质量高、目标类别全面、工程实用性强等特点，可广泛应用于自动驾驶感知、智慧交通监测、全天候目标检测算法研发以及复杂环境计算机视觉研究。

在自动驾驶从实验室走向大规模商用的进程中，全天候感知能力是最关键的里程碑之一。只有在暴雨浓雾中依然可靠运行的感知系统，才能真正赢得用户的信任，让自动驾驶技术从"晴天可用"进化为"全天候可靠"。