__getitem__.py

def __getitem__(self, idx):

    img = np.array(Image.open(os.path.join(self.db_root_dir, self.img_list[idx])).convert("RGB"), dtype=np.float32)
    gt = np.array(Image.open(os.path.join(self.db_root_dir, self.labels[idx])).convert("L"), dtype=np.float32)

    gt = ((gt/np.max([gt.max(), 1e-8])) > 0.5).astype(np.float32)
    #gt = gt.astype(np.bool).astype(np.float32)

    if self.transform is not None:

        augmentations = self.transform(image=img, mask=gt)
        img = augmentations["image"]
        gt = augmentations["mask"]

    if self.pad_mirroring:
        img = Pad(padding=self.pad_mirroring, padding_mode="reflect")(img)

    return img, gt