一种语义分割模型训练方法及装置

本发明涉及一种语义分割模型训练方法及装置，属于数字图像技术领域，该方法包括获取街道上的图像数据，设置初始语义分割模型；根据图像数据对初始语义分割模型进行训练，得到目标语义分割模型和输出结果；根据初始掩膜、伪标签和关键边缘区域集合进行权重计算，得到模型总损失；当模型总损失大于预设总损失阈值时，根据伪标签对图像数据进行扩充，并根据扩充之后的图像数据对目标语义分割模型进行训练，直至模型总损失不大于预设总损失阈值，确定模型训练完成；本发明提供了初始语义分割模型，可以对每张图像进行处理，无需手动对每张图像进行标记和处理；还可以对输出结果进行计算判断，保证了语义分割模型了准确率，减少了人工成本。