From fd3198c0c799e7943f7f27758e97670535c94979 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 2 Sep 2018 22:09:49 +0200 Subject: augment script --- data/recursive_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'data/recursive_dataset.py') diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 40b7ebf..dda9c95 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -13,7 +13,7 @@ class RecursiveDataset(BaseDataset): ### input A (label maps) self.dataset_size = 1000000 - + def __getitem__(self, index): ### input A (label maps) A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) -- cgit v1.2.3-70-g09d2