diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-15 02:47:42 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-15 02:47:42 +0200 |
| commit | 6caef752e3be24037024c83ea3c2f4d3de3c7f37 (patch) | |
| tree | d9f27eb89ef90530f754c5bf9ccc8bd926383ee3 /data | |
| parent | f19f765a93ffe3ac72d43eaa37993c2037449bba (diff) | |
test
Diffstat (limited to 'data')
| -rw-r--r-- | data/recursive_dataset.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 5cc3e4a..b1e0407 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -20,8 +20,9 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = self.opt.results_dir + self.opt.tag + "/frame_{:05d}.png".format(index) + A_path = self.opt.render_dir + self.opt.tag + "/frame_{:05d}.png".format(index) while not os.path.exists(A_path): + print(A_path) time.sleep(0.01) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |