北京交通大学自动化与智能学院

殷辰堃课题组论文发表于《International Conference on Robotics, Intelligent Control and Artificial Intelligence (RICAI)》

发布于 2024-01-08

    我所殷辰堃老师课题组完成的论文《Target Detection Algorithm Based on Improved YOLOv5 for Edge Computing Devices》于2023-12在《International Conference on Robotics, Intelligent Control and Artificial Intelligence (RICAI)》正式发表。论文摘要如下:

Abstract: This paper selects YOLOv5 as the basic model of edge computing device target detection. Aiming at the performance imbalance of the current deep neural network target detection algorithms and the high latency of deploying YOLOv5 on edge computing devices, an improved target detection algorithm based on YOLOv5 is named AG-YOLOv5 (Attention Ghost YOLOv5). The improvement of the algorithm mainly focuses on the backbone network and necking,to improve the comprehensive performance of the model. Firstly, in the backbone network, the Deep Stem module was used to replace convolutions, the improved C3Ghost is named S-C3Ghost(Strong C3Ghost) that was used to replace the CSP module, and SimSPPF was used instead of SPPF to achieve lightweight network and to improve inference speed. Secondly, attention mechanisms were added to the end and neck of the backbone network to improve the accuracy of object detection. The final function section utilizes Wlou instead of Clou as the loss function, and uses HardSwish activation function instead of SiLU, which improves the comprehensive performance of the model.

引用信息:Yanqiu Wang, Shangtai Jin, Chenkun Yin, Zejin Wu. Target Detection Algorithm Based on Improved YOLOv5 for Edge Computing Devices. International Conference on Robotics, Intelligent Control and Artificial Intelligence (RICAI), 2023-12. DOI: 10.1109/ricai60863.2023.10488977

论文链接:https://doi.org/10.1109/ricai60863.2023.10488977