
    cCi                     j    S r SSKJrJrJr  SSKJrJr  SSKJ	r	   " S S\5      r
 " S S\	5      rS/rg	)
z&Fast Video processor class for VJEPA2.   )IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STDPILImageResampling)UnpackVideosKwargs)BaseVideoProcessorc                       \ rS rSrSrg)VJEPA2VideoProcessorInitKwargs    N)__name__
__module____qualname____firstlineno____static_attributes__r       l/home/james-whalen/.local/lib/python3.13/site-packages/transformers/models/vjepa2/video_processing_vjepa2.pyr
   r
      s    Cr   r
   c                      ^  \ rS rSr\R
                  r\r\	r
S\" S5      0rSrSrSrSrSr\rS/rS\\   4U 4S jjrS	rU =r$ )
VJEPA2VideoProcessor   shortest_edgeg$I$Ir@   Tpixel_values_videoskwargsc                    > UR                  SS5      n[        U[        5      (       d+  [        U[        5      (       a  SU;  a  [	        S5      eUS   n[        US-  S-  5      nSU0US'   [
        TU ]  " S0 UD6  g )	N	crop_sizer   heightz@crop_size must be an integer or a dictionary with a 'height' key   r   sizer   )get
isinstanceintdict
ValueErrorsuper__init__)selfr   r   resize_size	__class__s       r   r&   VJEPA2VideoProcessor.__init__&   s{    JJ{C0	)S))i..()2K !cdd!(+I)c/C/0);7v"6"r   r   )r   r   r   r   r   BILINEARresampler   
image_meanr   	image_stdr"   r   r   	do_resize
do_rescaledo_center_cropdo_normalizer
   valid_kwargsmodel_input_namesr   r&   r   __classcell__)r)   s   @r   r   r      se    !**H&J$IS12DIIJNL1L./#(F!G # #r   r   N)__doc__image_utilsr   r   r   processing_utilsr   r   video_processing_utilsr   r
   r   __all__r   r   r   <module>r;      s7    - Z Z 4 8 8\ 7#- #0 "
"r   