PhrUddlZddlmZmZmZmZmZmZmZm Z ddl Z ddl m Z e e eee eedfZeeeZdedefdZiZeeeeeffed<dZd d d Zd Ze j0j3Dchc]"}e|es e|e j:s!|$c}Zd ZdddZ dZ!ycc}w)N)AnyCallableDictListOptionalSequenceTypeUnion) tv_tensorsinptreturncpt|tjxrt|tj SN) isinstancetorchTensorr TVTensor)r s vC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torchvision/transforms/v2/functional/_utils.pyis_pure_tensorr s( dELL ) W*T:CVCV2W.WW_KERNEL_REGISTRYcBtjfd}|S)Nc|jtjg|i|}tj||S)Nlike) as_subclassrrr wrap)r argskwargsoutputkernels rwrapperz*_kernel_tv_tensor_wrapper..wrappers9((6HHHvD11r functoolswrapsr!r"s` r_kernel_tv_tensor_wrapperr's%__V 2 2 NrTtv_tensor_wrappercrtj|ivrtd|ddfd}|S)Nz Functional z* already has a kernel registered for type .c`ttjr r t|n|<|Sr) issubclassr rr')r! input_typeregistryr)s r decoratorz,_register_kernel_internal..decorator*s6*j&9&9:?P &f -   r)r setdefault ValueError) functionalr.r)r0r/s `` @r_register_kernel_internalr4%sH**:r:HX;zl2\]g\hhijkk rcddl} t|jjj|S#t $rt d|ddwxYw)Nrz%Could not find functional with name 'z*' in torchvision.transforms.v2.functional.)$torchvision.transforms.v2.functionalgetattr transformsv2r3AttributeErrorr2)name torchvisions r_name_to_functionalr=5sU/{--00;;TBB 3D69c d  s )0A cbt|tr t|}n6t|rt |ddj dst d|dt|trt|tjst d|d|tvrt d|t||d S) zDecorate a kernel to register it for a functional and a (custom) tv_tensor type. See :ref:`sphx_glr_auto_examples_transforms_plot_custom_tv_tensors.py` for usage details. )r; __module__z$torchvision.transforms.v2.functionalzoKernels can only be registered on functionals from the torchvision.transforms.v2.functional namespace, but got r+zZKernels can only be registered for subclasses of torchvision.tv_tensors.TVTensor, but got zHKernels cannot be registered for the builtin tv_tensor classes, but got Fr() rstrr=callabler7 startswithr2typer-r r_BUILTIN_DATAPOINT_TYPESr4)r3 tv_tensor_clss rregister_kernelrGEs *c"(j9  J b 1 < <=c d!l! %  }d + =*J]J]0^$oQ (  00cdqcrstt $ZRW XXrF)allow_passthroughc &tj|}|std|jd|jD]!}||vr||cS|t j us!n|rdStd|jd|jd|d)Nz$No kernel registered for functional r+c|Sr)r rrs rz_get_kernel..qsTrz Functional F.z supports inputs of type z , but got z instead.) rgetr2__name____mro__r r TypeErrorkeys)r3r.rHr/clss r _get_kernelrSbs##J/H ? @S@S?TTUVWW!! (?C= J'' ' "11   ++,,EhmmoEVW,i ) rcxtj|ivrtd|dddfd}|S)Nz Functional 'z,' already has a kernel registered for type 'z'.cBtjfd}|S)NcVg|i|}t|}|fd|DS)Nc3LK|]}tj|yw)rN)r r).0or s r zY_register_five_ten_crop_kernel_internal..wrap..wrapper..s !PA*//!$"?"?s!$)rD)r rrr container_typer!s` rr"zF_register_five_ten_crop_kernel_internal..wrap..wrappers1D24262F!&\N!!P!PP Prr#r&s` rrz5_register_five_ten_crop_kernel_internal..wraps'   Q ! Q rcVttjr|n|<|Sr)r-r r)r!r.r/rs rr0z:_register_five_ten_crop_kernel_internal..decorators(/9*jFYFY/ZtF|`f r)rr1rP)r3r.r0r/rs ` @@r'_register_five_ten_crop_kernel_internalr]{sM**:r:HX,zl2^_i^jjlmnn r)"r$typingrrrrrrr r rr<r intfloat _FillType _FillTypeJITboolrr__annotations__r'r4r=__dict__valuesrrDr-rrErGrSr])objs0rrhsMMM " #uhsmXe_dB C U $ XXX :<$xdHn!556;$LP  &&--//C:c43HZX[]g]p]pMqC/ Y:>C2ws=B4 B4 B4