对比一致性学习的遥感图像语义分割模型训练方法及装置
摘要:
本申请公开了一种对比一致性学习的遥感图像语义分割模型训练方法及装置,涉及遥感图像分割技术领域。具体实现方案为:构建包含学生网络和教师网络的遥感图像语义分割网络,所述遥感图像语义分割网络为UNet语义分割网络;根据遥感图像数据集对所述遥感图像语义分割网络进行训练;将实时拍摄的遥感图像输入训练完成的遥感图像语义分割网络,并输出所述遥感图像的语义分割结果。本申请实施例通过学生网络和教师网络进行训练,以对遥感图像进行语义分割,获取各像素的类别。本申请实施例可以用少量的标记数据与大量的无标记数据半监督训练,就能达到用大量标记数据全监督训练的效果,减少对标签数据的需求,减少时间与人工成本。
0/0