# Dataset with labels
def __getitem__(self, item):
    x, y = get_image(item), get_label(item)
    return x, y

# Dataset without labels
def __getitem__(self, item):
    x = get_image(item)
    return x