AssertionError with prepare_data.py about gansformer

[dorarad]

Hi! I believe it's an issue with the files in the data directory you use. I suggest printing the sizes of self.shape and img.shape, as well as the image filename it uses when throwing the exception. And then you can track down the problematic image.

The image files are sorted using python sort() so if they are e.g. 1.png, 2.png, ..., 11.png it's going to sort them into: 1.png, 11.png, 2.png, and so it might be the case that you didn't remove the problematic image.
Also note that when you do find the image that isn't good you may just need to crop/pad/resize it to the right shape rather than deleting it. Good luck and let me know how it goes!

[andrew-alm]

The sorting was throwing me off. I modified the assert statement, and I guess the directory contains different sized images. I had been creating my own TF Data datasets mapping a resize function which made it so I wasn't aware.

Thanks for the help, this can be closed.