summaryrefslogtreecommitdiff
path: root/data/recursive_dataset.py
diff options
context:
space:
mode:
Diffstat (limited to 'data/recursive_dataset.py')
-rw-r--r--data/recursive_dataset.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index eea4ea8..998a96c 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -8,7 +8,7 @@ class RecursiveDataset(BaseDataset):
def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
- self.next_image = opt.dataroot
+ self.next_image = opt.dataroot + 'frame_0001.png'
self.dir_A = os.path.join(opt.dataroot)
self.A_paths = make_dataset(self.dir_A)
self.A_paths = sorted(self.A_paths)