box_convert

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

將 torch.Tensor 框從給定的 in_fmt 格式轉換為 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 是寬度和高度。

引數:

• boxes (Tensor[N, 4]) – 將要被轉換的框。

• in_fmt (str) – 給定框的輸入格式。支援的格式包括 [‘xyxy’, ‘xywh’, ‘cxcywh’]。

• out_fmt (str) – 給定框的輸出格式。支援的格式包括 [‘xyxy’, ‘xywh’, ‘cxcywh’]

返回:

轉換後的框。

返回型別:

Tensor[N, 4]