khddlmZmZmZmZmZmZmZddlZddlm Z ddl m Z ddl mZGddeZGdd eZGd d eZGd d eZy))AnyCallableDictListOptionalSequenceUnionN)nn) transforms) TransformcNeZdZdZdeeddffd ZdedefdZde fdZ xZ S) ComposeaComposes several transforms together. This transform does not support torchscript. Please, see the note below. Args: transforms (list of ``Transform`` objects): list of transforms to compose. Example: >>> transforms.Compose([ >>> transforms.CenterCrop(10), >>> transforms.PILToTensor(), >>> transforms.ConvertImageDtype(torch.float), >>> ]) .. note:: In order to script the transformations, please use ``torch.nn.Sequential`` as below. >>> transforms = torch.nn.Sequential( >>> transforms.CenterCrop(10), >>> transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)), >>> ) >>> scripted_transforms = torch.jit.script(transforms) Make sure to use only scriptable transformations, i.e. that work with ``torch.Tensor``, does not require `lambda` functions or ``PIL.Image``. r returnNct|t|ts t d|s t d||_y)N5Argument transforms should be a sequence of callableszPass at least one transform)super__init__ isinstancer TypeError ValueErrorr selfr __class__s `/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/transforms/v2/_container.pyrzCompose.__init__(s: *h/ST T:; ;$inputsc\t|dkD}|jD]}||}|r|n|f}SN)lenr rrneeds_unpacking transformoutputss rforwardzCompose.forward0s@f+/ @I(G /WgZF @rcrg}|jD]}|jd|dj|SNz  r appendjoinr format_stringts r extra_reprzCompose.extra_repr7=  -A  4s , -yy''r) __name__ __module__ __qualname____doc__rrrrr%strr/ __classcell__rs@rrr s?:%8H#5%$%ss(C(rrceZdZdZej Zd deee e jfde ddffd Z deeeffdZdedefd Zdefd ZxZS) RandomApplyaApply randomly a list of transformations with a given probability. .. note:: In order to script the transformation, please use ``torch.nn.ModuleList`` as input instead of list/tuple of transforms as shown below: >>> transforms = transforms.RandomApply(torch.nn.ModuleList([ >>> transforms.ColorJitter(), >>> ]), p=0.3) >>> scripted_transforms = torch.jit.script(transforms) Make sure to use only scriptable transformations, i.e. that work with ``torch.Tensor``, does not require `lambda` functions or ``PIL.Image``. Args: transforms (sequence or torch.nn.Module): list of transformations p (float): probability of applying the list of transforms r prNct|t|ttj fs t d||_d|cxkrdkstdtd||_ y)NzJArgument transforms should be a sequence of callables or a `nn.ModuleList`gg?z@`p` should be a floating point value in the interval [0.0, 1.0].) rrrrr ModuleListrr rr:)rr r:rs rrzRandomApply.__init__Tsc *x&?@hi i$qC_` ` _` `rc4|j|jdS)Nr r:r>)rs r _extract_params_for_v1_transformz,RandomApply._extract_params_for_v1_transform_s"ooDFF;;rrct|dkD}tjd|jk\r |r|S|dS|jD]}||}|r|n|f}S)Nrr)r torchrandr:r r!s rr%zRandomApply.forwardbsef+/ ::a=DFF ",6 ;&) ; @I(G /WgZF @rcrg}|jD]}|jd|dj|Sr'r)r,s rr/zRandomApply.extra_reprmr0r)g?)r1r2r3r4 _transformsr9_v1_transform_clsr rrr r<floatrrr5rr?r%r/r6r7s@rr9r9>sx&$// 5();R]])J#K PU `d <$sCx.< s s (C(rr9cVeZdZdZ d deedeeeddffd Z de de fdZ xZ S) RandomChoiceaApply single transformation randomly picked from a list. This transform does not support torchscript. Args: transforms (sequence or torch.nn.Module): list of transformations p (list of floats or None, optional): probability of each transform being picked. If ``p`` doesn't sum to 1, it is automatically normalized. If ``None`` (default), all transforms have the same probability. Nr r:rcNt|ts td|dgt|z}n:t|t|k7r#t dt|dt|t |||_t|}|Dcgc]}||z  c}|_ ycc}w)Nrrz4Length of p doesn't match the number of transforms: z != ) rrrr rrrr sumr:)rr r:totalprobrs rrzRandomChoice.__init__s *h/ST T 9c*o%A Vs: &STWXYTZS[[_`cdn`o_pqr r $A+,-4$,--s B"rcttjtj|jd}|j |}||Sr)intrA multinomialtensorr:r )rridxr#s rr%zRandomChoice.forwards?%##ELL$8!<=OOC( &!!r)N) r1r2r3r4rrrrrFrrr%r6r7s@rrHrHtsO $(.X&. DK .  .&"s"s"rrHcBeZdZdZdeeddffd ZdedefdZxZ S) RandomOrderzApply a list of transformations in a random order. This transform does not support torchscript. Args: transforms (sequence or torch.nn.Module): list of transformations r rNcft|ts tdt|||_y)Nr)rrrrrr rs rrzRandomOrder.__init__s+*h/ST T $rrct|dkD}tjt|jD]}|j|}||}|r|n|f}Sr)r rArandpermr )rrr"rQr#r$s rr%zRandomOrder.forwards\f+/>>#doo"67 @C,I(G /WgZF @r) r1r2r3r4rrrrr%r6r7s@rrSrSs3%8H#5%$% ssrrS)typingrrrrrrr rAr torchvisionr rDtorchvision.transforms.v2r rr9rHrSrrr[sMGGG 1/1(i1(h3()3(l""9""J)r