Laser & Optoelectronics Progress, Volume. 59, Issue 16, 1628008(2022)

Ship Classification and Detection Method for Optical Remote Sensing Images Based on Improved YOLOv5s

Qikai Zhou1, Wei Zhang1, Dongjin Li2, and Fu Niu1、*
Author Affiliations
  • 1Academy of Systems Engineering of Academy of Military Science of Chinese PLA, Beijing 100071, China
  • 2Beijing Institute of Control and Electronic Technology, Beijing 100038, China
  • show less

    Owing to the fault and leak detection problems caused by complex scenes and diverse scales in remote sensing image ship detection, a lightweight ship classification detection method based on improved YOLOv5s is proposed herein to realize real-time rapid ship classification and detection despite limited equipment computing capability. This method applies a lightweight and efficient channel attention technique to the backbone feature extraction network to obtain a novel feature extraction network with an improved ability to identify ships in complex remote sensing images. The feature maps with different levels obtained from the feature extraction network were input into the weighted bidirectional feature pyramid structure to optimize the fusion of high and low stage features of the backbone network, and experiments were conducted on the ship dataset of remote sensing images. The results show that the mean average precision of the improved network model has increased from 83.9% to 89.2% and the average precision for detecting aircraft carriers, warships, civil ships, and submarines has increased by 1.6 percentage points, 0.9 percentage points, 8.8 percentage points, and 9.5 percentage points, respectively. Additionally, the average detection speed and network complexity are considerably better than the other algorithms.

    Tools

    Get Citation

    Copy Citation Text

    Qikai Zhou, Wei Zhang, Dongjin Li, Fu Niu. Ship Classification and Detection Method for Optical Remote Sensing Images Based on Improved YOLOv5s[J]. Laser & Optoelectronics Progress, 2022, 59(16): 1628008

    Download Citation

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

    Category: Remote Sensing and Sensors

    Received: Dec. 29, 2021

    Accepted: Feb. 25, 2022

    Published Online: Aug. 8, 2022

    The Author Email: Niu Fu (niufu@vip.sina.com)

    DOI:10.3788/LOP202259.1628008

    Topics