From 07fa49cff8f20bb8e3b7dd785464e79e21e0dd8e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 12:36:44 +0200 Subject: return im paths from visualizer --- test.py | 5 +++-- util/visualizer.py | 1 + 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 -- cgit v1.2.3-70-g09d2