基于注意力机制的单幅图像到三维点云模型重建方法及系统

摘要

本发明提供了一种基于注意力机制的单幅图像到三维点云模型重建方法及系统，对待处理图像进行特征提取，得到图像全局特征矩阵和图像局部特征向量；利用一模型特征转换网络对图像全局特征向量进行处理，得到初始的稀疏点云模型特征，并生成点云模型；根据当前点云模型特征和图像局部特征矩阵，引入注意力机制，计算当前点云模型中每个子区域的图像关联特征矩阵；利用下一模型特征转换网络对当前点云模型特征和对应的图像关联特征矩阵进行处理，生成优化后的点云模型；重复上述优化过程，依次轮询每一个模型特征转换网络，得到最终的稠密点云模型。本发明对于扰动有较好的鲁棒性，在图像生成点云任务中具有良好的性能。