search

NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation
https://arxiv.org/abs/2006.06676
Other
4.13k stars 1.17k forks source link

Non-square Images? #289

Open researchhappy opened 1 year ago

researchhappy commented 1 year ago

Simple Question: Is there a way to input non-square and non-rectangular images? For example, if I am training the model on an apple, I want to only crop out the apple (no padding, no nothing) and train the model on only that particular part of the generation.

PrashantDixit0 commented 1 year ago

@researchhappy Yes you can input rectangular inputs but you have specify width and height parameter while preparing dataset using dataset_tool.py.

There is another way while using dataset_tool.py for data preperation set --cropsize parameter