summaryrefslogtreecommitdiff
path: root/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test.py')
-rw-r--r--test.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/test.py b/test.py
index 8444bd9..0a2fb9f 100644
--- a/test.py
+++ b/test.py
@@ -30,5 +30,7 @@ if __name__ == '__main__':
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)
+ if dataset.name() == 'RecursiveImageDataset':
+ dataset.append()
webpage.save()