^jS$*ddlmZddlZddlZddlmZddlmZddlm Z m Z m Z ddl Z ddlZddlmZddlmZddlmZmZmZdd lmZmZmZdd lmZmZdd Zdd Zdd Z ddZ!dZ"ddZ#d dZ$d!dZ%d"dZ&d#dZ'd$dZ(d%dZ)d&dZ*d'dZ+d(dZ,d(dZ-y))) annotationsN)Sequence)suppress)AnyCallableLiteral) tv_tensors)sequence_to_str)_check_sequence_input _setup_angle _setup_size)get_dimensionsget_sizeis_pure_tensor) _FillType _FillTypeJITctt|tttfst |dt |t|tr't |dvrtd|dt |t|tr6|D]1}t|ttfrt|dt |t|ttfrt|t|g}|St|trHt |dk(rt|dt|dg}|St|dt|dg}|S)Nz2 should be a number or a sequence of numbers. Got )zIf z0 is a sequence its length should be 1 or 2. Got z& should be a sequence of numbers. Got rr) isinstanceintfloatr TypeErrortypelen ValueError)argnameelements k/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torchvision/transforms/v2/_utils.py_setup_number_or_seqr!s1 cC1 24& RSWX[S\R]^__#x SXV%;3tf$TUXY\U]T^_``#x  aGgU|4 D6)OPTU\P]!_`` a#U|$Sz5:& J C " s8q=Q=%A-0C JQ=%A-0C Jct|tr!|jD] }t|y|1t|tj t tfs tdyy)NzNGot inappropriate fill arg, only Numbers, tuples, lists and dicts are allowed.) rdictvalues_check_fill_argnumbersNumbertuplelistr)fillvalues r r&r&*sZ$[[] #E E " #  JtgnneT5R$Slm m%T r"c||St|ttfs!t|Dcgc] }t|}}|Scc}wN)rrrr*)r+vs r _convert_fill_argr03sA  | dS%L )"&t*-Qa-- K.s>ct|t|tr(|jD]\}}t |||<|Sdt |iS)Nothers)r&rr$itemsr0)r+kr/s r _setup_fill_argr5@sQD$JJL +DAq'*DG + +D122r"c>||vr||Sd|vr|dStdy)Nr2zWThis should never happen, please open an issue on the torchvision repo if you hit this.) RuntimeError) fill_dict inpt_types r _get_fillr:Ks1I## Y ""nor"cd|d}t|ttfr+t|dvst d|Ds t |yt|t s t |y)NzEPadding must be an int or a 1, 2, or 4 element of tuple or list, got .)rrc3<K|]}t|tywr.)rr).0ps r z%_check_padding_arg..Xs3X1Jq#4F3Xs)rr)r*rallrr)paddingerr_msgs r _check_padding_argrETsfUV]U^^_`G'E4=) w.sUSSYY[H=TSUs""c3FK|]}d|jvs|yw)labelNrPrRs r rAz1_find_labels_default_heuristic..s XCIIKAW Xs!!zCould not infer where the labels are in the sample. Try passing a callable as the labels_getter parameter?If there are no labels in the sample by design, pass labels_getter=None.) rr)r*r collectionsabcMappingrr StopIterationnextkeys)inputs candidate_keys r _find_labels_default_heuristicr^es &5$-(f fkoo55 6FFLXY X  M - VUFKKMUU V m $ Y X XXM Y W  -  VV Y Ys-!C"'!C."C+.C7cZ|dk(rtSt|r|S|dStd|d)Ndefaultcyr.)_s r z&_parse_labels_getter..sr"zGlabels_getter should either be 'default', a callable, or None, but got r<)r^callabler) labels_getters r _parse_labels_getterrgsB !-- -   bcpbqqrsttr"cX td|DS#t$r tdwxYw)zgReturn the Bounding Boxes in the input. Assumes only one ``BoundingBoxes`` object is present. c3VK|]!}t|tjs|#ywr.)rr BoundingBoxesr?inpts r rAz%get_bounding_boxes..s_TJtZE]E]4^D_))z*No bounding boxes were found in the samplerZrYr flat_inputss r get_bounding_boxesrqs6 G_[___ GEFFG)cX td|DS#t$r tdwxYw)z^Return the keypoints in the input. Assumes only one ``KeyPoints`` object is present. c3VK|]!}t|tjs|#ywr.)rr KeyPointsrks r rAz get_keypoints..s[TJtZEYEY4ZD[rmz%No keypoints were found in the samplernros r get_keypointsrvs6 B[[[[[ B@AABrrc |Dchc]^}t|ttjtjjtj frt t|`}}|s tdt|dkDr tdtt||j\}}}|||fScc}w)z"Return Channel, Height, and Width.z)No image or video was found in the samplerz/Found multiple CxHxW dimensions in the sample: ) check_typerr ImagePILVideor)rrrrr sortedpop)rprlchwschws r query_chwrs    d^Z-=-=syyPZP`P`a b nT"# D  CDD TQJ?[abf[gKhJijkkhhjGAq! a7N sA#B>c |Dchc]}t|ttjtjjtj tj tjtjfrtt|}}|s tdt|dkDr tdtt||j!\}}||fScc}w)zReturn Height and Width.zGNo image, video, mask, bounding box of keypoint was found in the samplerz-Found multiple HxW dimensions in the sample: )rxrr ryrzr{Maskrjrur)rrrrr r|r})rprlsizesrrs r query_sizers          (($$   htn E  abb UaHY_`eYfIgHhijj 99;DAq a4K+ sBC)cd|D]+}t|trt||s y||s+yyNTFrr)objtypes_or_checks type_or_checks r rxrxs<( -7 t-L:c= )S``cRd r"c.|D]}t||syyr)rx)rprrls r has_anyrs# dO , r"cr|D]2}|D]*}t|trt||s n ||s*1yy)NFTr)rprrrls r has_allrsI(  D2<]D2Qz$ .WdeiWj   r")rz#int | float | Sequence[int | float]rstrreturnzSequence[float])r+'_FillType | dict[type | str, _FillType]rNone)r+rrr)r+rrzdict[type | str, _FillTypeJIT])rCzint | Sequence[int]rr)rKz3Literal['constant', 'edge', 'reflect', 'symmetric']rr)r\rrz torch.Tensor)rfz!str | Callable[[Any], Any] | NonerzCallable[[Any], Any])rp list[Any]rztv_tensors.BoundingBoxes)rprrztv_tensors.KeyPoints)rprrztuple[int, int, int])rprrztuple[int, int])rrrz(tuple[type | Callable[[Any], bool], ...]rbool)rprrztype | Callable[[Any], bool]rr). __future__rcollections.abcrVr'r contextlibrtypingrrr PIL.Imagerztorch torchvisionr torchvision._utilsr !torchvision.transforms.transformsr r r $torchvision.transforms.v2.functionalrrr+torchvision.transforms.v2.functional._utilsrrr!r&r0r5r:rErLr^rgrqrvrrrxrrrbr"r rs"$)) ".^^YYO(n 3p"_ )!Xu G B 4r"