关键词:
机器学习
损失函数
遥感图像
目标检测
注意力机制
摘要:
针对遥感图像目标检测中的复杂背景干扰,目标密集堆积导致遥感图像目标检测模型效果差的问题,通过对YOLOv5s目标检测模型进行改进.首先使用混合注意力机制对CBAM (convolutional block attention module)加以改进并添加至骨干网络中,使得模型所提取到的特征含有局部信息和全局信息,以加强模型识别复杂背景下目标的能力;其次使用超轻量级的动态上采样器DySample结构,减少模型参数并提高检测效率,最后使用EIoU损失函数提高对于待检测目标的定位水平.在RSOD和DIOR数据集上进行了实验验证,结果表明,改进后的YOLOv5s在遥感图像目标检测中的准确度要比原始模型高7.8%,同时能够满足遥感图像实时目标检测的需求;此外与其他目标检测模型相比,改进模型也能保有一定优势.