clip_boxes_to_image

torchvision.ops.clip_boxes_to_image(boxes: 张量, size: 元组[整型, 整型]) → 张量

将边界框裁剪到 size 指定图像尺寸内部。

注

对于裁剪 BoundingBoxes 对象，请考虑使用 ClampBoundingBoxes() 变换。

参数:

• boxes (张量[N, 4]) – 采用 (x1, y1, x2, y2) 格式的边界框，满足 0 <= x1 < x2 且 0 <= y1 < y2。

• size (元组[高, 宽]) – 图像尺寸

返回值:

裁剪后的边界框

返回值类型:

张量[N, 4]