[发明专利]图像语义分割网络的训练方法、装置、设备及存储介质

摘要

本申请公开了一种图像语义分割网络的训练方法、装置、设备及存储介质，涉及图像语义分割领域。该方法包括：根据训练图像集训练第一图像语义分割网络，训练图像集中的训练图像包含标注信息；通过掩膜网络对第一图像语义分割网络的网络权重进行掩膜处理，得到第二图像语义分割网络，掩膜网络用于筛选第一图像语义分割网络中对图像的特征分布不敏感的网络权重；根据训练图像集和测试图像集，对第二图像语义分割网络进行训练，训练图像集与测试图像集的分布不同。本申请实施例中，由于掩膜网络能够过滤对特征分布敏感的网络权重，因此经过网络权重筛选的第二图像语义分割网络对测试图像集也具有更好的泛化性，从而提高对测试图像集的测试准确性。

主权项

1.一种图像语义分割网络的训练方法，其特征在于，所述方法包括：根据训练图像集训练第一图像语义分割网络，所述训练图像集中的训练图像包含标注信息；通过掩膜网络对所述第一图像语义分割网络的网络权重进行掩膜处理，得到第二图像语义分割网络，所述掩膜网络用于筛选所述第一图像语义分割网络中对图像的特征分布不敏感的网络权重；根据所述训练图像集和所述测试图像集，对所述第二图像语义分割网络进行训练，所述测试图像集中的测试图像不包含所述标注信息，且所述训练图像集与所述测试图像集的分布不同。