a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Update LQ_dataset to support inference on split image videos

Browse Source
This commit is contained in:
James Betker 2020-05-23 21:05:49 -06:00
parent 74bb0fad33
commit 90073fc761
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
codes/data/LQ_dataset.py
View File

@ -33,14 +33,15 @@ class LQDataset(data.Dataset):
def __getitem__(self, index): def __getitem__(self, index):
if self.data_type == 'lmdb' and self.LQ_env is None: if self.data_type == 'lmdb' and self.LQ_env is None:
self._init_lmdb() self._init_lmdb()
actual_index = int(index / 2) actual_index = index # int(index / 2)
is_left = (index % 2) == 0 is_left = (index % 2) == 0
# get LQ image # get LQ image
LQ_path = self.paths_LQ[actual_index] LQ_path = self.paths_LQ[actual_index]
img_LQ = Image.open(LQ_path) img_LQ = Image.open(LQ_path)
left = 0 if is_left else 2000 left = 0 if is_left else 1920
img_LQ = F.crop(img_LQ, 74, left + 74, 1900, 1900) # crop input if needed.
#img_LQ = F.crop(img_LQ, 5, left + 5, 1900, 1900)
img_LQ = F.to_tensor(img_LQ) img_LQ = F.to_tensor(img_LQ)
img_name = osp.splitext(osp.basename(LQ_path))[0] img_name = osp.splitext(osp.basename(LQ_path))[0]
@ -49,4 +50,4 @@ class LQDataset(data.Dataset):
return {'LQ': img_LQ, 'LQ_path': LQ_path} return {'LQ': img_LQ, 'LQ_path': LQ_path}
def __len__(self): def __len__(self):
return len(self.paths_LQ) * 2 return len(self.paths_LQ) # * 2