71c559e66eaff89089fa447a3d7e04b6835356cf,dataset.py,BaseDataset,img_transform,#BaseDataset#Any#,56

Before Change

    def img_transform(self, img):
        // image to float
        img = img.astype(np.float32)
        img = img.transpose((2, 0, 1))
        img = self.normalize(torch.from_numpy(img.copy()))
        return img

After Change

    def img_transform(self, img):
        // 0-255 to 0-1
        img = np.float32(np.array(img)) / 255.
        img = img.transpose((2, 0, 1))
        img = self.normalize(torch.from_numpy(img.copy()))
        return img

Instances