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, y2 为左上角，w, h 为宽度和高度。

'cxcywh': 边界框表示为中心、宽度和高度，cx, cy 为框的中心，w, h 为宽度和高度。

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

'cxcywhr': 边界框表示为中心、宽度和高度，cx, cy 为框的中心，w, h 为宽度和高度。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]