Remove convert_bounding_boxes_to_keypoints

Author: NicolasHug

docs/source/transforms.rst

@@ -459,7 +459,6 @@ functionals
     v2.functional.to_pil_image
     v2.functional.to_dtype
     v2.functional.convert_bounding_box_format
-    v2.functional.convert_bounding_boxes_to_keypoints
 
 
 Deprecated

test/test_transforms_v2.py

@@ -7274,37 +7274,3 @@ def test_different_sizes(self, make_input1, make_input2, query):
     def test_no_valid_input(self, query):
         with pytest.raises(TypeError, match="No image"):
             query(["blah"])
-
-
-@pytest.mark.parametrize(
-    "boxes",
-    [
-        tv_tensors.BoundingBoxes(torch.tensor([[1.0, 1.0, 2.0, 2.0]]), format="XYXY", canvas_size=(4, 4)),
-        tv_tensors.BoundingBoxes(torch.tensor([[1.0, 1.0, 1.0, 1.0]]), format="XYWH", canvas_size=(4, 4)),
-        tv_tensors.BoundingBoxes(torch.tensor([[1.5, 1.5, 1.0, 1.0]]), format="CXCYWH", canvas_size=(4, 4)),
-        tv_tensors.BoundingBoxes(torch.tensor([[1.5, 1.5, 1.0, 1.0, 45]]), format="CXCYWHR", canvas_size=(4, 4)),
-        tv_tensors.BoundingBoxes(torch.tensor([[1.0, 1.0, 1.0, 1.0, 45.0]]), format="XYWHR", canvas_size=(4, 4)),
-        tv_tensors.BoundingBoxes(
-            torch.tensor([[1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 2.0, 1.0]]), format="XYXYXYXY", canvas_size=(4, 4)
-        ),
-    ],
-)
-def test_convert_bounding_boxes_to_keypoints(boxes: tv_tensors.BoundingBoxes):
-    kp = F.convert_bounding_boxes_to_keypoints(boxes)
-    assert kp.shape == (boxes.shape[0], 4, 2)
-    assert kp.dtype == boxes.dtype
-
-    # We manually convert the kp back into a BoundingBoxes, and convert that
-    # bbox back into the original `boxes` format to compare against it.
-    if tv_tensors.is_rotated_bounding_format(boxes.format):
-        reconverted = kp.reshape(-1, 8)
-        intermediate_format = tv_tensors.BoundingBoxFormat.XYXYXYXY
-    else:
-        reconverted = torch.cat([kp[..., 0, :], kp[..., 2, :]], dim=-1)
-        intermediate_format = tv_tensors.BoundingBoxFormat.XYXY
-
-    reconverted_bbox = F.convert_bounding_box_format(
-        tv_tensors.BoundingBoxes(reconverted, format=intermediate_format, canvas_size=kp.canvas_size),
-        new_format=boxes.format,
-    )
-    assert_equal(reconverted_bbox, boxes, atol=1e-5, rtol=0)

torchvision/transforms/v2/functional/__init__.py

@@ -6,7 +6,6 @@
     clamp_bounding_boxes,
     clamp_keypoints,
     convert_bounding_box_format,
-    convert_bounding_boxes_to_keypoints,
     get_dimensions_image,
     get_dimensions_video,
     get_dimensions,

torchvision/transforms/v2/functional/_meta.py

@@ -189,32 +189,6 @@ def _xyxyxyxy_to_keypoints(bounding_boxes: torch.Tensor) -> torch.Tensor:
     return bounding_boxes[:, [[0, 1], [2, 3], [4, 5], [6, 7]]]
 
 
-# Note: this doesn't have a corresponding transforms class.
-def convert_bounding_boxes_to_keypoints(bounding_boxes: tv_tensors.BoundingBoxes) -> tv_tensors.KeyPoints:
-    """Convert a set of bounding boxes to its edge points.
-
-    Args:
-        bounding_boxes (tv_tensors.BoundingBoxes): A set of ``N`` bounding boxes (of shape ``[N, 4]``)
-
-    Returns:
-        tv_tensors.KeyPoints: The edges, as a polygon of shape ``[N, 4, 2]``
-    """
-    if tv_tensors.is_rotated_bounding_format(bounding_boxes.format):
-        intermediate_format = BoundingBoxFormat.XYXYXYXY
-        to_keypoints = _xyxyxyxy_to_keypoints
-    else:
-        intermediate_format = BoundingBoxFormat.XYXY
-        to_keypoints = _xyxy_to_keypoints
-
-    bbox = _convert_bounding_box_format(
-        bounding_boxes.as_subclass(torch.Tensor),
-        old_format=bounding_boxes.format,
-        new_format=intermediate_format,
-        inplace=False,
-    )
-    return tv_tensors.KeyPoints(to_keypoints(bbox), canvas_size=bounding_boxes.canvas_size)
-
-
 def _cxcywhr_to_xywhr(cxcywhr: torch.Tensor, inplace: bool) -> torch.Tensor:
     if not inplace:
         cxcywhr = cxcywhr.clone()