+
    œ°³jsR  ã                   óB  a € R t,0 t R t^ RIt^ RIt^ RIHt ^ RIHt ^ RIH	t	 ^ RI
t
^ RIHt ^RIHt ^RIHtHt ^RIHtHt ^R	IHtHtHtHtHtHt ^R
IHtHt ^RI H!t!H"t" ]t#]t$. ROt%. ROt&Rt'] ^ k  ! R R]RR7      t( ! R R4      t)]" ! R R]4      4       t*R.t+R# )z Image processor class for Flava.N)ÚIterable)Ú	lru_cache)ÚAny)Ú
functional)ÚTorchvisionBackend)ÚBatchFeatureÚget_size_dict)Úgroup_images_by_shapeÚreorder_images)ÚOPENAI_CLIP_MEANÚOPENAI_CLIP_STDÚChannelDimensionÚ
ImageInputÚPILImageResamplingÚSizeDict)ÚImagesKwargsÚUnpack)Ú
TensorTypeÚauto_docstringgš™™™™™¹?c                   ó0   a € ] tR t^3t o RtV 3R ltRtV tR# )ÚFlavaImageProcessorKwargsaå  
return_image_mask (`bool`, *optional*, defaults to `False`):
    Whether to return the image mask. Can be overridden by the `return_image_mask` parameter in `preprocess`.
input_size_patches (`int`, *optional*, defaults to `14`):
    Number of patches in the image in height and width direction. 14x14 = 196 total patches. Can be overridden
    by the `input_size_patches` parameter in `preprocess`.
total_mask_patches (`int`, *optional*, defaults to `75`):
    Total number of patches that should be masked. Can be overridden by the `total_mask_patches` parameter in
    `preprocess`.
mask_group_min_patches (`int`, *optional*, defaults to `16`):
    Minimum number of patches that should be masked. Can be overridden by the `mask_group_min_patches`
    parameter in `preprocess`.
mask_group_max_patches (`int`, *optional*):
    Maximum number of patches that should be masked. Can be overridden by the `mask_group_max_patches`
    parameter in `preprocess`.
mask_group_min_aspect_ratio (`float`, *optional*, defaults to `0.3`):
    Minimum aspect ratio of the mask window. Can be overridden by the `mask_group_min_aspect_ratio` parameter
    in `preprocess`.
mask_group_max_aspect_ratio (`float`, *optional*):
    Maximum aspect ratio of the mask window. Can be overridden by the `mask_group_max_aspect_ratio` parameter
    in `preprocess`.
return_codebook_pixels (`bool`, *optional*, defaults to `False`):
    Whether to return the codebook pixel values.
codebook_do_resize (`bool`, *optional*, defaults to `True`):
    Whether to resize the input for codebook to a certain. Can be overridden by the `codebook_do_resize`
    parameter in `preprocess`. `codebook_size`.
codebook_size (`dict[str, int]`, *optional*, defaults to `{"height": 224, "width": 224}`):
    Resize the input for codebook to the given size. Can be overridden by the `codebook_size` parameter in
    `preprocess`.
codebook_resample (`PILImageResampling`, *optional*, defaults to `PILImageResampling.LANCZOS`):
    Resampling filter to use if resizing the codebook image. With torchvision < 0.27, LANCZOS is not
    supported for torch Tensors and BICUBIC is used as the closest alternative. Can be overridden by the
    `codebook_resample` parameter in `preprocess`.
codebook_do_center_crop (`bool`, *optional*, defaults to `True`):
    Whether to crop the input for codebook at the center. If the input size is smaller than
    `codebook_crop_size` along any edge, the image is padded with 0's and then center cropped. Can be
    overridden by the `codebook_do_center_crop` parameter in `preprocess`.
codebook_crop_size (`dict[str, int]`, *optional*, defaults to `{"height": 224, "width": 224}`):
    Desired output size for codebook input when applying center-cropping. Can be overridden by the
    `codebook_crop_size` parameter in `preprocess`.
codebook_do_rescale (`bool`, *optional*, defaults to `True`):
    Whether to rescale the input for codebook by the specified scale `codebook_rescale_factor`. Can be
    overridden by the `codebook_do_rescale` parameter in `preprocess`.
codebook_rescale_factor (`int` or `float`, *optional*, defaults to `1/255`):
    Defines the scale factor to use if rescaling the codebook image. Can be overridden by the
    `codebook_rescale_factor` parameter in `preprocess`.
codebook_do_map_pixels (`bool`, *optional*, defaults to `True`):
    Whether to map the pixel values of the codebook input to (1 - 2e)x + e. Can be overridden by the
    `codebook_do_map_pixels` parameter in `preprocess`.
codebook_do_normalize (`bool`, *optional*, defaults to `True`):
    Whether or not to normalize the input for codebook with `codebook_image_mean` and `codebook_image_std`. Can
    be overridden by the `codebook_do_normalize` parameter in `preprocess`.
codebook_image_mean (`Optional[Union[float, Iterable[float]]]`, *optional*, defaults to `[0, 0, 0]`):
    The sequence of means for each channel, to be used when normalizing images for codebook. Can be overridden
    by the `codebook_image_mean` parameter in `preprocess`.
codebook_image_std (`Optional[Union[float, Iterable[float]]]`, *optional*, defaults to `[0.5, 0.5, 0.5]`):
    The sequence of standard deviations for each channel, to be used when normalizing images for codebook. Can
    be overridden by the `codebook_image_std` parameter in `preprocess`.
c                óz  <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[;R&   S[ ;R&   S[ ;R	&   S[S[S[3,          ;R
&   S[;R&   S[ ;R&   S[S[S[3,          ;R&   S[ ;R&   S[S[,          ;R&   S[ ;R&   S[ ;R&   S[S[S[,          ,          ;R&   S[S[S[,          ,          ;R&   # )é   Úreturn_image_maskÚinput_size_patchesÚtotal_mask_patchesÚmask_group_min_patchesÚmask_group_max_patchesÚmask_group_min_aspect_ratioÚmask_group_max_aspect_ratioÚreturn_codebook_pixelsÚcodebook_do_resizeÚcodebook_sizeÚcodebook_resampleÚcodebook_do_center_cropÚcodebook_crop_sizeÚcodebook_do_rescaleÚcodebook_rescale_factorÚcodebook_do_map_pixelsÚcodebook_do_normalizeÚcodebook_image_meanÚcodebook_image_std)ÚboolÚintÚfloatÚdictÚstrr   )ÚformatÚ__classdict__s   "€Úq/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/flava/image_processing_flava.pyÚ__annotate__Ú&FlavaImageProcessorKwargs.__annotate__3   sH  ø‡ ‚ ñ| Ññ} ñ~ Ññ ñ@ ÑñA ñB  ÑñC ñD  ÑñE ñF "'Ñ&ñG ñH "'Ñ&ñI ñL !Ñ ñM ñN ÑñO ñP ™™S˜•>Ñ!ñQ ñR ÑñS ñT "Ñ!ñU ñV ™S¡#˜X�Ñ&ñW ñX ÑñY ñZ !¡5�[Ñ(ñ[ ñ\ !Ñ ñ] ñ^  Ññ_ ñ` ¡©%¥Õ0Ñ0ña ñb ¡©¥Õ/Ñ/òc ó    © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__©r2   s   @r3   r   r   3   s   ø‡ € ñ:÷ ƒ r6   r   F)Útotalc                   óN   a € ] tR t^ˆt o R	V 3R lR lltR tR tR tR tRt	V t
R# )
ÚFlavaMaskingGeneratorNc                óŽ   <€ V ^8„  d   QhRS[ S[S[ S[ 3,          ,          RS[ RS[ R,          RS[ RS[R,          RS[R,          /# )r   Ú
input_sizer   r   Nr   r   r   )r-   Útupler.   )r1   r2   s   "€r3   r4   Ú"FlavaMaskingGenerator.__annotate__‰   sh   ø€ ÷ oñ oá™%¡¡S �/Õ)ðoñ  ðoñ !$ d¥
ð	oñ
 !$ðoñ &+¨T¥\ðoñ &+¨T¥\ñor6   c                ób  € \        V\        4      '       g   V3^,          pVw  V n        V n        V P                  V P                  ,          V n        W n        W@n        Vf   TMTV n        T;'       g
    ^V,          p\        P                  ! V4      \        P                  ! V4      3V n
        R# )r   N)Ú
isinstancerF   ÚheightÚwidthÚnum_patchesr   r   r   ÚmathÚlogÚlog_aspect_ratio)ÚselfrE   r   r   r   r   r   s   &&&&&&&r3   Ú__init__ÚFlavaMaskingGenerator.__init__‰   s’   € ô ˜*¤e×,Ò,Ø$˜¨Õ*ˆJØ",ÑˆŒ�T”ZàŸ;™;¨¯©Õ3ˆÔØ"4Ôà&<Ô#Ø<RÒ<ZÑ&8Ð`vˆÔ#à&A×&dÐ&dÀQÐIdÕEdÐ#Ü!%§¢Ð*EÓ!FÌÏÊÐQlÓHmÐ nˆÖr6   c           	     óÎ   € R V P                   V P                  V P                  V P                  V P                  V P
                  ^ ,          V P
                  ^,          3,          pV# )z<MaskingGenerator(%d, %d -> [%d ~ %d], max = %d, %.3f ~ %.3f))rJ   rK   r   r   r   rO   )rP   Úrepr_strs   & r3   Ú__repr__ÚFlavaMaskingGenerator.__repr__Ÿ   s^   € ØQØ�K‰KØ�J‰JØ×'Ñ'Ø×'Ñ'Ø×#Ñ#Ø×!Ñ! !Õ$Ø×!Ñ! !Õ$ðU
õ 
ˆð ˆr6   c                ó2   € V P                   V P                  3# ©N)rJ   rK   )rP   s   &r3   Ú	get_shapeÚFlavaMaskingGenerator.get_shape«   s   € Ø�{‰{˜DŸJ™JÐ&Ð&r6   c           	     óº  € ^ p\        ^
4       EFÈ  p\        P                  ! V P                  V4      p\        P
                  ! \        P                  ! V P                  !  4      p\        \        \        P                  ! WV,          4      4      4      p\        \        \        P                  ! WV,          4      4      4      pW€P                  8  g   KÅ  WpP                  8  g   K×  \        P                  ! ^ V P                  V,
          4      p	\        P                  ! ^ V P                  V,
          4      p
WW—,           1WªV,           13,          P                  4       p^ Wx,          V,
          u;8  d   V8:  dT   M MPWW—,           1WªV,           13,          ^ 8H  p^WW—,           1WªV,           13,          V&   W<P                  4       ,          pV^ 8”  g   EKÈ   V# 	  V# )é    )ÚrangeÚrandomÚuniformr   rM   ÚexprO   r-   ÚroundÚsqrtrK   rJ   ÚrandintÚsum)rP   ÚmaskÚmax_mask_patchesÚdeltaÚ_attemptÚtarget_areaÚaspect_ratiorJ   rK   ÚtopÚleftÚ
num_maskedÚ	zeros_poss   &&&          r3   Ú_maskÚFlavaMaskingGenerator._mask®   sb  € ØˆÜ˜bŸ	ˆHÜ Ÿ.š.¨×)DÑ)DÐFVÓWˆKÜŸ8š8¤F§N¢N°D×4IÑ4IÑ$JÓKˆLÜœœtŸyšy¨Õ)CÓDÓEÓFˆFÜœœdŸiši¨Õ(BÓCÓDÓEˆEØ—z‘zÖ! f¯{©{Ö&:Ü—n’n Q¨¯©°fÕ(<Ó=�Ü—~’~ a¨¯©°eÕ);Ó<�à!¨­Ð"4°dÀE½\Ð6IÐ"IÕJ×NÑNÓP�
à�v•~¨
Õ2ÖFÐ6F×FØ $¨3­<Ð%7¸ÀuÅÐ9LÐ%LÕ MÐQRÑ R�IØOP�D˜s�|Ð+¨T¸5µLÐ-@Ð@ÕAÀ)ÑLØŸ]™]›_Õ,�Eà˜1—9ØØˆñ% "ð$ ˆr6   c                ó,  € \         P                  ! V P                  4       \         P                  R 7      p^ pW P                  8  dN   V P                  V,
          p\        W0P                  4      pV P                  W4      pV^ 8X  d    V# W$,          pK]  V# ))Údtype)ÚtorchÚzerosrY   r-   r   Úminr   ro   )rP   re   Ú
mask_countrf   rg   s   &    r3   Ú__call__ÚFlavaMaskingGenerator.__call__Ä   s   € Ü�{Š{˜4Ÿ>™>Ó+´5·9±9Ô=ˆØˆ
Ø×2Ñ2Ô2Ø#×6Ñ6¸ÕCÐÜ"Ð#3×5PÑ5PÓQÐà—J‘J˜tÓ6ˆEØ˜ŒzØð ˆð Õ#’
àˆr6   )rJ   rO   r   r   rL   r   rK   )é   éK   Né   ç333333Ó?N)r8   r9   r:   r;   rQ   rU   rY   ro   rw   r>   r?   r@   s   @r3   rC   rC   ˆ   s*   ø‡ € ÷oò oò,
ò'ò÷,ð r6   rC   c                   ó¬  a a€ ] tR t^Ôt o]t]P                  t]	t
]tR^àR^à/tR^àR^à/tRtRtRtRtRt^t^Kt^tRtRtRtRtRtR^pR^p/t]P:                  tRtR^pR^p/t Rt!Rt"Rt#Rt$]%t&]'t(V3R lV 3R llt)]*V3R	 lV 3R
 ll4       t+],V3R lV 3R ll4       t-].V3R lR l4       t/V3R lR lt0RV3R lV 3R lllt1V3R lR lt2V3R lR lt3Rt4Vt5V ;t6# )ÚFlavaImageProcessorrJ   rK   TFNr|   c                ó0   <€ V ^8„  d   QhRS[ S[,          /# )r   Úkwargs)r   r   )r1   r2   s   "€r3   r4   Ú FlavaImageProcessor.__annotate__÷   s   ø€ ÷ #ñ #¡Ñ(AÕ!Bñ #r6   c                ó(   <€ \         SV `  ! R/ VB  R # )Nr7   )ÚsuperrQ   )rP   r€   Ú	__class__s   &,€r3   rQ   ÚFlavaImageProcessor.__init__÷   s   ø€ Ü‰ÒÑ"˜6Ô"r6   c                ó<   <€ V ^8„  d   QhRS[ RS[S[,          RS[/# )r   Úimagesr€   Úreturn)r   r   r   r   )r1   r2   s   "€r3   r4   r�   û   s)   ø€ ÷ 4ñ 4¡ð 4±vÑ>WÕ7Xð 4Ñ]iñ 4r6   c                ó&   <€ \         SV `  ! V3/ VB # rX   )rƒ   Ú
preprocess)rP   r‡   r€   r„   s   &&,€r3   rŠ   ÚFlavaImageProcessor.preprocessú   s   ø€ ä‰wÒ! &Ñ3¨FÑ3Ð3r6   c                ó6   <€ V ^8„  d   QhRS[ S[S[3,          /# )r   Úimage_processor_dict)r/   r0   r   )r1   r2   s   "€r3   r4   r�   ÿ   s   ø€ ÷ 
Añ 
A©T±#±s°(­^ñ 
Ar6   c                ó²   <€ VP                  4       pRV9   d   VP                  R4      VR&   RV9   d   VP                  R4      VR&   \        SV `  ! V3/ VB # )zÝ
Overrides the `from_dict` method from the base class to make sure parameters are updated if image processor is
created using from_dict and kwargs e.g. `FlavaImageProcessor.from_pretrained(checkpoint, codebook_size=600)`
r"   r%   )ÚcopyÚpoprƒ   Ú	from_dict)Úclsr�   r€   r„   s   &&,€r3   r‘   ÚFlavaImageProcessor.from_dictþ   sd   ø€ ð  4×8Ñ8Ó:ÐØ˜fÔ$Ø4:·J±J¸Ó4OÐ  Ñ1Ø 6Ô)Ø9?¿¹ÐDXÓ9YÐ Ð!5Ñ6Ü‰wÒ Ð!5Ñ@¸Ñ@Ð@r6   c                ó    <€ V ^8„  d   QhRS[ /# )r   rˆ   ©rC   )r1   r2   s   "€r3   r4   r�     s   ø€ ÷ 
ñ 
ñ 
ñ
r6   c           	     ó$   € \        VVVVVVR 7      # ))rE   r   r   r   r   r   r•   )rP   r   r   r   r   r   r   s   &&&&&&&r3   Úmasking_generatorÚ%FlavaImageProcessor.masking_generator  s#   € ô %Ø)Ø1Ø#9Ø#9Ø(CØ(Cô
ð 	
r6   c                ó"   <€ V ^8„  d   QhRRRR/# )r   Úimageútorch.Tensorrˆ   r7   )r1   r2   s   "€r3   r4   r�     s   ø€ ÷ Gñ G ð G°>ñ Gr6   c                óN   € ^^\         ,          ,
          V,          \         ,           # )é   )ÚLOGIT_LAPLACE_EPS)rP   rš   s   &&r3   Ú
map_pixelsÚFlavaImageProcessor.map_pixels  s   € Ø�AÔ)Õ)Õ)¨UÕ2Ô5FÕFÐFr6   c                óÄ  <€ V ^8„  d   QhRS[ S[S[ ,          ,          S[S[S[ 3,          ,          S[,          R,          RS[ S[S[ ,          ,          S[S[S[ 3,          ,          S[,          R,          RS[R,          RS[S[S[,          ,          R,          RS[S[S[,          ,          R,          RS[ S[S[ ,          ,          S[S[S[ 3,          ,          S[,          R,          RS[ S[S[ ,          ,          S[S[S[ 3,          ,          S[,          R,          R	S[S[S[,          ,          R,          R
S[S[S[,          ,          R,          RRRS[R,          RS[/# )r   ÚsizeNÚ	crop_sizeÚdefault_to_squareÚ
image_meanÚ	image_stdr"   r%   r*   r+   r#   ú7PILImageResampling | tvF.InterpolationMode | int | NoneÚdata_formatrˆ   )	r-   r   r/   r0   r   r,   r.   Úlistr   )r1   r2   s   "€r3   r4   r�   !  sE  ø€ ÷ ,ñ ,á‘H™S•MÕ!¡D©©c¨¥NÕ2±XÕ=ÀÕDð,ñ ™¡#�Õ&©©c±3¨h­Õ7¹(ÕBÀTÕIð,ñ   $�;ð	,ñ
 ™D¡�KÕ'¨$Õ.ð,ñ ™4¡�;Õ&¨Õ-ð,ñ ™X¡c�]Õ*©T±#±s°(­^Õ;¹hÕFÈÕMð,ñ  ¡(©3¥-Õ/±$±s¹C°xµ.Õ@Á8ÕKÈdÕRð,ñ #¡T©%¥[Õ0°4Õ7ð,ñ "¡D©¥KÕ/°$Õ6ð,ð Uð,ñ &¨Õ,ð,ñ 
ñ,r6   c                óª  <€ \         SV `  ! RRVRVRVRVRVRV/VB pVe,   \        V\        4      '       g   \        R/ \	        WcR7      B pVe-   \        V\        4      '       g   \        R/ \	        VRR	7      B p\        V\
        4      '       d   \        V4      p\        V	\
        4      '       d   \        V	4      p	WlR
&   W|R&   WŒR&   WœR&   W¬R&   V# )zŠ
Update kwargs that need further processing before being validated
Can be overridden by subclasses to customize the processing of kwargs.
r¢   r£   r¤   r¥   r¦   r¨   )r¢   r¤   r%   )Ú
param_namer"   r*   r+   r#   r7   )rƒ   Ú_standardize_kwargsrI   r   r   r©   rF   )rP   r¢   r£   r¤   r¥   r¦   r"   r%   r*   r+   r#   r¨   r€   r„   s   &&&&&&&&&&&&,€r3   r¬   Ú'FlavaImageProcessor._standardize_kwargs!  s  ø€ ô& ‘Ò,ñ 
Øð
àð
ð 0ð
ð "ð	
ð
  ð
ð $ð
ð ñ
ˆð Ò$¬Z¸Äx×-PÒ-PÜ$Ñn¤}¸-Ô'mÑnˆMØÒ)´*Ð=OÔQY×2ZÒ2ZÜ!)Ñ!o¬MÐ:LÐYmÔ,nÑ!oÐÜÐ)¬4×0Ò0Ü"'Ð(;Ó"<ÐÜÐ(¬$×/Ò/Ü!&Ð'9Ó!:Ðà"/ˆÑØ'9Ð#Ñ$Ø(;Ð$Ñ%Ø'9Ð#Ñ$à&7Ð"Ñ#àˆr6   c                ó  <€ V ^8„  d   QhRS[ R,          RS[RS[RRRS[RS[R	S[R
S[RS[RS[RS[S[ S[,          ,          R,          RS[S[ S[,          ,          R,          RS[R,          RS[S[,          R,          RR/# )r   r‡   r›   Ú	do_resizer¢   Úresampler§   Údo_center_cropr£   Ú
do_rescaleÚrescale_factorÚdo_normalizeÚdo_map_pixelsr¥   Nr¦   Údisable_groupingÚreturn_tensorsrˆ   )r©   r,   r   r.   r0   r   )r1   r2   s   "€r3   r4   r�   O  sÑ   ø€ ÷ + ñ + á�^Õ$ð+ ñ ð+ ñ ð	+ ð
 Lð+ ñ ð+ ñ ð+ ñ ð+ ñ ð+ ñ ð+ ñ ð+ ñ ™D¡�KÕ'¨$Õ.ð+ ñ ™4¡�;Õ&¨Õ-ð+ ñ  �+ð+ ñ ™jÕ(¨4Õ/ð+ ð  
ñ!+ r6   c           	     ó¾  € \        WR 7      w  pp/ pVP                  4        F&  w  ppV'       d   V P                  VW4R7      pVVV&   K(  	  \        VV4      p\        VVR 7      w  pp/ pVP                  4        FS  w  ppV'       d   V P	                  VV4      pV P                  VWxW›V4      pV
'       d   V P                  VR7      pVVV&   KU  	  \        VV4      pV# ))r¶   )rš   r¢   r°   )rš   )r	   ÚitemsÚresizer
   Úcenter_cropÚrescale_and_normalizerŸ   )rP   r‡   r¯   r¢   r°   r±   r£   r²   r³   r´   rµ   r¥   r¦   r¶   r·   Úgrouped_imagesÚgrouped_images_indexÚresized_images_groupedÚshapeÚstacked_imagesÚresized_imagesÚprocessed_images_groupedÚprocessed_imagess   &&&&&&&&&&&&&&&        r3   Ú_preprocess_imageÚ%FlavaImageProcessor._preprocess_imageO  sý   € ô$ 0EÀVÔ/oÑ,ˆÐ,Ø!#ÐØ%3×%9Ñ%9Ö%;Ñ!ˆE�>ßØ!%§¡°>È Ó!`�Ø,:Ð" 5Ó)ñ &<ô (Ð(>Ð@TÓUˆô 0EÀ^ÐfvÔ/wÑ,ˆÐ,Ø#%Ð Ø%3×%9Ñ%9Ö%;Ñ!ˆE�>ßØ!%×!1Ñ!1°.À)Ó!L�à!×7Ñ7Ø 
¸LÐV_óˆN÷ Ø!%§¡°~ Ó!F�Ø.<Ð$ UÓ+ñ &<ô *Ð*BÐDXÓYÐàÐr6   c          B      ó®  <€ V ^8„  d   QhRS[ R,          RS[RS[RRRS[RS[R	S[R
S[RS[RS[S[ S[,          ,          R,          RS[S[ S[,          ,          R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RRRS[R,          RS[R,          RS[R,          RS[R,          RS[R,          RS[R,          R S[S[ S[,          ,          R,          R!S[S[ S[,          ,          R,          R"S[R,          R#S[S[,          R,          R$S[/!# )%r   r‡   r›   r¯   r¢   r°   r§   r±   r£   r²   r³   r´   r¥   Nr¦   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r¶   r·   rˆ   )r©   r,   r   r.   r-   r0   r   r   )r1   r2   s   "€r3   r4   r�   |  s2  ø€ ÷ YCñ YCá�^Õ$ðYCñ ðYCñ ð	YCð
 LðYCñ ðYCñ ðYCñ ðYCñ ðYCñ ðYCñ ™D¡�KÕ'¨$Õ.ðYCñ ™4¡�;Õ&¨Õ-ðYCñ   $�;ðYCñ   $�JðYCñ    $�Jð!YCñ" !$ d¥
ð#YCñ$ !$ d¥
ð%YCñ& &+¨T¥\ð'YCñ( &+¨T¥\ð)YCñ, !% t¥ð-YCñ. ! 4�Kð/YCñ0   $•ð1YCð2 Uð3YCñ4 "&¨¥ð5YCñ6 % t�Oð7YCñ8 " D�[ð9YCñ: "'¨¥ð;YCñ< !% t¥ð=YCñ>  $ d�{ð?YCñ@ #¡T©%¥[Õ0°4Õ7ðAYCñB "¡D©¥KÕ/°$Õ6ðCYCñD  �+ðEYCñF ™jÕ(¨4Õ/ðGYCñJ 
ñKYCr6   c!                ó\  € V P                  VVVVVVVVV	R V
VVV R7      p"RV"/p#V'       d%   V P                  VVVVVVVVVVVVVV R7      p$V$V#R&   V'       dE   V P                  VVVVVVR7      p%\        \        V4      4       U&u. uF
  p&V%! 4       NK  	  p'p&V'V#R&   \	        V#V R7      # u up&i )F)r‡   r¯   r¢   r°   r±   r£   r²   r³   r´   rµ   r¥   r¦   r¶   r·   Úpixel_valuesÚcodebook_pixel_values)r   r   r   r   r   r   Úbool_masked_pos)ÚdataÚtensor_type)rÅ   r—   r]   Úlenr   )(rP   r‡   r¯   r¢   r°   r±   r£   r²   r³   r´   r¥   r¦   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r¶   r·   r€   rÄ   rÌ   Úcodebook_processed_imagesÚmask_generatorÚ_Úmaskss(   &&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&,      r3   Ú_preprocessÚFlavaImageProcessor._preprocess|  s  € ðL  ×1Ñ1ØØØØØ)ØØ!Ø)Ø%ØØ!ØØ-Ø)ð 2ó 
Ðð" Ð,ð
ˆ÷ "Ø(,×(>Ñ(>ØØ,Ø"Ø*Ø6Ø,Ø.Ø6Ø2Ø4Ø.Ø,Ø!1Ø-ð )?ó )Ð%ð  -FˆDÐ(Ñ)çØ!×3Ñ3Ø#5Ø#5Ø'=Ø'=Ø,GØ,Gð 4ó ˆNô 05´S¸³[Ô/AÓBÑ/A¨!‘^Ö%Ñ/AˆEÐBØ&+ˆDÐ"Ñ#ä °>ÔBÐBùò Cs   ÂB)r7   gp?)NNNNNNNNNNN)7r8   r9   r:   r;   r   Úvalid_kwargsr   ÚBICUBICr°   ÚFLAVA_IMAGE_MEANr¥   ÚFLAVA_IMAGE_STDr¦   r¢   r£   r¯   r±   r²   r´   r   r   r   r   r   r   r   r    r!   r"   ÚLANCZOSr#   r$   r%   r&   r'   r(   r)   ÚFLAVA_CODEBOOK_MEANr*   ÚFLAVA_CODEBOOK_STDr+   rQ   r   rŠ   Úclassmethodr‘   r   r—   rŸ   r¬   rÅ   rÓ   r>   r?   Ú__classcell__)r„   r2   s   @@r3   r~   r~   Ô   sd  ù‡ € à,€LØ!×)Ñ)€HØ!€JØ€IØ�c˜7 CÐ(€DØ˜3 ¨Ð-€IØ€IØ€NØ€JØ€Lð ÐØÐØÐØÐØ!ÐØ"%ÐØ"&Ðà"ÐØÐØ˜s G¨SÐ1€MØ*×2Ñ2ÐØ"ÐØ" C¨°#Ð6ÐØÐØ%ÐØ!ÐØ ÐØ-ÐØ+Ð÷#ó #ð ÷4ò 4ó ð4ð ÷
Aò 
Aó ð
Að ÷
ó ð
÷$Gð G÷,õ ,÷\+ ð + ÷ZYC÷ YCð YCr6   r~   c                ó@   € V ^8„  d   Qh/ ^ \         9   d
   \        ;R&   # )r   rž   )Ú__conditional_annotations__r.   )r1   s   "r3   r4   r4      s   € × &Ñ &÷D Ò ”5Ñ òE 'r6   )ç        rà   rà   )ç      ð?rá   rá   )-rß   r<   rM   r^   Úcollections.abcr   Ú	functoolsr   Útypingr   rs   Útorchvision.transforms.v2r   ÚtvFÚimage_processing_backendsr   Úimage_processing_utilsr   r   Úimage_transformsr	   r
   Úimage_utilsr   r   r   r   r   r   Úprocessing_utilsr   r   Úutilsr   r   r×   rØ   rÚ   rÛ   rž   r   rC   r~   Ú__all__r4   )rß   s   @r3   Ú<module>rî      sª   øðö 'ã Û Ý $Ý Ý ã Ý 7å ;ß Aß E÷÷ ÷ 5÷ð $Ð Ø!€Ú%Ð Ú$Ð ØÐ Ó ôQ0 °Eõ Q0÷jIñ IðX ô@CÐ,ó @Có ð@CðF !Ð
!‚r6   