Spaces:
Running
on
Zero
Running
on
Zero
JianyuanWang
commited on
Commit
•
27a6ae5
1
Parent(s):
4ef72bc
update
Browse files
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:
|
3 |
-
size
|
|
|
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
|
|