Custom dataset with pytorch

[olegkorshunov]

Sometimes we have dataset with different formats for exampl in hdf5

and it would be nice to be able to write your own dataset that would return a picture and a label

from torch.utils.data import Dataset

class CustomImageClassificationDataset(Dataset):
    def __init__(self): pass
    def __len__(self): pass
    def __getitem__(self, idx):
        ...
        return image, label