Spaces:
Configuration error
Configuration error
fix init={} being modified inplace
Browse files
mast3r/cloud_opt/sparse_ga.py
CHANGED
@@ -14,6 +14,7 @@ import os
|
|
14 |
from collections import namedtuple
|
15 |
from functools import lru_cache
|
16 |
from scipy import sparse as sp
|
|
|
17 |
|
18 |
from mast3r.utils.misc import mkdir_for, hash_md5
|
19 |
from mast3r.cloud_opt.utils.losses import gamma_loss
|
@@ -166,7 +167,7 @@ def sparse_scene_optimizer(imgs, subsample, imsizes, pps, base_focals, core_dept
|
|
166 |
init={}, device='cuda', dtype=torch.float32,
|
167 |
matching_conf_thr=5., loss_dust3r_w=0.01,
|
168 |
verbose=True, dbg=()):
|
169 |
-
|
170 |
# extrinsic parameters
|
171 |
vec0001 = torch.tensor((0, 0, 0, 1), dtype=dtype, device=device)
|
172 |
quats = [nn.Parameter(vec0001.clone()) for _ in range(len(imgs))]
|
|
|
14 |
from collections import namedtuple
|
15 |
from functools import lru_cache
|
16 |
from scipy import sparse as sp
|
17 |
+
import copy
|
18 |
|
19 |
from mast3r.utils.misc import mkdir_for, hash_md5
|
20 |
from mast3r.cloud_opt.utils.losses import gamma_loss
|
|
|
167 |
init={}, device='cuda', dtype=torch.float32,
|
168 |
matching_conf_thr=5., loss_dust3r_w=0.01,
|
169 |
verbose=True, dbg=()):
|
170 |
+
init = copy.deepcopy(init)
|
171 |
# extrinsic parameters
|
172 |
vec0001 = torch.tensor((0, 0, 0, 1), dtype=dtype, device=device)
|
173 |
quats = [nn.Parameter(vec0001.clone()) for _ in range(len(imgs))]
|