xhddlmZddlmZmZmZmZddlZddlmZddl m Z ddl m Z Gdde ZGd d e ZGd d e ZGd de Zy))Sequence)AnyCallableOptionalUnionN)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 /home/developers/rajanand/mypropertyqr-fmb-refixing-v2/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.forward1s@f+/ @I(G /WgZF @rcrg}|jD]}|jd|dj|SNz  r appendjoinr format_stringts r extra_reprzCompose.extra_repr8=  -A  4s , -yy''r) __name__ __module__ __qualname____doc__rrrrr#strr- __classcell__rs@rr r s?:%8H#5%$%ss(C(rr ceZdZdZej 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 pr Nct|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 rr8)rr r8rs rrzRandomApply.__init__Usc *x&?@hi i$qC_` ` _` `rc4|j|jdS)Nr r8r<)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)rtorchrandr8r rs rr#zRandomApply.forwardcsef+/ ::a=DFF ",6 ;&) ; @I(G /WgZF @rcrg}|jD]}|jd|dj|Sr%r'r*s rr-zRandomApply.extra_reprnr.r)g?)r/r0r1r2 _transformsr7_v1_transform_clsrrrrr:floatrdictr3rr=r#r-r4r5s@rr7r7?sx&$// 5();R]])J#K PU `d <$sCx.< s s (C(rr7cVeZdZdZ 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 r8r cNt|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 != ) rrrrrrrr sumr8)rr r8totalprobrs rrzRandomChoice.__init__s *h/ST T 9c*o%A Vs: &STWXYTZS[[_`cdn`o_pqr r $A+,-4$,--s B"rcttjtj|jd}|j |}||Sr)intr? multinomialtensorr8r )rridxr!s rr#zRandomChoice.forwards?%##ELL$8!<=OOC( &!!r)N) r/r0r1r2rrrlistrDrrr#r4r5s@rrGrGusO $(.X&. DK .  .&"s"s"rrGcBeZdZdZdeeddffd 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 r Ncft|ts tdt|||_y)Nr)rrrrrr rs rrzRandomOrder.__init__s+*h/ST T $rrct|dkD}tjt|jD]}|j|}||}|r|n|f}Sr)rr?randpermr )rrr rPr!r"s rr#zRandomOrder.forwards\f+/>>#doo"67 @C,I(G /WgZF @r) r/r0r1r2rrrrr#r4r5s@rrSrSs3%8H#5%$% ssrrS)collections.abcrtypingrrrrr?r torchvisionr rBtorchvision.transforms.v2r r r7rGrSrrr\sM$11 1/1(i1(h3()3(l""9""J)r