基于密集编码标签的遥感图像旋转目标检测算法

• 作者: 王彦雅 李卫东 张伟娜
• 来源期刊: 河北省科学院学报
• 关键词: 旋转目标检测 遥感图像 环形平滑标签 密集编码标签
• 年，卷（期）: 2023,（1）
• 页码范围: 1-6,35
• 页数: 6页
• 分类号: TP391.4
• 语种: 中文

摘要

针对现有目标检测算法在遥感图像中检测精度低，容易漏检等问题，提出了一种遥感图像旋转目标检测算法，使用YOLOv5m作为基本框架。首先，使用环形平滑标签CSL(Circular Smooth Label，CSL)将角度回归预测转变为角度分类预测，解决回归预测中的角度周期性和边界可交换性的问题，提升检测精度。其次，使用密集编码标签(Densely Coded Label，DCL)替换稀疏编码标签，大幅减少预测层厚度，提升训练速度。实验表明，使用改进后的算法较基准算法mAP提升4.88%，模型训练速度与原模型速度基本相同，证明了算法的有效性。