^j] ddlmZmZddlmZddlmZmZddl m Z GddeZ GddeZ Gd d eZ Gd d eZy ))AnyUnion) tv_tensors) functional Transform)CLAMPING_MODE_TYPEceZdZdZej fZdeeejfddffd Z dej de ee fdej fdZ xZS) 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. formatreturnNc0t|||_yN)super__init__r )selfr __class__s j/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torchvision/transforms/v2/_meta.pyrz!ConvertBoundingBoxFormat.__init__s  inptparamscDtj||jS)N) new_format)Fconvert_bounding_box_formatr rrrs r transformz"ConvertBoundingBoxFormat.transforms,,TdkkJJr)__name__ __module__ __qualname____doc__r BoundingBoxes_transformed_typesrstrBoundingBoxFormatrdictrr __classcell__rs@rr r so%224uS**F*F%FGDKj66KS#XKS]SkSkKrr ceZdZdZd deeefddffd ZejfZ dejde ee fdejfdZ xZS) ClampBoundingBoxesa Clamp bounding boxes to their corresponding image dimensions. Args: clamping_mode: Default is "auto" which relies on the input box' ``clamping_mode`` attribute. Read more in :ref:`clamping_mode_tuto` for more details on how to use this transform. clamping_moder Nc0t|||_yr)rrr*rr*rs rrzClampBoundingBoxes.__init__$s *rrrcDtj||jS)N)r*)rclamp_bounding_boxesr*rs rrzClampBoundingBoxes.transform*s%%d$:L:LMMr)auto)rrrr rrr#rrr!r"r%rrr&r's@rr)r)sk+e,>,C&D+RV+%224Nj66NS#XNS]SkSkNrr)cpeZdZdZej fZdej deee fdej fdZ y)ClampKeyPointszClamp keypoints to their corresponding image dimensions. The clamping is done according to the keypoints' ``canvas_size`` meta-data. rrr c,tj|Sr)rclamp_keypointsrs rrzClampKeyPoints.transform6s  &&rN) rrrr r KeyPointsr"r%r#rrrrr1r1.sF %..0'j22'DcN'zOcOc'rr1ceZdZdZdeddffd ZejfZdejde e e fdejfdZ xZ S) SetClampingModea4Sets the ``clamping_mode`` attribute of the bounding boxes for future transforms. Args: clamping_mode: The clamping mode to set. Possible values are: "soft", "hard", or ``None``. Read more in :ref:`clamping_mode_tuto` for more details on how to use this transform. r*r Ncht|||_|jdvrtd|y)N)Nsofthardz.clamping_mode must be soft, hard or None, got )rrr* ValueErrorr,s rrzSetClampingMode.__init__Es= *   %; ;Mm_]^ ^ rCsE"@EKyK&NN& 'Y 'ir