JianyuanWang commited on
Commit
27a6ae5
1 Parent(s): 4ef72bc
vggsfm_code/cfgs/demo.yaml CHANGED
@@ -19,6 +19,7 @@ query_frame_num: 3
19
  robust_refine: 2
20
  BA_iters: 2
21
 
 
22
 
23
  load_gt: False
24
  visualize: False
 
19
  robust_refine: 2
20
  BA_iters: 2
21
 
22
+ low_mem: True
23
 
24
  load_gt: False
25
  visualize: False
vggsfm_code/examples/videos/apple_video.mp4 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:53a7f05247a574e0f77926345bb68a3b3c9044adcd6c6432c25f7e2ccc38b38b
3
- size 1846808
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d40b5b4eb3fc3bc4c06df2896a6bb281cf64775966b316a36768c8f6989accc9
3
+ size 1990141
vggsfm_code/vggsfm/models/triangulator.py CHANGED
@@ -325,6 +325,7 @@ class Triangulator(nn.Module):
325
  best_triangulated_points, best_inlier_num, best_inlier_mask = triangulate_tracks(
326
  extrinsics, tracks_normalized_refined, track_vis=pred_vis, track_score=pred_score, max_ransac_iters=128
327
  )
 
328
  # Determine valid tracks based on inlier numbers
329
  valid_tracks = best_inlier_num >= min_valid_track_length
330
  # Perform global bundle adjustment
 
325
  best_triangulated_points, best_inlier_num, best_inlier_mask = triangulate_tracks(
326
  extrinsics, tracks_normalized_refined, track_vis=pred_vis, track_score=pred_score, max_ransac_iters=128
327
  )
328
+
329
  # Determine valid tracks based on inlier numbers
330
  valid_tracks = best_inlier_num >= min_valid_track_length
331
  # Perform global bundle adjustment
vggsfm_code/vggsfm/utils/triangulation_helpers.py CHANGED
@@ -405,7 +405,7 @@ def local_refinement_tri(points1, extrinsics, inlier_mask, sorted_indices, lo_nu
405
  lo_points1 = lo_points1.reshape(B * lo_num, N, -1)
406
  lo_mask = lo_mask.reshape(B * lo_num, N)
407
  lo_extrinsics = extrinsics_expand.reshape(B * lo_num, N, 3, 4)
408
-
409
  # triangulate the inliers
410
  triangulated_points, tri_angles, invalid_che_mask = triangulate_multi_view_point_batched(
411
  lo_extrinsics, lo_points1, mask=lo_mask, compute_tri_angle=True, check_cheirality=True
 
405
  lo_points1 = lo_points1.reshape(B * lo_num, N, -1)
406
  lo_mask = lo_mask.reshape(B * lo_num, N)
407
  lo_extrinsics = extrinsics_expand.reshape(B * lo_num, N, 3, 4)
408
+
409
  # triangulate the inliers
410
  triangulated_points, tri_angles, invalid_che_mask = triangulate_multi_view_point_batched(
411
  lo_extrinsics, lo_points1, mask=lo_mask, compute_tri_angle=True, check_cheirality=True
viz_utils/__pycache__/viz_fn.cpython-310.pyc CHANGED
Binary files a/viz_utils/__pycache__/viz_fn.cpython-310.pyc and b/viz_utils/__pycache__/viz_fn.cpython-310.pyc differ