+
    œ°³j   ã                   ó¬   € R t ^ RIt^ RIHt ^RIHt ^RI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 ] ! R	 R
]4      4       tR
.tR# )z#Image processor class for VideoMAE.N)Ú
functional)ÚTorchvisionBackend)ÚBatchFeature)Úgroup_images_by_shapeÚreorder_images)ÚIMAGENET_STANDARD_MEANÚIMAGENET_STANDARD_STDÚ
ImageInputÚPILImageResamplingÚSizeDictÚmake_nested_list_of_images)ÚImagesKwargsÚUnpack)Ú
TensorTypeÚauto_docstringc                   óÐ   a a€ ] tR t^"t o]P
                  t]t]	t
R^à/tRtR^àR^à/tRtRtRtRtV3R lV 3R lltRV3R lR	 llt]V3R
 lV 3R ll4       tV3R lR ltRtVtV ;t# )ÚVideoMAEImageProcessorÚshortest_edgeFÚheightÚwidthTc                ó0   <€ V ^8„  d   QhRS[ S[,          /# )é   Úkwargs)r   r   )ÚformatÚ__classdict__s   "€Úw/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/videomae/image_processing_videomae.pyÚ__annotate__Ú#VideoMAEImageProcessor.__annotate__/   s   ø€ ÷ #ñ #¡©Õ!5ñ #ó    c                ó(   <€ \         SV `  ! R/ VB  R # )N© )ÚsuperÚ__init__)Úselfr   Ú	__class__s   &,€r   r"   ÚVideoMAEImageProcessor.__init__/   s   ø€ Ü‰ÒÑ"˜6Ô"r   c                ó,   <€ V ^8„  d   QhRS[ RS[RS[ /# )r   ÚimagesÚexpected_ndimsÚreturn)r	   Úint)r   r   s   "€r   r   r   2   s(   ø€ ÷ Qñ Q±
ð QÉCð QÑXbñ Qr   c                ó   € \        WR 7      # ))r(   )r   )r#   r'   r(   s   &&&r   Ú_prepare_images_structureÚ0VideoMAEImageProcessor._prepare_images_structure2   s   € Ü)¨&ÔPÐPr   c                ó<   <€ V ^8„  d   QhRS[ RS[S[,          RS[/# )r   Úvideosr   r)   )r	   r   r   r   )r   r   s   "€r   r   r   6   s(   ø€ ÷ 4ñ 4¡ð 4±v¹lÕ7Kð 4ÑP\ñ 4r   c                ó&   <€ \         SV `  ! V3/ VB # )aX  
videos (`ImageInput`):
    Video or batch of videos to preprocess. Expects a single video (list of frames) or a batch of videos
    (list of list of frames). Each frame can be a PIL image, numpy array, or torch tensor with pixel values
    ranging from 0 to 255. If passing in frames with pixel values between 0 and 1, set `do_rescale=False`.
)r!   Ú
preprocess)r#   r/   r   r$   s   &&,€r   r1   Ú!VideoMAEImageProcessor.preprocess5   s   ø€ ô ‰wÒ! &Ñ3¨FÑ3Ð3r   c                ó  <€ V ^8„  d   QhRS[ 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[S[,          R,          RS[/# )r   r'   ztorch.TensorÚ	do_resizeÚsizeÚresamplez7PILImageResampling | tvF.InterpolationMode | int | NoneÚdo_center_cropÚ	crop_sizeÚ
do_rescaleÚrescale_factorÚdo_normalizeÚ
image_meanNÚ	image_stdÚdisable_groupingÚreturn_tensorsr)   )ÚlistÚboolr   ÚfloatÚstrr   r   )r   r   s   "€r   r   r   ?   sÜ   ø€ ÷ *]ñ *]á‘T˜.Õ)Õ*ð*]ñ ð*]ñ ð	*]ð
 Lð*]ñ ð*]ñ ð*]ñ ð*]ñ ð*]ñ ð*]ñ ™D¡�KÕ'¨$Õ.ð*]ñ ™4¡�;Õ&¨Õ-ð*]ñ  �+ð*]ñ ™jÕ(¨4Õ/ð*]ñ  
ñ!*]r   c           	     ó  € \        VR VR7      w  pp/ pVP                  4        F%  w  ppV'       d   V P                  VW44      pVVV&   K'  	  \        VVR R7      p\        VR VR7      w  pp/ pVP                  4        F9  w  ppV'       d   V P	                  VV4      pV P                  VWxWšV4      pVVV&   K;  	  \        VVR R7      pV Uu. uF  p\        P                  ! V4      NK  	  pp\        RV/VR7      # u upi )T)Ú	is_nestedr>   )rE   Úpixel_values)ÚdataÚtensor_type)	r   ÚitemsÚresizer   Úcenter_cropÚrescale_and_normalizeÚtorchÚstackr   )r#   r'   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r   Úgrouped_imagesÚgrouped_images_indexÚresized_images_groupedÚshapeÚstacked_imagesÚresized_imagesÚprocessed_images_groupedÚprocessed_imagesÚvideo_framesrF   s   &&&&&&&&&&&&&&,          r   Ú_preprocessÚ"VideoMAEImageProcessor._preprocess?   s1  € ô" 0EØ˜dÐ5Eô0
Ñ,ˆÐ,ð "$ÐØ%3×%9Ñ%9Ö%;Ñ!ˆE�>ßØ!%§¡¨^¸TÓ!L�Ø,:Ð" 5Ó)ñ &<ô (Ð(>Ð@TÐ`dÔeˆä/DØ dÐ=Mô0
Ñ,ˆÐ,ð $&Ð Ø%3×%9Ñ%9Ö%;Ñ!ˆE�>ßØ!%×!1Ñ!1°.À)Ó!L�Ø!×7Ñ7Ø 
¸LÐV_óˆNð /=Ð$ UÓ+ñ &<ô *Ð*BÐDXÐdhÔiÐñ GWÓWÑFV°lœŸš LÖ1ÑFVˆÐWÜ .°,Ð!?È^Ô\Ð\ùò Xs   ÃC=r    )é   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r
   ÚBILINEARr6   r   r<   r   r=   r5   Údefault_to_squarer8   r4   r7   r9   r;   r"   r,   r   r1   rX   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r$   r   s   @@r   r   r   "   s�   ù‡ € à!×*Ñ*€HØ'€JØ%€IØ˜SÐ!€DØÐØ˜3 ¨Ð-€IØ€IØ€NØ€JØ€L÷#ó #÷Qò Qð ÷4ò 4ó ð4÷*]÷ *]ð *]r   r   )Ú__doc__rM   Útorchvision.transforms.v2r   ÚtvFÚimage_processing_backendsr   Úimage_processing_utilsr   Úimage_transformsr   r   Úimage_utilsr   r   r	   r
   r   r   Úprocessing_utilsr   r   Úutilsr   r   r   Ú__all__r    r   r   Ú<module>rn      sX   ðñ *ã Ý 7å ;Ý 2ß E÷÷ ÷ 5ß /ð ôF]Ð/ó F]ó ðF]ðR $Ð
$‚r   