import os import torch from torchvision.utils import save_image sample_dir = '/content/drive/MyDrive/computer_vision/image_anotation/generated_mask' if not os.path.exists(sample_dir): os.makedirs(sample_dir) if(epoch == n_epochs-1 ): fake_image_and_labels = 'fake_image_and_labels-{0:0=4d}.png'.format(i) #save_image(fake, os.path.join(sample_dir, fake_fname), nrow=8) save_image(fake[i], os.path.join(sample_dir, fake_image_and_labels), nrow=1)
var
This content, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)