Fix --optimize-disk-space

stages/run_opensfm.py

@@ -35,18 +35,20 @@ class ODMOpenSfMStage(types.ODM_Stage):
         octx.extract_cameras(tree.path("cameras.json"), self.rerun())
         self.update_progress(70)
 
-        if args.optimize_disk_space:
+        def cleanup_disk_space():
-            for folder in ["features", "matches", "exif", "reports"]:
+            if args.optimize_disk_space:
-                folder_path = octx.path(folder)
+                for folder in ["features", "matches", "exif", "reports"]:
-                if os.path.islink(folder_path):
+                    folder_path = octx.path(folder)
-                    os.unlink(folder_path)
+                    if os.path.islink(folder_path):
-                else:
+                        os.unlink(folder_path)
-                    shutil.rmtree(folder_path)
+                    else:
+                        shutil.rmtree(folder_path)
 
         # If we find a special flag file for split/merge we stop right here
         if os.path.exists(octx.path("split_merge_stop_at_reconstruction.txt")):
             log.ODM_INFO("Stopping OpenSfM early because we found: %s" % octx.path("split_merge_stop_at_reconstruction.txt"))
             self.next_stage = None
+            cleanup_disk_space()
             return
 
         updated_config_flag_file = octx.path('updated_config.txt')
@@ -205,6 +207,8 @@ class ODMOpenSfMStage(types.ODM_Stage):
         if not args.skip_report:
             octx.export_stats(self.rerun())
 
+        cleanup_disk_space()
+
         if args.optimize_disk_space:
             os.remove(octx.path("tracks.csv"))
             if io.file_exists(octx.recon_backup_file()):