diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 19:37:23 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 19:37:23 +0200 |
| commit | 92223bd44b891536bf0423f9b7f43c29414331c6 (patch) | |
| tree | 1404d5b5a4ee0bd3dba81d32b0ab20027abcecec | |
| parent | f49d7f6397d9e7c3e256ce01cc2dd33b84714987 (diff) | |
k
| -rw-r--r-- | data/recursive_dataset.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index b4135dc..b51777f 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -26,7 +26,6 @@ class RecursiveDataset(BaseDataset): print('initialize') self.opt = opt self.root = opt.dataroot - im.append(opt.dataroot + 'frame_0000.png') self.dir_A = os.path.join(opt.dataroot) self.A_paths = make_dataset(self.dir_A) self.A_paths = sorted(self.A_paths) @@ -36,7 +35,7 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = "recursive/frame_{:04d}.png".format(index) + A_path = "/home/lens/code/pytorch-CycleGAN-and-pix2pix/recursive/frame_{:04d}.png".format(index) print('next image: ' + A_path) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |