From 8969a3ce70d99f301641b8011c6da91fe494c027 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sun, 10 May 2020 10:48:05 -0600
Subject: [PATCH] Add capability to start at arbitrary frames
---
codes/data/LQ_dataset.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/codes/data/LQ_dataset.py b/codes/data/LQ_dataset.py
index 277c1671..535c591a 100644
--- a/codes/data/LQ_dataset.py
+++ b/codes/data/LQ_dataset.py
@@ -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):