def __getitem__(self, idx):
    image = Image.open(os.path.join(self.path_to_images, self.df.index[idx]))
    image = image.convert('RGB')
    label = np.zeros(len(self.PRED_LABEL), dtype=int)
    for i in range(0, len(self.PRED_LABEL)):
        if (self.df[self.PRED_LABEL[i].strip()].iloc[idx].astype('int') > 0):
            label[i] = self.df[self.PRED_LABEL[i].strip()].iloc[idx].astype('int')
    if self.transform:
        image = self.transform(image)
    return (image, label, self.df.index[idx])
def __getitem__(self, idx):
    image = Image.open(os.path.join(self.path_to_images, self.df.index[idx]))
    image = image.convert('RGB')
    label = np.zeros(len(self.PRED_LABEL), dtype=int)
    for i in range(0, len(self.PRED_LABEL)):
        if (self.df[self.PRED_LABEL[i].strip()].iloc[idx].astype('int') > 0):
            label[i] = self.df[self.PRED_LABEL[i].strip()].iloc[idx].astype('int')
    if self.transform:
        image = self.transform(image)
    return (image, label, self.df.index[idx])