快捷方式

masks_to_boxes

torchvision.ops.masks_to_boxes(masks: Tensor) Tensor[源代码]

计算提供的掩码周围的边界框。

返回包含边界框的 [N, 4] 张量。框采用 (x1, y1, x2, y2) 格式,其中 0 <= x1 < x20 <= y1 < y2

参数:

masks (Tensor[N, H, W]) – 要转换的掩码,其中 N 是掩码的数量,(H, W) 是空间维度。

返回值:

边界框

返回类型:

Tensor[N, 4]

使用 masks_to_boxes 的示例

将掩码重新用于边界框

将掩码重新用于边界框

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的答案

查看资源