masks_to_boxes¶
- torchvision.ops.masks_to_boxes(masks: Tensor) Tensor[源代码]¶
计算提供掩码周围的边界框。
返回一个包含边界框的 [N, 4] 张量。框的格式为
(x1, y1, x2, y2)，其中0 <= x1 <= x2和0 <= y1 <= y2。警告
在大多数情况下，输出将保证
x1 < x2和y1 < y2。但如果输入是退化的，例如，如果掩码是单行或单列，则输出可能出现 x1 = x2 或 y1 = y2 的情况。- 参数:
masks (Tensor[N, H, W]) – 要转换的掩码，其中 N 是掩码的数量，(H, W) 是空间维度。
- 返回:
边界框
- 返回类型:
Tensor[N, 4]
使用
masks_to_boxes的示例
