Laser & Optoelectronics Progress, Volume. 60, Issue 22, 2212008(2023)

Infrared Ship Detection Using Attention Mechanism and Multiscale Fusion

Shen Zhang1, Lin Hu1,2, Xiang'e Sun1,2、*, and Meihua Liu1,2
Author Affiliations
  • 1School of Electronic Information, Yangtze University, Jingzhou 434023, Hubei , China
  • 2Intelligence Research Institute, Yangtze University, Jingzhou 434023, Hubei , China
  • show less

    To address the issue of inadequate accuracy and real-time performance of infrared ship target detection methods on coastal defense scenarios, a novel lightweight ship detection algorithm based on improved YOLOv7 framework is proposed. This framework incorporates several enhancements to augment its capabilities. First, to achieve model lightweight processing, the algorithm integrates the MobileNetv3 network into the architecture of the Backbone network. This addition contributes to efficient computation and model size reduction.Second, an attention mechanism is introduced within the Neck network to mitigate noise and interference, thereby improving the network's feature extraction capability. In addition, we employ a bidirectional weighted feature pyramid to enhance feature fusion within the network, promoting more effective information integration. Finally, the algorithm incorporates Wise IoU to optimize the loss function, improving convergence speed and model accuracy. Experimental evaluations on the Arrow dataset demonstrate noteworthy improvements over the standard YOLOv7 approach. Specifically, the proposed enhanced algorithm exhibits a 0.9 percentage points increase in accuracy, 2.5 percentage points increase in recall, and 1.2 percentage points increase in mean average precision (mAP) at IoU thresholds of 0.5 and 0.5∶0.95. In addition, it achieves approximately 38.4% reduction in model parameters and a 65.5% reduction in floating point operations per second (FLOPs). This enhanced algorithm delivers superior inspection accuracy while meeting the speed requirements for efficient ship inspection. Consequently, it effectively enables high-speed and high-precision ship detection.

    Tools

    Get Citation

    Copy Citation Text

    Shen Zhang, Lin Hu, Xiang'e Sun, Meihua Liu. Infrared Ship Detection Using Attention Mechanism and Multiscale Fusion[J]. Laser & Optoelectronics Progress, 2023, 60(22): 2212008

    Download Citation

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

    Category: Instrumentation, Measurement and Metrology

    Received: Jun. 5, 2023

    Accepted: Jul. 24, 2023

    Published Online: Nov. 6, 2023

    The Author Email: Sun Xiang'e (xinges2000@yangtzeu.edu.cn)

    DOI:10.3788/LOP231462

    Topics