Skip to content

Commit 64e104e

Browse files
committed
set clamp=False in reference for both rotated and non-rotated
1 parent 9e0c2dd commit 64e104e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/test_transforms_v2.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1301,11 +1301,11 @@ def _reference_horizontal_flip_bounding_boxes(self, bounding_boxes: tv_tensors.B
13011301
)
13021302

13031303
helper = (
1304-
functools.partial(reference_affine_rotated_bounding_boxes_helper, flip=True, clamp=False)
1304+
functools.partial(reference_affine_rotated_bounding_boxes_helper, flip=True)
13051305
if tv_tensors.is_rotated_bounding_format(bounding_boxes.format)
13061306
else reference_affine_bounding_boxes_helper
13071307
)
1308-
return helper(bounding_boxes, affine_matrix=affine_matrix)
1308+
return helper(bounding_boxes, affine_matrix=affine_matrix, clamp=False)
13091309

13101310
@pytest.mark.parametrize("format", list(tv_tensors.BoundingBoxFormat))
13111311
@pytest.mark.parametrize(
@@ -1910,11 +1910,11 @@ def _reference_vertical_flip_bounding_boxes(self, bounding_boxes: tv_tensors.Bou
19101910
)
19111911

19121912
helper = (
1913-
functools.partial(reference_affine_rotated_bounding_boxes_helper, flip=True, clamp=False)
1913+
functools.partial(reference_affine_rotated_bounding_boxes_helper, flip=True)
19141914
if tv_tensors.is_rotated_bounding_format(bounding_boxes.format)
19151915
else reference_affine_bounding_boxes_helper
19161916
)
1917-
return helper(bounding_boxes, affine_matrix=affine_matrix)
1917+
return helper(bounding_boxes, affine_matrix=affine_matrix, clamp=False)
19181918

19191919
@pytest.mark.parametrize("format", list(tv_tensors.BoundingBoxFormat))
19201920
@pytest.mark.parametrize("fn", [F.vertical_flip, transform_cls_to_functional(transforms.RandomVerticalFlip, p=1)])

0 commit comments

Comments
 (0)