基于复杂场景的小目标交通标志检测方法研究
THE DETECTION METHOD OF SMALL TARGET TRAFFIC SIGNS IN COMPLEX SCENARIOS
-
摘要: 针对复杂场景的小目标交通标志检测过程中易出现漏检、误检的问题,提出一种基于改进YOLOx-s模型的小目标交通标志检测方法。改进的YOLOx-s模型以ConvNext-T为骨干特征提取网络,并结合Meta-ACON、RFP、Focal-EioU方法,对数据增强后的CCTSDB数据集进行端到端的训练,实现了在不增加注意力机制的情况下,提升模型对小目标的检测性能,同时保持了模型的简洁性。实验结果表明,改进后的模型更加关注目标样本中的小目标交通标志,mAP值提高了5.31百分点。Abstract: For the problem of small-target traffic signs being easily missed and mis-detected in complex scenes, a small-target traffic sign detection method based on the improved YOLOx-s model is proposed. The improved YOLOx-s model used ConvNext-T as the backbone feature extraction network and combined Meta-ACON, RFP, and Focal-EioU methods for end-to-end training of the data-enhanced CCTSDB dataset. It was achieved to improve the detection performance of the model for small targets without increasing the attention mechanism, while maintaining the simplicity of the model. The experimental results show that the improved model pays more attention to small traffic signs in the target sample, and the mAP value improved by 5.31 percentage points.
下载: