Figure 1: PolyDiffuse for floorplan and HD map reconstruction: Starting from an initial proposal (e.g., from a human annotator or an existing method), the sensor-conditioned denoising process of our Guided Set Diffusion Model (GS-DM) generates shape reconstructions in a few sampling steps, initialized and directed by the guidance networks. The initial proposal above mimics simple human inputs that indicate rough locations and specify the number of vertices for the polygonal shapes.

Figure 2: Illustration of the forward and reverse processes of PolyDiffuse with floorplan data.


Download PDF     Arxiv     Supplementary    

     title={PolyDiffuse: Polygonal Shape Reconstruction via Guided Set Diffusion Models},
     author={Chen, Jiacheng and Deng, Ruizhi and Furukawa, Yasutaka},
     journal={arXiv preprint arXiv:2306.01461},

Code / Pre-trained Models

Check out the code and pre-trained models in our Github repo.


This research is partially supported by NSERC Discovery Grants with Accelerator Supplements and DND/NSERC Discovery Grant Supplement, NSERC Alliance Grants, and John R. Evans Leaders Fund (JELF). We thank the Digital Research Alliance of Canada and BC DRI Group for providing computational resources.