1 files changed, 4 insertions, 1 deletions
diff --git a/test.py b/test.py
index 8444bd9..651a164 100644
--- a/test.py
+++ b/test.py
@@ -29,6 +29,9 @@ if __name__ == '__main__':
         visuals = model.get_current_visuals()
         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)
+        ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio)
+        if dataset.name() == 'RecursiveImageDataset':
+            # dataset.append(ims)
+            print ims
 
     webpage.save()