Chinese Journal of Liquid Crystals and Displays, Volume. 36, Issue 10, 1437(2021)

Lightweight SSD object detection method based on feature fusion

WU Tian-cheng1、*, WANG Xiao-quan1,2, CAI Yi-jun1, JING You-bo2, and CHEN Cheng-ying1
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    An improved scheme based on feature fusion for the lightweight object detection algorithm deployed in embedded devices is proposed in this paper. The algorithm is based on the MobileNetV2-SSD lightweight object detection algorithm with the idea of FPN feature fusion. It combines the three feature layers of MobileNetV2-SSD that contain more semantic information, and is regenerated based on the fused feature layer Feature pyramid for object detection. The improved model and the original model are compared and tested on the PASCAL VOC data set. Compared with the original model, the mAP is increased by 3.6%, and reaches 76.5%. At the same time, the detection effect of small targets is significantly improved. Finally, it is tested in the embedded device Jetson AGX Xavier. The improved model has a detection speed close to the original model when the network structure is more complex, reaching the detection speed of 22FPS, which can realize real-time object detection on embedded devices.

    Tools

    Get Citation

    Copy Citation Text

    WU Tian-cheng, WANG Xiao-quan, CAI Yi-jun, JING You-bo, CHEN Cheng-ying. Lightweight SSD object detection method based on feature fusion[J]. Chinese Journal of Liquid Crystals and Displays, 2021, 36(10): 1437

    Download Citation

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

    Category:

    Received: Jan. 7, 2021

    Accepted: --

    Published Online: Nov. 6, 2021

    The Author Email: WU Tian-cheng (787568968@qq.com)

    DOI:10.37188/cjlcd.2021-0007

    Topics