2024年度  第9期


标题:基于高斯分布损失的遥感图像旋转目标检测
作者:朱玉鹏 张文涛 孙山林 杜浩
作者单位:桂林电子科技大学电子工程与自动化学院,广西 桂林 541004
关键字:深度学习;遥感图像;高斯分布;旋转目标检测
摘要:针对遥感图像目标检测物体长宽比例大、方向任意等特点,一般检测器存在检测精度不高的问题,提出一种基于YOLOv5的遥感图像旋转目标检测,使用旋转回归损失KLD Loss,把旋转矩形包络框转换为高斯分布,提供了更加准确的回归预测,避免了边界问题。实现结果表明,所提算法在DOTA数据集检测精度(mAP50)达到了77.97%,相比基础模型YOLOv5m提升了5.75个百分点。实验证明该算法在高精度的遥感图像旋转目标检测的有效性。