From 43e6a1e2e1cefa1509dce69e0f0de714994d873e Mon Sep 17 00:00:00 2001 From: reidmovis <59581602+reidmovis@users.noreply.github.com> Date: Tue, 28 Oct 2025 17:08:00 -0400 Subject: [PATCH] dot_size_configurability --- tapnet/utils/viz_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tapnet/utils/viz_utils.py b/tapnet/utils/viz_utils.py index df954b4..cb6b555 100644 --- a/tapnet/utils/viz_utils.py +++ b/tapnet/utils/viz_utils.py @@ -49,6 +49,7 @@ def paint_point_track( point_tracks: np.ndarray, visibles: np.ndarray, colormap: Optional[List[Tuple[int, int, int]]] = None, + dot_size_multiple: int ) -> np.ndarray: """Converts a sequence of points to color code video. @@ -65,7 +66,7 @@ def paint_point_track( if colormap is None: colormap = get_colors(num_colors=num_points) height, width = frames.shape[1:3] - dot_size_as_fraction_of_min_edge = 0.015 + dot_size_as_fraction_of_min_edge = (0.0015 x dot_size_multiple) radius = int(round(min(height, width) * dot_size_as_fraction_of_min_edge)) diam = radius * 2 + 1 quadratic_y = np.square(np.arange(diam)[:, np.newaxis] - radius - 1)