[发明专利]图像语义分割模型的训练方法、可读存储介质及电子设备

摘要

一种图像语义分割的训练方法包括：将预先标注语义分割信息的训练图像输入到图像语义分割模型中，得到包含特征图及预测的语义分割结果；将特征图输入到困难样本挖掘单元中，以计算出训练图像的困难样本；依据预测语义分割结果和预先标注的语义分割信息，统计预测错误的像素，将预测语义分割结果中预测错误的像素作为参考困难样本；依据预测语义分割结果和预先标注的语义分割信息、困难样本和参考困难样本，对基础图像语义分割模型及困难样本挖掘单元的参数进行修正。本发明还提供了一种应用该图像语义分割的训练方法的可读存储介质及电子设备。本发明图像语义分割模型的训练方法、可读存储介质及电子设备可提高语义分割结果的准确率。

主权项

1.一种图像语义分割的训练方法，其特征在于，所述训练方法包括：将预先标注语义分割信息的训练图像输入到图像语义分割模型中，得到包含语义信息的特征图及预测的语义分割结果；将对应所述训练图像的所述特征图输入到困难样本挖掘单元中，以通过所述困难样本挖掘单元计算出所述训练图像的困难样本；依据所述训练图像的所述预测语义分割结果和预先标注的语义分割信息，统计预测错误的像素，将所述预测语义分割结果中预测错误的像素作为参考困难样本；及依据所述训练图像的所述预测语义分割结果和预先标注的语义分割信息、所述计算得到的困难样本和参考困难样本，对所述基础图像语义分割模型及所述困难样本挖掘单元的参数进行修正。