paths

1 files changed, 12 insertions, 0 deletions

diff --git a/crop-equirectangular.py b/crop-equirectangular.py
index 47a56cd..16f19da 100644
--- a/crop-equirectangular.py
+++ b/crop-equirectangular.py
@@ -1,6 +1,7 @@
 import os
 import glob
 import argparse
+from shutil import rmtree
 from PIL import Image
 
 # This script generates crops with a specific aspect ratio from a 360 video.
@@ -44,6 +45,17 @@ path_0 = os.path.join("sequences", opt.label + "_a")
 path_1 = os.path.join("sequences", opt.label + "_b")
 path_2 = os.path.join("sequences", opt.label + "_c")
 
+if os.path.exists(path_0):
+  rmtree(path_0)
+if os.path.exists(path_1):
+  rmtree(path_1)
+if os.path.exists(path_2):
+  rmtree(path_2)
+
+os.makedirs(path_0)
+os.makedirs(path_1)
+os.makedirs(path_2)
+
 for i, fn in enumerate(glob.glob(os.path.join(opt.folder, '*.png'))):
   fn = "frame_{:04d}.png".format(i)