一种用于高斯基元补全的扩散模型的训练方法及装置

本申请实施例公开了一种用于高斯基元补全的扩散模型的训练方法，方法包括：获取训练样本，训练样本为包括目标物体的图像样本；利用图像样本获得目标物体对应的结构化高斯基元样本；对结构化高斯基元样本进行采样，获得三维局部块，将三维局部块对应的区域在目标物体样本的所述三维网格空间中去除，得到输入样本；获得扩散模型基于输入样本生成的针对输入样本的输出高斯基元；训练的目标包括：最小化针对输入样本的输出高斯基元与结构化高斯基元样本之间的差异。本申请通过对结构化高斯基元进行采样和去除的方式获得输入样本，由扩散模型对局部缺失区域进行补全，使生成的高斯基元能够更好地拟合实际的三维形状，实现更高质量的补全。