A Weak Supervision Learning Paradigm for Oriented Ship Detection in SAR Image

Tingxuan Yue, Yanmei Zhang*, Jin Wang, Yanbing Xu, Pengyun Liu

*此作品的通讯作者

科研成果: 期刊稿件文章同行评审

4 引用 (Scopus)

摘要

The advancement of convolutional neural networks (CNNs) has greatly promoted the development of ship detection in SAR images. Oriented ship detection is practical and imperative, but the development of its fully-supervised methods is limited by the insufficiency of data annotated in an oriented bounding box (OBB). Therefore, we proposed a weakly supervised learning (WSL) paradigm to train the network by only data annotated in a horizontal bounding box (HBB) for oriented ship detection. The paradigm follows a coarse-to-fine prediction route, and the ships’ coarse orientations and high-quality pseudolabels both mined from images are used to train the network combined with the HBB annotations. The designed orientation initialization decoder (OID) utilizes the output from the prediction branch of ship coarse orientation (PBSCO) and the regression branch to decode coarse OBBs, based on the horizontal proposals, and then these OBBs are refined by the fine prediction head, which is jointly trained by high-quality pseudolabels and HBB annotation. Extensive experiments were conducted on the SAR ship detection dataset (SSDD) and the high-resolution SAR images dataset (HRSID) with both OBB and HBB annotations, and the results show that our proposed WSL paradigm can train network to attain the AP50 and R in the same level of mainstream fully-supervised methods in remote sensing.

源语言英语
文章编号5207812
页(从-至)1-12
页数12
期刊IEEE Transactions on Geoscience and Remote Sensing
62
DOI
出版状态已出版 - 2024

指纹

探究 'A Weak Supervision Learning Paradigm for Oriented Ship Detection in SAR Image' 的科研主题。它们共同构成独一无二的指纹。

引用此