[发明专利]一种训练图像语义分割模型的方法

摘要

本发明提供一种训练图像语义分割模型的方法。该方法包括：预先训练知识蒸馏的教师模型并固定；在知识蒸馏的教师模型和学生模型之间嵌入目标类掩膜函数和非目标类掩膜函数，分别结合教师模型和学生模型的中间特征图、输出图，构建对应的目标类掩膜特征和非目标类掩膜特征；基于构建的目标类掩膜特征、非目标类掩膜特征的KL散度构建知识蒸馏的损失函数；基于交叉熵损失函数和所述知识蒸馏的损失函数构建训练学生模型的损失函数对所述学生模型进行训练以获得图像语义分割模型。通过本发明提供的技术方案使知识蒸馏过程中轻量级的学生模型更关注目标类区域与非目标类区域两种知识，由此得到图像语义分割模型具有速度快、准确率高的优点。