Laser & Optoelectronics Progress, Volume. 59, Issue 16, 1628008(2022)
Ship Classification and Detection Method for Optical Remote Sensing Images Based on Improved YOLOv5s
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.
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
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)