针对现有深度学习模型体量大,难以在机载边缘硬件环境上直接部署并流畅开展推理的问题,本文提出了一种轻量化的无人机遥感影像多目标检测模型。该模型利用分组卷积核与通道混排结构构建了特征提取结构的基本层,使用轻型通道注意力机制...针对现有深度学习模型体量大,难以在机载边缘硬件环境上直接部署并流畅开展推理的问题,本文提出了一种轻量化的无人机遥感影像多目标检测模型。该模型利用分组卷积核与通道混排结构构建了特征提取结构的基本层,使用轻型通道注意力机制加强模型对实际特征的聚焦程度,使用双层特征强化结构输出多尺度的特征图,通过改进的非极大值抑制算法实现目标框的筛选输出,并利用多源无人机影像数据集对模型进行训练和验证。试验结果表明,本文所提出模型在测试数据集上平均精度均值可以达到88.91%,较其余三组对照模型分别提高了7.07%、16.64%和11.56%。训练后模型仅有42.5 MB,在测试环境下的每秒检测张数(Frames Per Second,FPS)可以达到57,证明本文提出的模型能够在有限的内存及计算资源硬件条件下快速且精准地完成检测任务。展开更多
文摘针对现有深度学习模型体量大,难以在机载边缘硬件环境上直接部署并流畅开展推理的问题,本文提出了一种轻量化的无人机遥感影像多目标检测模型。该模型利用分组卷积核与通道混排结构构建了特征提取结构的基本层,使用轻型通道注意力机制加强模型对实际特征的聚焦程度,使用双层特征强化结构输出多尺度的特征图,通过改进的非极大值抑制算法实现目标框的筛选输出,并利用多源无人机影像数据集对模型进行训练和验证。试验结果表明,本文所提出模型在测试数据集上平均精度均值可以达到88.91%,较其余三组对照模型分别提高了7.07%、16.64%和11.56%。训练后模型仅有42.5 MB,在测试环境下的每秒检测张数(Frames Per Second,FPS)可以达到57,证明本文提出的模型能够在有限的内存及计算资源硬件条件下快速且精准地完成检测任务。