6f90dac721f38b365ad87e1633860f53b9ca44db,discogan/data_loader.py,DataLoader,load_data,#DataLoader#Any#Any#,10

Before Change

        batch_B = [path_B[i] for i in idx]

        imgs_A, imgs_B = [], []
        for img_path_A, img_path_B in zip(batch_A, batch_B):
            img_A = self.imread(img_path_A)
            img_B = self.imread(img_path_B)

            img_A = scipy.misc.imresize(img_A, self.img_res)
            img_B = scipy.misc.imresize(img_B, self.img_res)

            if not is_testing and np.random.random() > 0.5:
                    img_A = np.fliplr(img_A)
                    img_B = np.fliplr(img_B)

            imgs_A.append(img_A)
            imgs_B.append(img_B)

        imgs_A = np.array(imgs_A)/127.5 - 1.
        imgs_B = np.array(imgs_B)/127.5 - 1.

        return imgs_A, imgs_B

After Change

        for img in batch:
            img = self.imread(img)
            h, w, _ = img.shape
            half_w = int(w/2)
            img_A = img[:, :half_w, :]
            img_B = img[:, half_w:, :]

            img_A = scipy.misc.imresize(img_A, self.img_res)

Instances