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的範例
