Laser & Optoelectronics Progress, Volume. 60, Issue 4, 0415008(2023)

Attention Mechanism-Based Improved Lightweight Target Detection Algorithm

Mei Jin, Yihui Li*, Liguo Zhang, and Zijian Ma
Author Affiliations
  • School of Electrical Engineering,Yanshan University, Qinhuangdao 066000, Hebei, China
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Nov. 12, 2021

    Accepted: Jan. 5, 2022

    Published Online: Feb. 13, 2023

    The Author Email: Li Yihui (liyihui819@163.com)

    DOI:10.3788/LOP212947

    Topics