yocabon commited on
Commit
0e7bc28
·
unverified ·
1 Parent(s): 9e25e47

Update demo.py

Browse files
Files changed (1) hide show
  1. mast3r/demo.py +1 -0
mast3r/demo.py CHANGED
@@ -170,6 +170,7 @@ def get_reconstructed_scene(outdir, gradio_delete_cache, model, device, silent,
170
  cache_dir = tempfile.mkdtemp(suffix='_cache', dir=outdir)
171
  else:
172
  cache_dir = os.path.join(outdir, 'cache')
 
173
  scene = sparse_global_alignment(filelist, pairs, cache_dir,
174
  model, lr1=lr1, niter1=niter1, lr2=lr2, niter2=niter2, device=device,
175
  opt_depth='depth' in optim_level, shared_intrinsics=shared_intrinsics,
 
170
  cache_dir = tempfile.mkdtemp(suffix='_cache', dir=outdir)
171
  else:
172
  cache_dir = os.path.join(outdir, 'cache')
173
+ os.makedirs(cache_dir, exist_ok=True)
174
  scene = sparse_global_alignment(filelist, pairs, cache_dir,
175
  model, lr1=lr1, niter1=niter1, lr2=lr2, niter2=niter2, device=device,
176
  opt_depth='depth' in optim_level, shared_intrinsics=shared_intrinsics,