diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index b20b082..8b79d9b 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -2,15 +2,17 @@ import os.path from data.base_dataset import BaseDataset, get_transform from data.image_folder import make_dataset from PIL import Image +import random class NextImage(): def __init__(self): self.next_image = "lol" + self.id = str(random.randint(1,100)) def append(self, path): self.next_image = path - print("APPEND " + self.next_image) + print("APPEND " + self.id + ' ' + self.next_image) def get(self): - print("GET " + self.next_image) + print("GET " + self.id + ' ' + self.next_image) return self.next_image im = NextImage() |