Click here to Skip to main content
15,867,834 members
Please Sign up or sign in to vote.
1.00/5 (1 vote)
See more:
Python
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)


What I have tried:

Python
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)
Posted
Comments
Richard Deeming 24-Feb-23 4:29am    
You seem to have forgotten to ask a question.

This content, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)



CodeProject, 20 Bay Street, 11th Floor Toronto, Ontario, Canada M5J 2N8 +1 (416) 849-8900