9jddlmZddlmZmZmZmZddlZddlmZddl 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||_yN5Argument transforms should be a sequence of callablesPass at least one transform)super__init__ isinstancer TypeError ValueErrorr selfr __class__s d/media/conek/DATA/Code/OCR/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|s td||_d|cxkrdkstdtd||_ y)NzJArgument transforms should be a sequence of callables or a `nn.ModuleList`rgg?z@`p` should be a floating point value in the interval [0.0, 1.0].) rrrrr ModuleListrrr r:)rr r:rs rrzRandomApply.__init__Usr *x&?@hi i:; ;$qC_` ` _` `rc4|j|jdS)Nr r:r>)rs r _extract_params_for_v1_transformz,RandomApply._extract_params_for_v1_transformbs"ooDFF;;rrct|dkD}tjd|jk\r |r|S|dS|jD]}||}|r|n|f}S)Nrr)r torchrandr:r r!s rr%zRandomApply.forwardesef+/ ::a=DFF ",6 ;&) ; @I(G /WgZF @rcrg}|jD]}|jd|dj|Sr'r)r,s rr/zRandomApply.extra_reprpr0r)g?)r1r2r3r4 _transformsr9_v1_transform_clsrrrrr<floatrdictr5rr?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:r cht|ts td|s td|dgt |z}n:t |t |k7r#tdt |dt |t |||_t|}|Dcgc]}||z  c}|_ ycc}w)Nrrrz4Length of p doesn't match the number of transforms: z != ) rrrrr rrr 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) r1r2r3r4rrrlistrFrrr%r6r7s@rrIrIwsO $(.X&. DK .  .("s"s"rrIcBeZdZdZdeeddffd 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 Nct|ts td|s tdt|||_yr)rrrrrrr rs rrzRandomOrder.__init__s:*h/ST T:; ; $rrct|dkD}tjt|jD]}|j|}||}|r|n|f}Sr)r rArandpermr )rrr"rRr#r$s rr%zRandomOrder.forwards\f+/>>#doo"67 @C,I(G /WgZF @r) r1r2r3r4rrrrr%r6r7s@rrUrUs3%8H#5%$%ssrrU)collections.abcrtypingrrrrrAr torchvisionr rDtorchvision.transforms.v2r r r9rIrUrrr^sM$11 1/1(i1(h5()5(p#"9#"L)r