基于半监督学习的轻量级遥感图像分割方法
摘要:
本发明公开基于半监督学习的轻量级遥感图像分割方法,包括S1、收集原始图像数据集,标注部分数据集以得到部分标注的训练数据集;S2、构建两个完全一致的语义分割模型;S3、将无标签数据同时输入两个语义分割模型,计算输出的一致性损失;S4、将有标签数据同时输入两个语义分割模型,计算交叉熵损失和骰子损失的混合损失作为有监督损失;S5、总损失为一致性损失和两个模型的混合损失三者按权重相加;S6、对总体损失进行优化和反向梯度传播,更新网络参数;S7、在同一测试集分别测试训练好的两个模型,选择其中精度较高的一个模型作为最终模型。本发明可提高遥感图像分割的效率,在较低的参数量和计算复杂度下实现了更高的精度。
公开/授权文献
0/0