question about dataset size

switchablenorms / CelebAMask-HQ

A large-scale face dataset for face parsing, recognition, generation and editing.

2.09k stars 348 forks source link

Open ax-2020 opened 4 years ago

ax-2020 commented 4 years ago

Hello, glad to find this project. I have a question about dataset size. How to set up a model to load images of different sizes?

steven413d commented 4 years ago

Hi, please refer to face_parsing/data_loader.py.

ax-2020 commented 4 years ago

Do I only need to modify this file? Because my dataset has different image sizes, I want to generate different mask sizes, thanks

ax-2020 commented 4 years ago

Hi，Can the size of the mask be the same as the size of the original image, rather than just a square shape？