1 files changed, 3 insertions, 2 deletions
diff --git a/test.py b/test.py
index 0a2fb9f..651a164 100644
--- a/test.py
+++ b/test.py
@@ -29,8 +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()
+            # dataset.append(ims)
+            print ims
 
     webpage.save()