复杂场景下的交通标志小目标检测算法
DOI:
CSTR:
作者:
作者单位:

上海应用技术大学 计算机科学与信息工程学院

作者简介:

通讯作者:

中图分类号:

U491;TP391;TN919.82

基金项目:

国家自然科学基金资助项目(52402381)


Small target detection algorithm for traffic signs in complex scenes
Author:
Affiliation:

Fund Project:

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

    在交通标志识别应用中,待检测目标多为小目标,易出现漏检、误检等问题。针对这些问题,基于YOLOv8s算法设计了一种改进的交通标志识别算法,FKDS-YOLOv8s。使用FasterBlock重构C2f模块,形成新的轻量化模块C2f-Faster,既提升模型特征提取能力,又降低了计算开销;基于SENet和ResNeXt模型设计一种新的检测头Detect_SR,使模型能够有效地聚焦于小目标的关键特征;融入轻量且高效的动态上采样器DySample,显著减少了GPU内存消耗;通过增加上采样和Prediction输出层次,模型能够捕捉丰富的位置信息,有效解决了YOLOv8s模型在处理小目标时信息不足的问题;引入Shape-IoU损失函数,优化了原CIoU在边框回归中的不足;此外,在Neck部分融入了本文新设计的注意力机制DKN-Attention,在上采样和下采样过程中定位微小物体场景的注意力区域,提升了远处小型交通标志的特征提取和识别能力。实验在中国交通标志数据集TT100K上进行,结果表明,FKDS-YOLOv8s相比基准模型,在查准率(P)、查全率(R)和mAP50上分别提升了5.9%、4.2%和6.3%。较传统方法,FKDS-YOLOv8s在性能上表现出显著优势。

    Abstract:

    In the application of traffic sign recognition, most of the targets to be detected are small targets, which are prone to problems such as missed detection and false detection. In order to solve these problems, an improved traffic sign recognition algorithm, FKDS-YOLOv8s, was designed based on the YOLOv8s algorithm. FasterBlock is used to reconstruct the C2f module to form a new lightweight module C2f-Faster, which not only improves the feature extraction ability of the model, but also reduces the computational overhead. Based on the SENet and ResNeXt models, a new detection head Detect_SR was designed to enable the model to effectively focus on the key features of small targets. DySample, a lightweight and efficient dynamic upsampler, significantly reduces GPU memory consumption. By increasing the output level of upsampling and prediction, the model can capture rich position information, which effectively solves the problem of insufficient information when the YOLOv8s model processes small targets. The Shape-

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-10-08
  • 最后修改日期:2024-12-03
  • 录用日期:2024-12-04
  • 在线发布日期:
  • 出版日期:
文章二维码
×
《电子测量技术》
财务封账不开票通知