Fix cache_dir error

app.py

@@ -28,7 +28,7 @@ if __name__ == '__main__':
             with gr.Tab("MASt3R Demo"):
                 mast3r_demo_tab()
             with gr.Tab("Gaussian Splatting Demo"):
-                gs_demo_tab(cache_path)
+                gs_demo_tab()
 
         demo.launch(show_error=True, share=None, server_name=None, server_port=None) 
         # demo.launch(show_error=True, share=None, server_name='0.0.0.0', server_port=5555) 

demo/demo_globals.py

@@ -7,8 +7,9 @@ import torch
 from mast3r.utils.misc import hash_md5
 from mast3r.model import AsymmetricMASt3R
 
+DATASET_DIR = "colmap_data"
 weights_path = "naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"
-# weights_path = '/app/wild-gaussian-splatting/mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth'
+weights_path = '/app/wild-gaussian-splatting/mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth'
 tmpdirname = tempfile.TemporaryDirectory(suffix='demo')
 chkpt_tag = hash_md5(weights_path)
 CACHE_PATH = os.path.join(tmpdirname.name, chkpt_tag)

demo/gs_train.py

@@ -8,7 +8,7 @@ import gradio as gr
 import importlib.util
 from dataclasses import dataclass, field
 
-# import spaces
+import spaces
 
 
 @dataclass
@@ -60,7 +60,7 @@ class TrainingArgs:
     checkpoint_iterations: list[int] = field(default_factory=lambda: [7_000, 15_000, 30_000])
     start_checkpoint: str = None
 
-# @spaces.GPU(duration=20)
+@spaces.GPU(duration=20)
 def train(
     data_source_path, sh_degree, model_path, images, resolution, white_background, data_device, eval,
     convert_SHs_python, compute_cov3D_python, debug,

demo/mast3r_demo.py

@@ -34,7 +34,7 @@ import matplotlib.pyplot as pl
 import torch
 
 
-from demo_globals import CACHE_PATH, MODEL, DEVICE, SILENT
+from demo_globals import CACHE_PATH, MODEL, DEVICE, SILENT, DATASET_DIR
 
 class SparseGAState():
     def __init__(self, cache_dir=None, outfile_name=None):
@@ -219,12 +219,12 @@ def get_reconstructed_scene(image_size, current_scene_state,
                                     opt_depth='depth' in optim_level, shared_intrinsics=shared_intrinsics,
                                     matching_conf_thr=matching_conf_thr, **kw)
 
-    base_colmapdata_dir = os.path.join(CACHE_PATH, 'colmap_data')
+    base_colmapdata_dir = os.path.join(CACHE_PATH, DATASET_DIR)
     os.makedirs(base_colmapdata_dir, exist_ok=True)
     colmap_data_dir = get_next_dir(base_colmapdata_dir)
     # 
     save_colmap_scene(scene, colmap_data_dir, min_conf_thr, clean_depth)
-    
+
     if current_scene_state is not None and \
        current_scene_state.outfile_name is not None:
         outfile_name = current_scene_state.outfile_name