M-ReDet: A mamba-based method for remote sensing ship object detection and fine-grained recognition
by Xuhui Liu, Chi Feng, Shuran Zi, Zhengkun Qin, Qinghe Guan
Ship object detection and fine-grained recognition of remote sensing images are hot topics in remote sensing image processing, with applications in fishing vessel operation command, merchant ship navigation route planning, and other fields. In order to improve the detection accuracy for different types of remote sensing ship objects, this paper proposes a ship object perception and feature refinement method based on the improved ReDet, called Mamba-ReDet (M-ReDet). First, this paper designs a ship object fine-grained feature extraction backbone (Mamba-ReResNet, M-ReResNet), which selects and reconstructs the unique features of different types of ship objects through the Mamba’s selective memory to improve the algorithm’s ability to extract fine-grained features. Secondly, the M-ReDet consists of the Ship Object Perception Module (SOPM) and the Ship Feature Refinement Module (SFRM), which can extract the ship’s spatial position information from the feature map, fuse different scales of spatial position information and use this information to refine the fine-grained features to improve the detection accuracy of the algorithm for different categories of ships. Finally, we use the KFIoU and Focal Loss as the regression loss and classification loss of the algorithm to improve the accuracy of the training. The experimental results show that the mAP0.5 of the M-ReDet algorithm on the FAIR1M(ship) and DOTAv1.0 visible light (RGB) remote sensing image datasets are 43.29% and 82.09%, respectively, which is 2.78% and 3.34% higher than that of the ReDet.