search

woodfrog / poly-diffuse

Code for "PolyDiffuse: Polygonal Shape Reconstruction via Guided Set Diffusion Models", NeurIPS 2023
https://poly-diffuse.github.io/
Other
118 stars 9 forks source link

normal map in dataloader #4

Closed jdk9405 closed 9 months ago

jdk9405 commented 9 months ago

Hello, Thank you for sharing the amazing work!

I have a question about the "normal" parameter in the dataloader: https://github.com/woodfrog/poly-diffuse/blob/main/src/datasets/polygon_datasets.py#L50 I noticed that normal maps consist of values of either 0 or 255. How can I preprocess my custom dataset to get the normal map before testing your code?

woodfrog commented 9 months ago

Hi, thanks for your interest. This normal issue seems to be inherited from an early work, MonteFloor, which provides the data pre-processing scripts for the Structured3D dataset. You can refer to this issue in my other repo for reference.

jdk9405 commented 9 months ago

Thank you for the quick response. I'll refer to the link.