3D Contour Generation based on Diffusion Probabilistic Models

The contours of objects effectively represent essential 3D information such as shapes and boundaries. Existing contour detection methods mostly rely on thresholding or neural networks to classify points as edge or non-edge points. However, these methods often lack generalization ability on datasets with different shapes, leading to issues such as missing contours and discontinuous distribution. To address this, we propose a 3D point cloud contour generation method based on the denoising diffusion probabilistic model (DDPM). Our method treats the complete point cloud as an explicit condition to guide the generation of contour from noise. Specifically, our method trains the DDPM to be a conditional generative network customized for contour generation tasks. In the network, we design the Conditional Feature Extraction (CFE) module that obtains multi-scale feature information, and the Conditional Feature Fusion (CFF) module embeds this information in the generation process to guide contour generation. The experimental results demonstrate the effectiveness of our method. The source code of our method is available at: https://github.com/djzgroup/ContourGeneration.