基于图像掩码及特征融合的交通标志检测

TRAFFIC SIGN DETECTION BASED ON IMAGE MASK AND FEATURE FUSION

  • 摘要: 为解决交通标志检测中小目标检测问题,提出基于目标检测网络 YOLOX 的特征循环融合目标检测模型。通过特征循环融合的方法,促进不同特征图之间位置信息及语义信息的相互补充,利用区域通道注意力机制加深模型在融合过程中对小目标的关注。利用原图像生成图像掩码,通过掩码辅助优化损失函数,从而降低模型对非交通标志的误检,并利用数据平衡的数据增强方法提高模型整体检测性能。通过在公开数据集上进行实验,验证了该模型能够有效地提高交通标志小目标检测精度。

     

    Abstract: In order to solve the problem of small object detection in traffic sign detection, a feature recurrent fusion object detection model based on YOLOX is proposed. The feature recurrent fusion method was used to deepen the mutual complementation of position information and semantic information between different feature maps, and the channel attention mechanism was used to deepen the model’sattention to small targets in the fusion process. The original image was used to generate image mask, and the mask was used to optimize loss function, thereby reducing the false detection of non-traffic signs by the model, and using the data-balanced data augmentation method to improve the overall detection performance of the model. Experimental results on public datasets show that the proposed method can effectively improve the detection accuracy of small targets in traffic signs.

     

/

返回文章
返回