return im paths from visualizer

2 files changed, 4 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()
diff --git a/util/visualizer.py b/util/visualizer.py
index a98b512..35ea0be 100644
--- a/util/visualizer.py
+++ b/util/visualizer.py
@@ -148,3 +148,4 @@ class Visualizer():
             txts.append(label)
             links.append(image_name)
         webpage.add_images(ims, txts, links, width=self.win_size)
+        return ims