Laser & Optoelectronics Progress, Volume. 60, Issue 4, 0415008(2023)
Attention Mechanism-Based Improved Lightweight Target Detection Algorithm
A lightweight YOLOv4s based on an attention mechanism is proposed to address the low accuracy and slow speed issue of the general target detection algorithm in multi-target life scenes. First, CSPDarknet53-s was used as the backbone network to extract image features, and these features were selected using the attention block. Subsequently, the feature pyramid network was adopted to fuse the features. Finally, the YOLOv4s head was used to process the two outputs after the feature fusion to improve the multi-target detection ability in living scenes. According to the experiment results, the YOLOv4s algorithm outperforms the prior algorithm in the PASCAL VOC and MS COCO datasets, exhibiting improvement in the mean average precision and average precision. Compared with the lightweight algorithm Efficientdet, the YOLOv4s algorithm also has a certain improvement in the AP on the MS COCO dataset, and achieves effective significant target detection.
Get Citation
Copy Citation Text
Mei Jin, Yihui Li, Liguo Zhang, Zijian Ma. Attention Mechanism-Based Improved Lightweight Target Detection Algorithm[J]. Laser & Optoelectronics Progress, 2023, 60(4): 0415008
Category: Machine Vision
Received: Nov. 12, 2021
Accepted: Jan. 5, 2022
Published Online: Feb. 13, 2023
The Author Email: Li Yihui (liyihui819@163.com)