0134a038d242a57e765de9f7288b974ff72646f0,codes/data/LRHR_dataset.py,LRHRDataset,__init__,#LRHRDataset#,19

Before Change

        elif self.data_type == "lmdb":
            keys = pickle.load(open(osp.join(opt["dataroot_GT"], "_keys_cache.p"), "rb"))
            self.paths_GT = sorted([key for key in keys if not key.endswith(".meta")])
            keys = pickle.load(open(osp.join(opt["dataroot_LR"], "_keys_cache.p"), "rb"))
            self.paths_LR = sorted([key for key in keys if not key.endswith(".meta")])
        assert self.paths_GT, "Error: GT path is empty."
        if self.paths_LR and self.paths_GT:
            assert len(self.paths_LR) == len(self.paths_GT), \
                "GT and LR datasets have different number of images - {}, {}.".format(\

After Change

        self.opt = opt
        self.data_type = self.opt["data_type"]
        self.paths_LR, self.paths_GT = None, None
        self.sizes_LR, self.sizes_GT = None, None
        self.LR_env, self.GT_env = None, None  // environment for lmdb

        self.paths_GT, self.sizes_GT = util.get_image_paths(self.data_type, opt["dataroot_GT"])

Instances