一种基于多层特征交互金字塔和轻量化增强检测头的遥感图像全方位目标检测方法

本发明公开了一种基于多层特征交互金字塔和轻量化增强检测头的遥感图像全方位目标检测算法，属于计算机视觉领域。所述方法包括如下步骤：1.对遥感图像数据集进行预处理。2.搭建遥感图像全方位目标检测模型：设计了多层特征交互金字塔，通过聚合多层特征图获得中间特征图和融合特征图，实现跨层的特征融合，避免信息交互局限在相邻层之间，同时生成旋转不变的特征表示，增强旋转目标的特征信息；构建了轻量化增强检测头，采用共享增强卷积的策略减少模型的参数量，并通过中心差分卷积提取特征信息，让模型能捕捉到旋转目标的细节特征；采用Lamp剪枝方法在不损失精度的基础上降低模型的复杂度。3.构建模型的损失函数，引入KLD散度损失函数解决旋转目标检测中角度周期性变化的问题。4.迭代训练模型，直到模型达到收敛，得到最佳权重。5.将得到的最佳权重对测试集进行测试，获得评估结果。所述方法在保证旋转目标检测精度的同时减少了模型的参数量和计算复杂度。