复杂天气条件下的道路交通目标检测方法
DOI:
作者:
作者单位:

中国人民公安大学 交通管理学院

作者简介:

通讯作者:

中图分类号:

TN911.73-34;TP391.41

基金项目:

北斗高精度时空同步的自主式道路交通系统安全态势感知、评价与控制技术


Road Traffic Target Detection Method Under Complex Weather Conditions
Author:
Affiliation:

1.School of Traffic Management,People'2.'3.s Public Security University of China,Beijing 10037,China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对现有模型在复杂道路场景中小目标特征模糊、复杂背景干扰、远距离目标定位不准及传统损失函数泛化性不足等问题,本文基于RT-DETR-R18框架,提出了一种名为PM-DETR(pyramid and multi-cognitive detection Transformer)道路交通目标检测算法。首先在主干网络构建部分注意力特征提取模块,通过PATConv(partial attention convolution)的三个子模块PAT_ch、PAT_sf、PAT_sp分别优化模型中低层的细粒度特征提取与高层全局关联建模能力;然后在颈部网络采用AIFI_MSM(attention-based intra-scale feature interaction with multi-Scale attention and multi-cognitive adapter)模块替换原有AIFI模块,结合CA(coordinate attention)模块,增强模型对复杂天气、明暗变化的鲁棒性并扩大感受野,改善了尺度失配导致的漏检问题;最后将损失函数由GIoU替换为Inner-Shape-IoU(inner-shape intersection over union),提升非正方形目标与小目标的定位精度。将PM-DETR模型应用于BDD100K(Berkeley DeepDrive 100K dataset)数据集构建含过曝、雨雾雪等复杂场景的实验数据集,该模型的准确率(P)、召回率(R)、mAP50及mAP50-95分别提升至77.547%、61.694%、70.706%和37.809%,相较于RT-DETR-R18模型分别提升3.913%、1.89%、4.051%、2.312%,且在ACDC和RTTS数据集上进一步验证了模型良好的泛化能力。

    Abstract:

    To address the issues of blurred small-target features, strong complex background interference, inaccurate long-distance target localization, and insufficient generalization of traditional loss functions in complex road scenes, a road traffic object detection algorithm named PM-DETR (Pyramid and Multi-cognitive Detection Transformer) is proposed based on the RT-DETR-R18 framework. First, a partial attention feature extraction module is constructed in the backbone network, where three sub-modules of PATConv (Partial Attention Convolution)—namely PAT_ch, PAT_sf, and PAT_sp are respectively adopted to optimize fine-grained feature extraction in low-level layers and global correlation modeling in high-level layers. Then, the original AIFI module in the neck network is replaced with the proposed AIFI\_MSM (Attention-based Intra-scale Feature Interaction with Multi-scale Attention and Multi-cognitive Adapter) module, which is combined with a CA (Coordinate Attention) module to enhance model robustness against complex weather and illumination variations, expand the receptive field, and alleviate missed detections caused by scale mismatch. Finally, the GIoU loss is replaced by Inner-Shape-IoU (Inner-Shape Intersection over Union) to improve the localization accuracy of non-rectangular objects and small targets. The PM-DETR model is evaluated on the BDD100K (Berkeley DeepDrive 100K dataset) dataset, where an experimental dataset covering overexposure, rain, fog, snow, and other complex scenarios is constructed. The proposed model achieves a precision (P) of 77.547%, a recall (R) of 61.694%, an mAP50 of 70.706%, and an mAP50-95 of 37.809%, which are improvements of 3.913%, 1.89%, 4.051%, and 2.312% over the baseline RT-DETR-R18 model, respectively. Furthermore, the strong generalization capability of the model is further validated on the ACDC and RTTS datasets.

    参考文献
    相似文献
    引证文献
引用本文

刘毅博,马社强,王晟由. 复杂天气条件下的道路交通目标检测方法[J]. 科学技术与工程, , ():

复制
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2026-04-02
  • 最后修改日期:2026-06-22
  • 录用日期:2026-07-31
  • 在线发布日期:
  • 出版日期:
×
2026年会通知 | “技术经济学驱动智能经济生态构建与治理变革”——中国技术经济学会第三十三届学术年会(2026)会议通知暨征文启事(第一轮)
亟待确认版面费归属稿件,敬请作者关注