Breaking News: Grepper is joining You.com. Read the official announcement!

generate image dataset pytorch

Innocent Iguana answered on May 30, 2023 Popularity 7/10 Helpfulness 3/10

answer generate image dataset pytorch

generate image dataset pytorch

Comment

# Create custom dataset objectfor image
class Dataset(Dataset):
  
    def __init__(self, csv_file, data_dir, transform=None):   # Constructor        
        self.data_dir=data_dir         # Image directory        
        self.transform = transform # The transform is goint to be used on image
        data_dircsv_file=os.path.join(self.data_dir,csv_file)
        self.data_name= pd.read_csv(data_dircsv_file)    # Load the CSV file contians image info    
        self.len=self.data_name.shape[0] # Number of images in dataset
    
    def __len__(self): # Get the length
        return self.len
    
    def __getitem__(self, idx):   # Getter 
        img_name=os.path.join(self.data_dir,self.data_name.iloc[idx, 1]) # Image file path
        image = Image.open(img_name)        # Open image file
        y = self.data_name.iloc[idx, 0]     # The class label for the image    
        if self.transform: # If there is any transform method, apply it onto the image
            image = self.transform(image)
        return image, y # getter will return both image and label

# Create the dataset objects
dataset = Dataset(csv_file=csv_file, data_dir=directory)

image=dataset[0][0] # First image in dataset
y=dataset[0][1] # First image's label in dataset

import torchvision.transforms as transforms
# Combine two transforms: crop and convert to tensor. Apply the compose to MNIST dataset
croptensor_data_transform = transforms.Compose([transforms.CenterCrop(20), transforms.ToTensor()])
dataset = Dataset(csv_file=csv_file , data_dir=directory,transform=croptensor_data_transform )
print("The shape of the first element tensor: ", dataset[0][0].shape)

def show_data(data_sample, shape = (28, 28)): # Helper func to show image
    plt.imshow(data_sample[0].numpy().reshape(shape), cmap='gray')
    plt.title('y = ' + data_sample[1])
    
show_data(dataset[0],shape = (20, 20)) # Plot the first element in the dataset

# Construct the compose for multiple transformations back to back.
fliptensor_data_transform = transforms.Compose([transforms.RandomVerticalFlip(p=1),transforms.ToTensor()])
dataset = Dataset(csv_file=csv_file , data_dir=directory,transform=fliptensor_data_transform )
show_data(dataset[1])

Popularity 7/10 Helpfulness 3/10 Language python

Source: Grepper

Tags: dataset generate image python pytorch

Link to this answer
Share Copy Link

Contributed on May 30 2023

Innocent Iguana

0 Answers Avg Quality 2/10

generate image dataset pytorch

Contents

More Related Answers

generate image dataset pytorch

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.