demo/demo_globals.py

import sys
sys.path.append('wild-gaussian-splatting/mast3r/')

import os
import tempfile
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'
tmpdirname = tempfile.TemporaryDirectory(suffix='demo')
chkpt_tag = hash_md5(weights_path)
CACHE_PATH = os.path.join(tmpdirname.name, chkpt_tag)
os.makedirs(CACHE_PATH, exist_ok=True)

DEVICE = device = 'cuda' if torch.cuda.is_available() else 'cpu'
MODEL = AsymmetricMASt3R.from_pretrained(weights_path).to(DEVICE)
SILENT = False