masks_to_boxes¶ torchvision.ops.masks_to_boxes(masks: Tensor) → Tensor[源代码]¶ 计算提供的掩码周围的边界框。 返回包含边界框的 [N, 4] 张量。框采用 (x1, y1, x2, y2) 格式,其中 0 <= x1 < x2 和 0 <= y1 < y2。 参数: masks (Tensor[N, H, W]) – 要转换的掩码,其中 N 是掩码的数量,(H, W) 是空间维度。 返回值: 边界框 返回类型: Tensor[N, 4] 使用 masks_to_boxes 的示例 将掩码重新用于边界框 将掩码重新用于边界框