new scripts using canny

1 files changed, 2 insertions, 1 deletions

diff --git a/test.py b/test.py
index a53932d..1f82875 100644
--- a/test.py
+++ b/test.py
@@ -25,7 +25,8 @@ if __name__ == '__main__':
 
     render_dir = opt.results_dir + opt.name + "/exp:" + opt.experiment + "/"
 
-    rmtree(render_dir)
+    if os.path.exists(render_dir):
+      rmtree(render_dir)
     mkdirs(render_dir)
     copyfile(opt.start_img, render_dir + "frame_00000.png")