Skip to content

Commit 8dc9ce4

Browse files
Add affine transformation tests for rotated boxes
Test Plan: Unit tests: ```bash pytest test/test_transforms_v2.py -vvv -k "TestAffine and test_transform_bounding_boxes_correctness" ```
1 parent 580ae4b commit 8dc9ce4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_transforms_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1641,7 +1641,7 @@ def test_functional_bounding_boxes_correctness(self, format, angle, translate, s
16411641

16421642
torch.testing.assert_close(actual, expected)
16431643

1644-
@pytest.mark.parametrize("format", SUPPORTED_BOX_FORMATS)
1644+
@pytest.mark.parametrize("format", list(tv_tensors.BoundingBoxFormat))
16451645
@pytest.mark.parametrize("center", _CORRECTNESS_AFFINE_KWARGS["center"])
16461646
@pytest.mark.parametrize("seed", list(range(5)))
16471647
def test_transform_bounding_boxes_correctness(self, format, center, seed):

0 commit comments

Comments
 (0)