From 516cfaecf34c73c4af4435a60622bca934b090e1 Mon Sep 17 00:00:00 2001 From: jsalvasoler Date: Wed, 11 Mar 2026 10:02:49 +0100 Subject: [PATCH] fix: clamping to avoid v2.ElasticTransform IndexError --- torchvision/transforms/v2/functional/_geometry.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchvision/transforms/v2/functional/_geometry.py b/torchvision/transforms/v2/functional/_geometry.py index 84986629ba6..6e66f418ae5 100644 --- a/torchvision/transforms/v2/functional/_geometry.py +++ b/torchvision/transforms/v2/functional/_geometry.py @@ -2454,6 +2454,8 @@ def elastic_bounding_boxes( points = points.ceil_() index_xy = points.to(dtype=torch.long) index_x, index_y = index_xy[:, 0], index_xy[:, 1] + index_x = index_x.clamp(0, inv_grid.shape[2] - 1) + index_y = index_y.clamp(0, inv_grid.shape[1] - 1) # Transform points: t_size = torch.tensor(canvas_size[::-1], device=displacement.device, dtype=displacement.dtype)