box_convert

torchvision.ops.box_convert(boxes: Tensor, in_fmt: str, out_fmt: str) → Tensor

将给定 in_fmt 格式的 torch.Tensor 边界框转换为 out_fmt 格式。

注意

要在不同格式之间转换 torch.Tensor 或 BoundingBoxes 对象，请考虑使用 convert_bounding_box_format()。或者查看相应的变换 ConvertBoundingBoxFormat()。

支持的 in_fmt 和 out_fmt 字符串有：

'xyxy': 边界框通过左上角 x1, y1 和右下角 x2, y2 表示。这是 torchvision 工具所期望的格式。

'xywh': 边界框通过左上角 x1, y1、宽度 w 和高度 h 表示。

'cxcywh': 边界框通过中心点 cx, cy、宽度 w 和高度 h 表示。

'xywhr': 边界框通过左上角 x1, y1、宽度 w、高度 h 和旋转角度 r 表示。r 是相对于框中心，在图像平面上逆时针旋转的度数 \(|r|\)。

'cxcywhr': 边界框通过中心点 cx, cy、宽度 w、高度 h 和旋转角度 r 表示。r 是相对于框中心，在图像平面上逆时针旋转的度数 \(|r|\)。

'xyxyxyxy': 边界框通过四个顶点表示：左上角 x1, y1、右上角 x2, y2、右下角 x3, y3 和左下角 x4, y4。

参数:

• boxes (Tensor[N, K]) – 将要转换的边界框。K 是坐标数量（无旋转边界框为 4，有旋转边界框为 5 或 8）。

• in_fmt (str) – 给定边界框的输入格式。支持的格式为 [‘xyxy’, ‘xywh’, ‘cxcywh’, ‘xywhr’, ‘cxcywhr’, ‘xyxyxyxy’]。

• out_fmt (str) – 给定边界框的输出格式。支持的格式为 [‘xyxy’, ‘xywh’, ‘cxcywh’, ‘xywhr’, ‘cxcywhr’, ‘xyxyxyxy’]。

返回:

转换后的格式的边界框。

返回类型:

Tensor[N, K]