Update main_test_SUNet.py

main_test_SUNet.py

@@ -44,7 +44,10 @@ def main():
 
     inp_dir = args.input_dir
     out_dir = args.result_dir
-
+    
+    if os.path.exists(out_dir):
+        shutil.rmtree(out_dir)
+    
     os.makedirs(out_dir, exist_ok=True)
 
     files = natsorted(glob.glob(os.path.join(inp_dir, '*')))