Attention feature fusion network for small traffic sign detection
Engineering Research Express(2022)
摘要
Abstract Object detection has made great progress with the rise of convolutional neural networks in recent years. Traffic sign detection is a research hotspot for object detection tasks. The existing detection models have the problems of inaccurate positioning and low classification accuracy when detecting small traffic signs. To address these issues, in this paper, we propose a small traffic sign detection method based on YOLOv4. Specifically, we design an attention-based feature fusion module including attention spatial pyramid pooling (ASPP) and attention path aggregation networks (APAN). ASPP highlights useful small object information and suppresses invalid interference information in the background. APAN reduces information loss during feature fusion. A large number of experimental results on public datasets show that the method in this paper improves the detection performance of the model. In terms of small traffic sign detection, the method improves YOLOv4 by 12 mAP, and meets the real-time requirements of automatic driving detection (more than 50 FPS).
更多查看译文
关键词
attention,detection,feature
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要