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

    def __getitem__(self, index):

        img_pair = self.transform(Image.open(self.files[index % len(self.files)]))
        _, h, w = img_pair.shape
        half_w = int(w/2)

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

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

    def __len__(self):

Instances