mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add capability to start at arbitrary frames

Browse Source
This commit is contained in:
James Betker 2020-05-10 10:48:05 -06:00
parent 03351182be
commit 8969a3ce70
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/data/LQ_dataset.py
View File

@ -12,10 +12,15 @@ class LQDataset(data.Dataset):
super(LQDataset, self).__init__()
self.opt = opt
self.data_type = self.opt['data_type']
if 'start_at' in self.opt.keys():
self.start_at = self.opt['start_at']
else:
self.start_at = 0
self.paths_LQ, self.paths_GT = None, None
self.LQ_env = None # environment for lmdb
self.paths_LQ, self.sizes_LQ = util.get_image_paths(self.data_type, opt['dataroot_LQ'])
self.paths_LQ = self.paths_LQ[self.start_at:]
assert self.paths_LQ, 'Error: LQ paths are empty.'
def _init_lmdb(self):