search

SheldonTsui / GOF_NeurIPS2021

The codebase for our paper "Generative Occupancy Fields for 3D Surface-Aware Image Synthesis" (NeurIPS 2021)
Apache License 2.0
102 stars 6 forks source link

HI，I have some questions about the preprocess of CelebA dataset #3

Closed wasdzzzzzz closed 2 years ago

wasdzzzzzz commented 2 years ago

Congrats on this great work, and thanks a lot for open-sourcing the code! In the paper, you mention that “we crop all images in CelebA from the top of the hair to the bottom of the chin as a pre-processing step.” Can you provide specific steps or code scripts for pre-processing methods？ Thanks a lot in advance.

SheldonTsui commented 2 years ago

@JJJLLLTTT This processing has already been included in the datasets.py. Please refer to https://github.com/SheldonTsui/GOF_NeurIPS2021/blob/1e85de85de45acde9b32c41268108b1d936cef03/datasets.py#L24 Note that, the CelebA dataset must be aligned in advance where the image size is 218 x 178.

wasdzzzzzz commented 2 years ago

Thank you very much for your reply~