timothybrooks / instruct-pix2pix

Other
6.41k stars 542 forks source link

Train on non-square aspect ratios? #91

Open andreemic opened 1 year ago

andreemic commented 1 year ago

The original IP2P is trained on 256x256 images. Is it possible to train IP2P on a non-square aspect ratio like 768x512 (3:2)?

There is some hard-coded squareness in edit_dataset.py and the config but does it is it a restriction of the architecture or does it only have to do with the preprocessing code and so on?

Thanks!