b591a7aecce5b8de42a0dd7bba62780675d34fce,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

    def __len__(self):

After Change

        // Pad to square resolution
        img, _ = pad_to_square(img, 0)
        // Resize
        img = resize(img.unsqueeze(0), self.img_size).squeeze(0)

        return img_path, img

Instances