通过中心点-船头点提取的任意朝向舰船检测
计算机视觉与模式识别
2021-11-29 v3
摘要
遥感图像中的舰船检测在各种应用中起着关键作用，近年来受到越来越多的关注。然而，现有的任意朝向舰船检测方法通常基于一组预定义的旋转锚框发展而来。这些预定义框不仅导致角度预测不准确，还引入了额外的超参数和高计算成本。此外，现有方法未充分利用舰船尺寸的先验知识，阻碍了其检测精度的提升。针对上述问题，本文提出一种基于中心点-船头点提取的检测器（称为 CHPDet），以实现遥感图像中任意朝向舰船检测。我们的 CHPDet 将任意朝向舰船表示为带有用于确定方向的船头点的旋转框，并使用旋转高斯核将标注映射为目标热图。通过关键点估计寻找舰船中心，然后回归舰船尺寸与船头点。还使用了方向不变模型（OIM）来生成方向不变特征图。最后，我们利用目标尺寸作为先验对结果进行微调。此外，我们引入了一个在固定地面采样距离（GSD）下用于遥感图像多类任意朝向舰船检测的新数据集，命名为 FGSD2021。在 FGSD2021 及另外两个广泛使用的数据集 HRSC2016 和 UCAS-AOD 上的实验结果表明，我们的 CHPDet 达到了最先进性能，并能很好区分船首与船尾。代码与 FGSD2021 数据集可在 https://github.com/zf020114/CHPDet 获取。
引用
@article{arxiv.2101.11189,
title = {Arbitrary-Oriented Ship Detection through Center-Head Point Extraction},
author = {Feng Zhang and Xueying Wang and Shilin Zhou and Yingqian Wang and Yi Hou},
journal= {arXiv preprint arXiv:2101.11189},
year = {2021}
}
备注
Accepted by IEEE Transactions on Geoscience and Remote Sensing (TGRS)