o V h{@sTddlmZmZmZddlmZddlmZm Z Gddde Z Gddde Z dS) )AnyDictUnion) tv_tensors) functional Transformcs^eZdZdZejfZdeeej fddffdd Z dejde ee fdejfd d Z ZS) ConvertBoundingBoxFormata_Convert bounding box coordinates to the given ``format``, eg from "CXCYWH" to "XYXY". Args: format (str or tv_tensors.BoundingBoxFormat): output bounding box format. Possible values are defined by :class:`~torchvision.tv_tensors.BoundingBoxFormat` and string values match the enums, e.g. "XYXY" or "XYWH" etc. formatreturnNcst||_dSN)super__init__r )selfr  __class__S/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/transforms/v2/_meta.pyr s  z!ConvertBoundingBoxFormat.__init__inptparamscCstj||jdS)N) new_format)Fconvert_bounding_box_formatr rrrrrr transformsz"ConvertBoundingBoxFormat.transform)__name__ __module__ __qualname____doc__r BoundingBoxes_transformed_typesrstrBoundingBoxFormatr rrr __classcell__rrrrrs  *rc@s:eZdZdZejfZdejdeee fdejfddZ dS)ClampBoundingBoxeszClamp bounding boxes to their corresponding image dimensions. The clamping is done according to the bounding boxes' ``canvas_size`` meta-data. rrr cCs t|Sr )rclamp_bounding_boxesrrrrr#s zClampBoundingBoxes.transformN) rrrrrrrrr rrrrrrr#s&r#N) typingrrr torchvisionrtorchvision.transforms.v2rrrrr#rrrrs