o V h"@sLddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZddlZddlZddlmZddlmZddlmZmZmZddlmZmZmZdd l m!Z!m"Z"dHddZ#dIddZ$dJddZ%dKddZ&ddZ'dLd"d#Z(dMd&d'Z)dNd+d,Z*dOd0d1Z+dPd5d6Z,dQd8d9Z-dRd;d<Z.dSdAdBZ/dTdDdEZ0dTdFdGZ1dS)U) annotationsN)suppress) AnyCallableDictListLiteralSequenceTupleTypeUnion) tv_tensors)sequence_to_str)_check_sequence_input _setup_angle _setup_size)get_dimensionsget_sizeis_pure_tensor) _FillType _FillTypeJITarg.Union[int, float, Sequence[Union[int, float]]]namestrreturnSequence[float]cCst|tttfst|dt|t|tr*t|dvr*td|dt|t|trF|D]}t|ttfsEt|dt|q1t|ttfrWt|t|g}|St|tr|t|dkrpt|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)rrelementr'T/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/transforms/v2/_utils.py_setup_number_or_seqs$  r)fill3Union[_FillType, Dict[Union[Type, str], _FillType]]NonecCsNt|tr|D]}t|q dS|dur#t|tjttfs%tddSdS)NzNGot inappropriate fill arg, only Numbers, tuples, lists and dicts are allowed.) rdictvalues_check_fill_argnumbersNumbertuplelistr")r*valuer'r'r(r/)s   r/rrcCs0|dur|St|ttfsddt|D}|S)NcSsg|]}t|qSr')r!).0vr'r'r( ;sz%_convert_fill_arg..)rr r!r3)r*r'r'r(_convert_fill_arg2s r8$Dict[Union[Type, str], _FillTypeJIT]cCs@t|t|tr|D] \}}t|||<q |Sdt|iS)Nothers)r/rr-itemsr8)r*kr6r'r'r(_setup_fill_arg?s   r=cCs,||vr||Sd|vr|dStddS)Nr:zWThis should never happen, please open an issue on the torchvision repo if you hit this.) RuntimeError) fill_dict inpt_typer'r'r( _get_fillJs  rApaddingUnion[int, Sequence[int]]cCsZd|d}t|ttfr"t|dvstdd|Ds t|dSt|ts+t|dS)NzEPadding must be an int or a 1, 2, or 4 element of tuple or list, got .)rrcss|]}t|tVqdSN)rr )r5pr'r'r( Wsz%_check_padding_arg..)rr2r3r$allr%r )rBerr_msgr'r'r(_check_padding_argSs  rK padding_mode3Literal['constant', 'edge', 'reflect', 'symmetric']cCs|dvrtddS)N)constantedgereflect symmetriczBPadding mode should be either constant, edge, reflect or symmetric)r%)rLr'r'r(_check_padding_mode_arg_srRinputsr torch.TensorcCst|ttfr |d}t|r|St|tjjs td|dd}tt t dd| D}Wdn1s.css |] }d|vr|VqdS)labelNrVrXr'r'r(rHrZzCould 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.) rr2r3r collectionsabcMappingr%r StopIterationnextkeys)rS candidate_keyr'r'r(_find_labels_default_heuristicds.  rc labels_getter&Union[str, Callable[[Any], Any], None]Callable[[Any], Any]cCs8|dkrtSt|r |S|durddStd|d)NdefaultcSsdSrFr')_r'r'r(sz&_parse_labels_getter..zGlabels_getter should either be 'default', a callable, or None, but got rD)rccallabler%)rdr'r'r(_parse_labels_gettersrk flat_inputs List[Any]tv_tensors.BoundingBoxescCs,z tdd|DWStytdw)zgReturn the Bounding Boxes in the input. Assumes only one ``BoundingBoxes`` object is present. css |] }t|tjr|VqdSrF)rr BoundingBoxesr5inptr'r'r(rHrZz%get_bounding_boxes..z*No bounding boxes were found in the sample)r`r_r%)rlr'r'r(get_bounding_boxess  rrTuple[int, int, int]cCsTdd|D}|s tdt|dkrtdtt||\}}}|||fS)z"Return Channel, Height, and Width.cSs2h|]}t|ttjtjjtjfrtt|qSr') check_typerr ImagePILVideor2rrpr'r'r( s  zquery_chw..z)No image or video was found in the samplerz/Found multiple CxHxW dimensions in the sample: r"r$r%rsortedpop)rlchwschwr'r'r( query_chws  rTuple[int, int]cCsPdd|D}|s tdt|dkrtdtt||\}}||fS)zReturn Height and Width.c Ss:h|]}t|ttjtjjtjtjtjfrtt |qSr') rtrr rurvrwMaskror2rrpr'r'r(rxs zquery_size..z=No image, video, mask or bounding box was found in the samplerz-Found multiple HxW dimensions in the sample: ry)rlsizesr~rr'r'r( query_sizes  robjtypes_or_checks.Tuple[Union[Type, Callable[[Any], bool]], ...]boolcCs6|D]}t|trt||rdS||rdSqdSNTFrr#)rr type_or_checkr'r'r(rtsrt"Union[Type, Callable[[Any], bool]]cGs|D] }t||r dSqdSr)rt)rlrrqr'r'r(has_anys  rcGs@|D]}|D]}t|trt||rn||rnqdSqdS)NFTr)rlrrrqr'r'r(has_allsr)rrrrrr)r*r+rr,)r*rrr)r*r+rr9)rBrCrr,)rLrMrr,)rSrrrT)rdrerrf)rlrmrrn)rlrmrrs)rlrmrr)rrrrrr)rlrmrrrr)2 __future__rcollections.abcr\r0 contextlibrtypingrrrrrr r r r PIL.Imagervtorch torchvisionr torchvision._utilsr!torchvision.transforms.transformsrrr$torchvision.transforms.v2.functionalrrr+torchvision.transforms.v2.functional._utilsrrr)r/r8r=rArKrRrcrkrrrrrtrrr'r'r'r(s6  ,       ,