diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/recursive_dataset.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index b14bd66..ecbc7df 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -20,9 +20,9 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = self.opt.render_dir + "/frame_{:05d}.png".format(index) + A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) while not os.path.exists(A_path): - # print(A_path) + print(A_path) time.sleep(0.01) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |