Arbitrary-Oriented Ship Detection Based on RetinaNet for Remote Sensing Images
Mingming Zhu, Guoping Hu, Hao Zhou, Shiqiang Wang, Yule Zhang, Shijie Yue, Yu Bai, Kexin Zang
Abstract
Aiming to address the problems of arbitrary orientations, large aspect ratios, and dense arrangements in ship detection, an arbitrary-oriented ship detection method based on RetinaNet is proposed. Our proposed method includes a rotated RetinaNet, a refined network, a feature alignment module, and an improved loss function. First, the rotated RetinaNet achieves rotation detection by using a feature pyramid network, rotated anchors, the skew intersection-over-union (IoU), and skew nonmaximum suppression. Then, the refined network and feature alignment module are introduced to achieve better detection accuracy. Finally, to address the boundary discontinuity, the loss function is improved by introducing the IoU constant factor. Considering the problems with the HRSC2016 dataset, we establish a new dataset with more accurate labels and more images and object samples. Through an ablation study, we thoroughly analyze the validity of the proposed rotated RetinaNet, feature alignment module, and improved loss function. The experimental results show that our method is superior to other state-of-the-art methods.