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