NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation

Home Page:https://arxiv.org/abs/2006.06676

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AssertionError : list(image.shape) == self.image_shape

ku60 opened this issue · comments

commented

I intended to train network.
I prepared training data with png format and pass the folder path.
first time, it went well, but after I added new images, program says "AssertionError" in training/dataset.py, line88 in __getitem__ : assert list(image.shape) == self.image_shape
But the shape of images I passed are all the same (128,128), I checked and the first data is the same size.
Anyone helps me ?

Hello, for training the model on your custom dataset, the images should have the following format:
Resolution: power of 2 (1024x1024, 512x512, 256x256, 128x128 .....)
Format: .png
image.channel==3, which means only RGB images are supported.
Please remove any grayscale(channel=1) or RGBA(channel=4) images from your dataset or convert them into RGB format.
I hope this will solve your problem...