diff options
| -rw-r--r-- | test.py | 5 | ||||
| -rw-r--r-- | util/visualizer.py | 1 |
2 files changed, 4 insertions, 2 deletions
@@ -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 |