一种基于注意力机制的模型训练方法和点云分割方法

本说明书公开了一种基于注意力机制的模型训练方法和点云分割方法。所述方法包括：获取训练样本，其中，训练样本中包含有支持集点云数据和查询集点云数据；将训练样本输入语义分割模型，通过特征提取网络，确定训练样本对应的特征数据；对特征数据进行转置处理，得到转置特征数据，通过第一注意力网络，对转置特征数据进行加权处理，得到第一加权特征，通过第二注意力网络，对特征数据进行加权处理，得到第二加权特征；根据第一加权特征和第二加权特征，确定目标特征；根据支持集点云数据对应的目标特征与查询集点云数据对应的目标特征，对查询集点云数据进行语义分割，根据语义分割结果和其标签之间的偏差，确定损失值并对进行模型训练。