Zhang Guirong, Peng Yiming, Wang Hai
School of Automotive and Traffic Engineering, Jiangsu University, Zhenjiang 212013, China.
Sensors (Basel). 2023 Jul 20;23(14):6543. doi: 10.3390/s23146543.
With the rapid development of the autonomous driving industry, there is increasing research on related perception tasks. However, research on road surface traffic sign detection tasks is still limited. There are two main challenges to this task. First, when the target object's pixel ratio is small, the detection accuracy often decreases. Second, the existing publicly available road surface traffic sign datasets have limited image data. To address these issues, this paper proposes a new instance segmentation network, RTS R-CNN, for road surface traffic sign detection tasks based on Mask R-CNN. The network can accurately perceive road surface traffic signs and provide important information for the autonomous driving decision-making system. Specifically, CSPDarkNet53_ECA is proposed in the feature extraction stage to enhance the performance of deep convolutional networks by increasing inter-channel interactions. Second, to improve the network's detection accuracy for small target objects, GR-PAFPN is proposed in the feature fusion part, which uses a residual feature enhancement module (RFA) and atrous spatial pyramid pooling (ASPP) to optimize PAFPN and introduces a balanced feature pyramid module (BFP) to handle the imbalanced feature information at different resolutions. Finally, data augmentation is used to generate more data and prevent overfitting in specific scenarios. The proposed method has been tested on the open-source dataset Ceymo, achieving a Macro -score of 87.56%, which is 2.3% higher than the baseline method, while the inference speed reaches 23.5 FPS.
随着自动驾驶行业的快速发展,对相关感知任务的研究日益增多。然而,对路面交通标志检测任务的研究仍然有限。该任务主要存在两个挑战。第一,当目标物体的像素占比小时,检测精度往往会下降。第二,现有的公开路面交通标志数据集的图像数据有限。为了解决这些问题,本文基于Mask R-CNN提出了一种用于路面交通标志检测任务的新实例分割网络RTS R-CNN。该网络能够准确感知路面交通标志,并为自动驾驶决策系统提供重要信息。具体而言,在特征提取阶段提出了CSPDarkNet53_ECA,通过增加通道间交互来提升深度卷积网络的性能。其次,为提高网络对小目标物体的检测精度,在特征融合部分提出了GR-PAFPN,它使用残差特征增强模块(RFA)和空洞空间金字塔池化(ASPP)来优化PAFPN,并引入平衡特征金字塔模块(BFP)来处理不同分辨率下不平衡的特征信息。最后,使用数据增强来生成更多数据并防止在特定场景下出现过拟合。所提方法在开源数据集Ceymo上进行了测试,Macro-score达到87.56%,比基线方法高2.3%,同时推理速度达到23.5 FPS。