一种基于条件扩散模型的桌面层物体生成方法

本发明公开了一种基于条件扩散模型的桌面层物体生成方法，该方法首先获取桌面场景数据，包括不同视角的图片和家具点云数据，并进行预处理。其次利用预处理后训练数据集中的不同视角的图片，以及对应的语义标签，通过预训练模型提取桌面风格语义特征；利用预处理后训练数据集中的家具点云数据，提取几何特征与约束特征，得到表面点云和边界点云。然后采用U型网络构建扩散模型，整合桌面风格语义特征、表面点云和边界点云，得到生成桌面层的模型。最后利用训练得到的扩散模型，选择不同的场景，推理生成对应场景下的桌面摆设，实现桌面层物体布局生成。本发明能基于不同的桌面场景准确生成对应的物件摆设。