bugfix #12
Browse files
mast3r/cloud_opt/sparse_ga.py
CHANGED
@@ -306,14 +306,6 @@ def sparse_scene_optimizer(imgs, subsample, imsizes, pps, base_focals, core_dept
|
|
306 |
for s in imgs_slices:
|
307 |
is_matching_ok[s.img1, s.img2] = matching_check(s.confs)
|
308 |
|
309 |
-
# Subsample preds_21
|
310 |
-
subsamp_preds_21 = {}
|
311 |
-
for imk, imv in preds_21.items():
|
312 |
-
subsamp_preds_21[imk] = {}
|
313 |
-
for im2k, (pred, conf) in preds_21[imk].items():
|
314 |
-
idxs = anchors[imgs.index(im2k)][1]
|
315 |
-
subsamp_preds_21[imk][im2k] = (pred[idxs], conf[idxs]) # anchors subsample
|
316 |
-
|
317 |
# Prepare slices and corres for losses
|
318 |
dust3r_slices = [s for s in imgs_slices if not is_matching_ok[s.img1, s.img2]]
|
319 |
loss3d_slices = [s for s in imgs_slices if is_matching_ok[s.img1, s.img2]]
|
|
|
306 |
for s in imgs_slices:
|
307 |
is_matching_ok[s.img1, s.img2] = matching_check(s.confs)
|
308 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
309 |
# Prepare slices and corres for losses
|
310 |
dust3r_slices = [s for s in imgs_slices if not is_matching_ok[s.img1, s.img2]]
|
311 |
loss3d_slices = [s for s in imgs_slices if is_matching_ok[s.img1, s.img2]]
|