一种高分辨率遥感图像实例分割模型的训练方法及装置

基本信息

申请号 CN202110987535.2 申请日 -
公开(公告)号 CN114155247A 公开(公告)日 2022-03-08
申请公布号 CN114155247A 申请公布日 2022-03-08
分类号 G06T7/10(2017.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 张鹏;章一骥;马天舒;尹鹏飞;潘婷;张瑞祥;杨文 申请(专利权)人 航天恒星科技有限公司
代理机构 中国航天科技专利中心 代理人 臧春喜
地址 100086北京市海淀区知春路82号院
法律状态 -

摘要

摘要 本发明公开了一种高分辨率遥感图像实例分割模型的训练方法及装置。包括:获取有标签的第一遥感图像和无标签的第二遥感图像;将第一遥感图像和第二遥感图像输入至初始高分辨率遥感图像实例分割模型;基于监督分支网络对第一遥感图像进行处理,并根据处理结果得到监督分支网络的监督损失函数;基于一致性分支网络对第一遥感图像和第二遥感图像进行处理,并根据处理结果得到一致性分支网络的一致性损失函数;基于监督损失函数和一致性损失函数,计算得到模型损失函数;根据模型损失函数调整初始高分辨率遥感图像实例分割模型的模型参数,并迭代上述过程,直至得到目标高分辨率遥感图像实例分割模型。本发明可以有效地利用未标注数据提高模型分割精度。