From f43e17d69c7121ff658dcca4aecf22d7a10c94f0 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 13:39:13 +0200 Subject: k --- data/__init__.py | 2 +- test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/data/__init__.py b/data/__init__.py index 6da79f3..cd868fc 100644 --- a/data/__init__.py +++ b/data/__init__.py @@ -64,7 +64,7 @@ class CustomDatasetDataLoader(BaseDataLoader): class RecursiveDatasetDataLoader(BaseDataLoader): def name(self): - return 'CustomDatasetDataLoader' + return 'RecursiveDatasetDataLoader' def initialize(self, opt): #BaseDataLoader.initialize(self, opt) diff --git a/test.py b/test.py index 53a1d4b..ad42b2d 100644 --- a/test.py +++ b/test.py @@ -32,7 +32,7 @@ if __name__ == '__main__': print('%04d: process image... %s' % (i, img_path)) ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio) print(dataset.name()) - if dataset.name() == 'RecursiveImageDataset': + if dataset.name() == 'RecursiveDatasetDataLoader': # dataset.append(ims) print(ims) -- cgit v1.2.3-70-g09d2