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 non-maximum suppression (NMS). 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.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.