基于交叉领域类别感知的领域适应语义分割方法及装置

摘要

本发明实施例提供一种基于交叉领域类别感知的领域适应语义分割方法及装置，该方法包括：将源图像风格转换为目标图像风格后，分别进行特征提取及分类；将特征图和分类分数图输入交叉域类别感知模块；分别通过两个交叉域类别感知器的交叉域类别中心生成器对特征图类别中心进行调整使特征图的类别中心靠近；分别通过类别注意力模块对特征图的分类模糊特征点进行调整，得到第一注意力特征图和第二注意力特征图进而进行语义分割。本发明实施例模型在提取某一领域特征时关注另一领域数据特征的类别中心，并结合注意力机制，重点对两个领域中分类模糊的像素点特征进行调整，使不同领域同种类特征的类别中心一致，减小特征分布的差异，实现领域适应。