数字创新中心

Center for Digital Innovation

Direct Oriented Ship Localization Regression in Remote Sensing Imagery with Curriculum Learning

Weiwei Guo, Huiyuan Chen, Zenghui Zhang, YanHua Zhang, Wenxian Yu

2021 IEEE International Geoscience and Remote Sensing Symposium IGARSS

摘要/Abstract

由于尺度、方向和分布的巨大变化,遥感图像中准确有效的船舶检测仍然是一项具有挑战性的任务。在本文中,我们提出了一种无锚船舶检测器,可以直接回归船舶定位参数,提供比以前的方法更简单的管道。然后以多任务方式训练检测网络,其中不仅包含船舶中心点图和定向边界框,还包含船舶掩模。我们没有固定多个任务损失之间的权重,而是采用课程学习策略,在训练过程中逐渐调整损失权重,以便网络可以在早期阶段学习有判别性的船舶特征,并在训练继续时获得更多的定位信息。真实数据集上的实验结果证明了我们提出的方法的有效性和效率。

Accurate and efficient ship detection in remote sensing images still remains a challenging task due to the large variations of scales, orientations and distributions. In this paper, we propose an anchor-free ship detector that directly regresses ship localization parameters, offering a simpler pipeline over the previous methods. The detection network is then trained in a multi-task fashion which contains not only the ship center-point maps and oriented bounding boxes but the ship masks. Instead of fixing the weights among the multiple task losses, we adopt a curriculum learning strategy which gradually adapts the loss weights during the training process so that the network can learn the discriminative ship features at the early stage and obtain more localization information while training continues. Experimental results on real dataset demonstrate the effectiveness and efficiency of our proposed method.

相关信息/Info

作者/Authors

链接/Link

Weiwei Guo, Huiyuan Chen, Zenghui Zhang, YanHua Zhang, Wenxian Yu

https://ieeexplore.ieee.org/abstract/document/9554670

图片/Figures