Applied Sciences, Vol. 13, Pages 1746: A Lightweight YOLOv5 Optimization of Coordinate Attention

1 year ago 25

Applied Sciences, Vol. 13, Pages 1746: A Lightweight YOLOv5 Optimization of Coordinate Attention

Applied Sciences doi: 10.3390/app13031746

Authors: Jun Wu Jiaming Dong Wanyu Nie Zhiwei Ye

As Machine Learning technologies evolve, there is a desire to add vision capabilities to all devices within the IoT in order to enable a wider range of artificial intelligence. However, for most mobile devices, their computing power and storage space are affected by factors such as cost and the tight supply of relevant chips, making it impossible to effectively deploy complex network models to small processors with limited resources and to perform efficient real-time detection. In this paper, YOLOv5 is studied to achieve the goal of lightweight devices by reducing the number of original network channels. Then detection accuracy is guaranteed by adding a detection head and CA attention mechanism. The YOLOv5-RC model proposed in this paper is 30% smaller and lighter than YOLOv5s, but still maintains good detection accuracy. YOLOv5-RC network models can achieve a good balance between detection accuracy and detection speed, with potential for its widespread use in industry.

Read Entire Article