6917d8a3dbf6617f02b1247fd50fbc20d3cb6fd4,codes/data/LR_dataset.py,LRDataset,__getitem__,#LRDataset#Any#,29

Before Change

        // get LR image
        LR_path = self.paths_LR[index]
        if self.opt["data_type"] == "img":
            img_LR = cv2.imread(LR_path, cv2.IMREAD_UNCHANGED)
        else:
            img_LR = util.read_lmdb_img(self.LR_env, LR_path)
        img_LR = img_LR.astype(np.float32) / 255.
        if img_LR.ndim == 2:
            img_LR = np.expand_dims(img_LR, axis=2)
        H, W, C = img_LR.shape

        // channel conversion
        if self.opt["color"]:
            img_LR = util.channel_convert(C, self.opt["color"], [img_LR])[0]

        // HWC to CHW, BGR to RGB, numpy to tensor
        if img_LR.shape[2] == 3:
            img_LR = cv2.cvtColor(img_LR, cv2.COLOR_BGR2RGB)
        img_LR = torch.from_numpy(np.ascontiguousarray(np.transpose(img_LR, (2, 0, 1)))).float()

        return {"LR": img_LR, "LR_path": LR_path}

After Change

        // HWC to CHW, BGR to RGB, numpy to tensor
        if img_LR.shape[2] == 3:
            img_LR = img_LR[:, :, [2, 1, 0]]
        img_LR = torch.from_numpy(np.ascontiguousarray(np.transpose(img_LR, (2, 0, 1)))).float()

        return {"LR": img_LR, "LR_path": LR_path}

Instances