def __getitem__(self, idx):
    image_root = self.image_roots[idx]
    image = Image.open(image_root)
    image = image.convert('RGB')
    if (self.transform is not None):
        image = self.transform(image)
    label = self.labels[idx]
    if (self.target_transform is not None):
        label = self.target_transform(label)
    return (image, label, image_root)
def __getitem__(self, idx):
    image_root = self.image_roots[idx]
    image = Image.open(image_root)
    image = image.convert('RGB')
    if (self.transform is not None):
        image = self.transform(image)
    label = self.labels[idx]
    if (self.target_transform is not None):
        label = self.target_transform(label)
    return (image, label, image_root)