755483e22a22c7445293ec19fbc4c0487433a5d4,implementations/discogan/datasets.py,ImageDataset,__getitem__,#ImageDataset#,17

Before Change

        item_A = self.transform(Image.open(self.files_A[index % len(self.files_A)]))

        if self.unaligned:
            item_B = self.transform(Image.open(self.files_B[random.randint(0, len(self.files_B) - 1)]))
        else:
            item_B = self.transform(Image.open(self.files_B[index % len(self.files_B)]))

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

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]

Instances