From 06d78b634d7f05dc74c873a2bc3109dcd73ce68e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 14:15:37 +0200 Subject: k --- data/recursive_dataset.py | 6 ++++-- 1 file 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() -- cgit v1.2.3-70-g09d2