ConvertBoundingBoxFormat — Torchvision 0.22 documentation (original) (raw)

class torchvision.transforms.v2.ConvertBoundingBoxFormat(format: Union[str, BoundingBoxFormat])[source]¶

Convert bounding box coordinates to the given format, eg from “CXCYWH” to “XYXY”.

Parameters:

format (str or tv_tensors.BoundingBoxFormat) – output bounding box format. Possible values are defined by BoundingBoxFormat and string values match the enums, e.g. “XYXY” or “XYWH” etc.

transform(inpt: BoundingBoxes, params: Dict[str, Any]) → BoundingBoxes[source]¶

Method to override for custom transforms.

See How to write your own v2 transforms