9dcac66250b2d26d9b2505ad04841f3f4b8bc2a1,utils/datasets.py,ImageFolder,__getitem__,#ImageFolder#,39

Before Change

    def __getitem__(self, index):
        img_path = self.files[index % len(self.files)]
        // Extract image
        img = lycon.load(img_path)
        img, _ = pad_to_square(img, 127)
        // Resize
        img = lycon.resize(img, height=self.img_size, width=self.img_size, interpolation=lycon.Interpolation.NEAREST)
        // Channels-first
        img = np.transpose(img, (2, 0, 1))
        // As pytorch tensor
        img = torch.from_numpy(img).float() / 255.0

        return img_path, img

After Change

    def __getitem__(self, index):
        img_path = self.files[index % len(self.files)]
        // Extract image as PyTorch tensor
        img = transforms.ToTensor()(Image.open(img_path))
        // Pad to square resolution
        img, _ = pad_to_square(img, 0)
        // Resize
        img = resize(img, self.img_size)

Instances