基于视觉Transformer的遥感图像分割方法、装置、介质及设备

本申请公开了基于视觉Transformer的遥感图像分割方法、装置、介质及设备。该方法包括：获取遥感图像数据集，遥感图像数据集包括查询遥感图像与支持遥感图像；将遥感数据集输入到遥感图像分割模型中得到预测的遥感图像分割结果：将查询遥感图像与支持遥感图像输入到可调分支中得到可调分支原型向量；将查询遥感图像与支持遥感图像输入到冻结分支中得到冻结分支原型向量；基于两种原型向量得到预测的遥感图像分割结果；将遥感图像输入到训练好的遥感图像分割模型中，得到遥感图像分割结果；基于遥感图像分割结果与真值标签，进行损失函数的计算，根据损失函数对遥感图像分割模型进行迭代训练，得到训练好的遥感图像分割模型；本申请能够防止过拟合。