diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-04-19 02:11:42 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-04-19 02:11:42 +0200 |
| commit | b32f538d539fcd47696f8ae3bcf6cb64ba5b75ca (patch) | |
| tree | f9532879ed5ed374cb412874df7a8d9d8b48a717 /data/recursive_dataset.py | |
| parent | 92223bd44b891536bf0423f9b7f43c29414331c6 (diff) | |
more general
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index b51777f..4bd0798 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -35,7 +35,7 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = "/home/lens/code/pytorch-CycleGAN-and-pix2pix/recursive/frame_{:04d}.png".format(index) + A_path = self.opt.results_dir + "/" + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index) print('next image: ' + A_path) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |