摘要
本公开的实施例提供了一种遥感图像目标识别模型训练方法、装置、设备及存储介质。该方法包括:获取训练集;构建遥感图像目标识别模型;其中,遥感图像目标识别模型包括编码器、解码器、识别头;解码器具有扩大感受野的背景信息提取模块,用于提取遥感图像中目标的背景信息;在识别头处通过角度敏感的概率交并比损失函数计算预测框与标注框之间的角度交并比损失值,并根据角度交并比损失值更新模型参数;使用训练集对构建的遥感图像目标识别模型进行训练,得到训练完成的遥感图像目标识别模型。以此方式,可以利用遥感图像中目标的背景信息及角度信息增强模型对目标的识别能力,从而训练得到目标识别效果较优的遥感图像目标识别模型。
技术关键词
识别模型训练方法
更新模型参数
解码器
遥感图像数据
识别头
训练集
编码器
Sigmoid函数
海林格距离
全局平均池化
模块
计算机
电子设备
处理器通信
识别装置
识别方法
指令