add 'aspect_ratio' in the test code

1 files changed, 1 insertions, 1 deletions

diff --git a/test.py b/test.py
index fc0f1bb..65e79d7 100644
--- a/test.py
+++ b/test.py
@@ -28,6 +28,6 @@ for i, data in enumerate(dataset):
     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)
+    visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio)
 
 webpage.save()