• Laser & Optoelectronics Progress
  • Vol. 59, Issue 16, 1628008 (2022)
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
    DOI: 10.3788/LOP202259.1628008 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.