setting up recursive dataset

1 files changed, 2 insertions, 0 deletions

diff --git a/test.py b/test.py
index 8444bd9..0a2fb9f 100644
--- a/test.py
+++ b/test.py
@@ -30,5 +30,7 @@ if __name__ == '__main__':
         img_path = model.get_image_paths()
         print('%04d: process image... %s' % (i, img_path))
         visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio)
+        if dataset.name() == 'RecursiveImageDataset':
+            dataset.append()
 
     webpage.save()