update visualizer

yiranran · Apr 20, 2020 · 1f3b920 · 1f3b920
1 parent 5b0c56a
commit 1f3b920
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/render-to-video/util/visualizer.py b/render-to-video/util/visualizer.py
@@ -11,6 +11,7 @@
 # save image to the disk
 def save_images(webpage, visuals, image_path, aspect_ratio=1.0, width=256):
     image_dir = webpage.get_image_dir()
+    short_path = ntpath.basename(image_path[0])
     short_path1 = ntpath.basename(ntpath.dirname(image_path[0]))
     short_path = short_path1 + '-' + short_path
     name = os.path.splitext(short_path)[0]