clip_boxes_to_image

torchvision.ops.clip_boxes_to_image(boxes: Tensor, size: tuple[int, int]) → Tensor

将边界框剪辑到大小为 size 的图像内。

注意

要剪辑 BoundingBoxes 对象，请考虑改用变换 ClampBoundingBoxes()。

参数:

• boxes (Tensor[N, 4]) – 格式为 (x1, y1, x2, y2) 的边界框，其中 0 <= x1 < x2 和 0 <= y1 < y2。

• size (Tuple[height, width]) – 图像的大小

返回:

剪辑后的边界框

返回类型:

Tensor[N, 4]