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]