You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hello
I'm trying to train WGAN in order to increase the amount of images (3211 img) that I have from my own data-set. The size of my images are 1242x375, but only for testing purposes of this net I am resizing them to 256x256.
To add my own data-set, I just added this code lines to the main.py file:
Transformaciones is the data augmentation module that i have build.
I launch the training with 6 extra layers and 800 epoch.
The thing is that generator net only produces noise. I tried to use the generator net when the results were like lossG: 0.08 approx, and I only had noise in the images.
I am doing something wrong? Is because I have only a few images to train?
Someone has tried to use an own data-set and managed to made it work?
Thanks!
The text was updated successfully, but these errors were encountered:
Hello
I'm trying to train WGAN in order to increase the amount of images (3211 img) that I have from my own data-set. The size of my images are 1242x375, but only for testing purposes of this net I am resizing them to 256x256.
To add my own data-set, I just added this code lines to the main.py file:
elif opt.dataset == 'own': normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225]) dataset = dset.ImageFolder(root=opt.dataroot, transform=transforms.Compose([ transformaciones.Transformaciones(), transforms.RandomHorizontalFlip(0.1), transforms.Resize((opt.imageSize,opt.imageSize)), transforms.ToTensor(), normalize, ]))
Transformaciones is the data augmentation module that i have build.
I launch the training with 6 extra layers and 800 epoch.
The thing is that generator net only produces noise. I tried to use the generator net when the results were like lossG: 0.08 approx, and I only had noise in the images.
I am doing something wrong? Is because I have only a few images to train?
Someone has tried to use an own data-set and managed to made it work?
Thanks!
The text was updated successfully, but these errors were encountered: