get_data

Module Contents

customDataset

get_data(config)
sample_return(root)

get_data(config)

class customDataset(root, transform=None)

Bases: torch.utils.data.Dataset

__getitem__(index)

Retrieves a sample from the dataset at the given index. Args:

index (int): Index of the sample to retrieve.

Returns:

img (PIL.Image): The image data. label (int): The label for the image data.

__len__()

sample_return(root)