8f420a3b35944dcfa470dee958ef61dee221ce02,implementations/dualgan/datasets.py,MNISTM,__getitem__,#MNISTM#Any#,60

Before Change

        // doing this so that it is consistent with all other datasets
        // to return a PIL Image
        img = Image.fromarray(img.squeeze().numpy(), mode="RGB")

        if self.transform is not None:
            img = self.transform(img)

After Change

        item_A = img_pair[:, :, :half_w]
        item_B = img_pair[:, :, half_w:]

        return {"A": item_A, "B": item_B}

    def __len__(self):
        return len(self.files)

Instances