遥感图像目标识别模型训练方法、装置、设备及存储介质

AITNT
正文
推荐专利
遥感图像目标识别模型训练方法、装置、设备及存储介质
申请号:CN202410860677
申请日期:2024-06-28
公开号:CN118691976A
公开日期:2024-09-24
类型:发明专利
摘要
本公开的实施例提供了一种遥感图像目标识别模型训练方法、装置、设备及存储介质。该方法包括:获取训练集;构建遥感图像目标识别模型;其中,遥感图像目标识别模型包括编码器、解码器、识别头;解码器具有扩大感受野的背景信息提取模块,用于提取遥感图像中目标的背景信息;在识别头处通过角度敏感的概率交并比损失函数计算预测框与标注框之间的角度交并比损失值,并根据角度交并比损失值更新模型参数;使用训练集对构建的遥感图像目标识别模型进行训练,得到训练完成的遥感图像目标识别模型。以此方式,可以利用遥感图像中目标的背景信息及角度信息增强模型对目标的识别能力,从而训练得到目标识别效果较优的遥感图像目标识别模型。
技术关键词
识别模型训练方法 更新模型参数 解码器 遥感图像数据 识别头 训练集 编码器 Sigmoid函数 海林格距离 全局平均池化 模块 计算机 电子设备 处理器通信 识别装置 识别方法 指令
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号