How to use my own dataset about pytorch-generative-model-collections

[ArghyaPal]

Hi,

open the dataloader.py file
and, add

elif dataset == 'AWA':
data_loader = DataLoader(
datasets.ImageFolder('data/AWA', transform=transform),
batch_size=batch_size, shuffle=True)

then, open main.py
and modify this line:
parser.add_argument('--dataset', type=str, default='mnist', choices=[ 'mnist', 'fashion-mnist', 'cifar10', 'cifar100', 'svhn', 'stl10', 'lsun-bed'],

to, parser.add_argument('--dataset', type=str, default='mnist', choices=['AWA', 'mnist', 'fashion-mnist', 'cifar10', 'cifar100', 'svhn', 'stl10', 'lsun-bed'],

[HamzahNizami]

Do the images have to be in a specific format or size.
Thanks in advance

[HamzahNizami]

Hi,

open the dataloader.py file
and, add

elif dataset == 'AWA':
data_loader = DataLoader(
datasets.ImageFolder('data/AWA', transform=transform),
batch_size=batch_size, shuffle=True)

then, open main.py
and modify this line:
parser.add_argument('--dataset', type=str, default='mnist', choices=[ 'mnist', 'fashion-mnist', 'cifar10', 'cifar100', 'svhn', 'stl10', 'lsun-bed'],

to, parser.add_argument('--dataset', type=str, default='mnist', choices=['AWA', 'mnist', 'fashion-mnist', 'cifar10', 'cifar100', 'svhn', 'stl10', 'lsun-bed'],

Do the images have to be in a specific format or size.
Thanks in advance

[ArghyaPal]

No, I don't think so. Please go ahead and use your dataset.. please reply back, if you encounter any problem related to image size

[HamzahNizami]

Okay I will try soon.

I have a dataset of my own JPG images. These images are of strawberries. The image I want to generate is of strawberries.

My images are in colour, will this affect training on this GitHub. Do I need to change to gray-scale.

Finally @ArghyaPal have you successfully trained this model on your dataset? Any help would be greatly appreciated.

Thanks in advance!

[HamzahNizami]

No, I don't think so. Please go ahead and use your dataset.. please reply back, if you encounter any problem related to image size

@ArghyaPal

Hi, I'm currently getting the following error.

Even though my images are jpg as shown in the image below. Any help would be appreicated.

[HamzahNizami]

I managed to fix this issue by placing them in another folder called 1, so my directory is now, data/strawberriesBlack/1/B(1).jpg

However, my results are completely blank after 50 epochs on GAN

My dataset is currently coloured, do you think I should change to grayscale?
How big should said dataset actually be?

Thanks in advance

[evdelph]

@ArghyaPal I am having issues with either size or color scaling with these models, . I am trying to run color images of ships through GAN but I get shape errors (see issue 25). Is there a fix to this?