updated dimensions order in poses_to_napari_tracks conversion function

niksirbi · niksirbi · commit 6322cfbea80c · 2025-04-07T14:25:00.000+01:00
diff --git a/movement/napari/convert.py b/movement/napari/convert.py
@@ -58,7 +58,7 @@ def poses_to_napari_tracks(ds: xr.Dataset) -> tuple[np.ndarray, pd.DataFrame]:
     n_tracks = n_individuals * n_keypoints
     # Construct the napari Tracks array
     # Reorder axes to (individuals, keypoints, frames, xy)
-    yx_cols = np.transpose(ds.position.values, (1, 2, 0, 3)).reshape(-1, 2)[
+    yx_cols = np.transpose(ds.position.values, (3, 2, 0, 1)).reshape(-1, 2)[
         :, [1, 0]  # swap x and y columns
     ]
     # Each keypoint of each individual is a separate track

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ def poses_to_napari_tracks(ds: xr.Dataset) -> tuple[np.ndarray, pd.DataFrame]:`
`58`	`58`	`n_tracks = n_individuals * n_keypoints`
`59`	`59`	`# Construct the napari Tracks array`
`60`	`60`	`# Reorder axes to (individuals, keypoints, frames, xy)`
`61`		`- yx_cols = np.transpose(ds.position.values, (1, 2, 0, 3)).reshape(-1, 2)[`
	`61`	`+ yx_cols = np.transpose(ds.position.values, (3, 2, 0, 1)).reshape(-1, 2)[`
`62`	`62`	`:, [1, 0] # swap x and y columns`
`63`	`63`	`]`
`64`	`64`	`# Each keypoint of each individual is a separate track`