masks_to_boxes¶
- torchvision.ops.masks_to_boxes(masks: Tensor) Tensor[源]¶
計算所提供 masks 的邊界框。
返回一個包含邊界框的 [N, 4] 形狀的張量。邊界框的格式為
(x1, y1, x2, y2)，且0 <= x1 <= x2和0 <= y1 <= y2。警告
在大多數情況下，輸出會保證
x1 < x2和y1 < y2。但如果輸入是退化的，例如如果一個 mask 是單行或單列，那麼輸出可能出現 x1 = x2 或 y1 = y2 的情況。- 引數:
masks (Tensor[N, H, W]) – 要轉換的 masks，其中 N 是 masks 的數量，(H, W) 是空間維度。
- 返回值:
邊界框
- 返回型別:
Tensor[N, 4]
使用
masks_to_boxes的示例
