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]