f0f56c2e15da58d1d191d936ee85a61021ab2c9d,pytorch/ade20k_dataset.py,ADE20KSegmentation,__getitem__,#ADE20KSegmentation#Any#,53

Before Change

            img, lbl = self.augmentations(img, lbl)

        if self.is_transform:
            img, lbl = self.transform(img, lbl)

        return img, lbl

After Change

        assert (len(self.images) == len(self.masks))

    def __getitem__(self, index):
        img = Image.open(self.images[index]).convert("RGB")
        mask = Image.open(self.masks[index])

        if self.mode == "train":
            img, mask = self._sync_transform(img, mask)

Instances