paul-corbalan/wasserstein-gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Save each image in its own file.

Browse Source
This commit is contained in:
kopytjuk 2018-12-26 16:03:28 +01:00
parent ef94e5ae17
commit ccb22bee22
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
generate.py
View File

@ -56,7 +56,8 @@ if __name__=="__main__":
netG.cuda() netG.cuda()
fixed_noise = fixed_noise.cuda() fixed_noise = fixed_noise.cuda()
fake = netG(Variable(fixed_noise, volatile=True)) fake = netG(fixed_noise)
fake.data = fake.data.mul(0.5).add(0.5) fake.data = fake.data.mul(0.5).add(0.5)
vutils.save_image(fake.data, os.path.join(opt.output_dir, "generated.png")) for i in range(opt.nimages):
vutils.save_image(fake.data[i, ...].reshape((1, nc, imageSize, imageSize)), os.path.join(opt.output_dir, "generated_%02d.png"%i))