Skip to content

Commit 0345b8f

Browse files
batch triangulation function
1 parent d65f347 commit 0345b8f

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

eks/multicam_smoother.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -754,9 +754,10 @@ def triangulate_3d_models(marker_array, camgroup) -> np.ndarray:
754754
tri = np.zeros((M, K, T, 3), dtype=float)
755755
for m in range(M):
756756
for k in range(K):
757-
for t in range(T):
758-
xy_views = [raw[m, c, t, k, :2] for c in range(C)]
759-
tri[m, k, t] = camgroup.triangulate(np.array(xy_views), fast=True)
757+
# Batch all T frames together: shape (C, T, 2)
758+
xy_views = raw[m, :, :, k, :2] # (C, T, 2)
759+
# triangulate expects (C, N, 2) and returns (N, 3)
760+
tri[m, k, :, :] = camgroup.triangulate(xy_views, fast=True)
760761
return tri
761762

762763

0 commit comments

Comments
 (0)