push

1 files changed, 5 insertions, 1 deletions

diff --git a/test.py b/test.py
index 849a8db..33edb61 100644
--- a/test.py
+++ b/test.py
@@ -9,7 +9,7 @@ from shutil import move, copyfile
 from PIL import Image, ImageOps
 from skimage.transform import resize
 from scipy.misc import imresize
-from shutil import copyfile
+from shutil import copyfile, rmtree
 import numpy as np
 
 import subprocess
@@ -25,6 +25,7 @@ if __name__ == '__main__':
 
     render_dir = opt.results_dir + opt.name + "/exp:" + opt.experiment + "/"
 
+    rmtree(render_dir)
     mkdirs(render_dir)
     copyfile(opt.start_img, render_dir + "frame_00000.png")
 
@@ -80,6 +81,9 @@ if __name__ == '__main__':
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()
 
+    print("________")
+    print("\n")
+    
     cmd = ("scp", render_dir + opt.name + "_" + opt.experiment + ".mp4", "jules@asdf.us:asdf/neural/")
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()