The ship classification and detection method of optical remote sensing image based on improved YOLOv7-tiny

Abstract

In view of the fault and leak detection problems caused by complex scenes of offshore area in remote sensing image ship detection, a lightweight ship classification detection method is proposed based on improved YOLOv7-tiny. On the one hand, this method stacks a lightweight feature extraction module and applies it to the backbone feature extraction network, which significantly reduces the parameter and computational complexity and does not weaken the network's ability of feature extraction. On the other hand, this method introduces spatial information into the feature pyramid, raising the discrimination of features at different scales, to improve the classification and detection ability of the network. This method has been tested on the remote sensing image ship data set. The experimental results show that the average accuracy of ship classification detection based on the improved network is increased by 2.9%. Meanwhile, the parameter quantity and computational complexity are better than YOLOv7-tiny, with a 15% reduction in parameter quantity and a 24% reduction in computational complexity.