基于高阶空间交互增强的轻量级模型用于光学遥感图像小舰船检测
计算机视觉与模式识别
2023-04-11 v1
摘要
准确可靠的光学遥感图像小舰船检测对海上监视系统至关重要,但现有方法常难以平衡检测性能与计算复杂度。本文提出一种基于高阶空间交互的轻量级框架 \textit{HSI-ShipDetectionNet},适用于卫星和无人机等资源受限平台部署。HSI-ShipDetectionNet 包含专为微小舰船设计的预测分支以及用于降低复杂度的轻量级混合注意力模块。此外,高阶空间交互模块的使用提升了高级特征理解与建模能力。我们使用公开 Kaggle 海洋舰船检测数据集评估模型,并与包括小目标检测模型、轻量级检测模型和舰船检测模型在内的多个 SOTA 模型比较。结果表明,HSI-ShipDetectionNet 在召回率和平均精度均值(mAP)上优于其他模型,且轻量并适合部署于资源受限平台。
引用
@article{arxiv.2304.03812,
title = {High-order Spatial Interactions Enhanced Lightweight Model for Optical Remote Sensing Image-based Small Ship Detection},
author = {Yifan Yin and Xu Cheng and Fan Shi and Xiufeng Liu and Huan Huo and Shengyong Chen},
journal= {arXiv preprint arXiv:2304.03812},
year = {2023}
}