
    Z j                     T    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/rg)z&Fast Video processor class for VJEPA2.   )IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STDPILImageResampling)UnpackVideosKwargs)BaseVideoProcessorc                   ~   ^  \ rS rSr\R
                  r\r\	r
S\" S5      0rSrSrSrSrSrS\\   4U 4S jjrSrU =r$ )	VJEPA2VideoProcessor   shortest_edgeg$I$Ir@   T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   size )get
isinstanceintdict
ValueErrorsuper__init__)selfr   r   resize_size	__class__s       ڃ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/vjepa2/video_processing_vjepa2.pyr   VJEPA2VideoProcessor.__init__    s{    JJ{C0	)S))i..()2K !cdd!(+I)c/C/0);7v"6"    r   )__name__
__module____qualname____firstlineno__r   BILINEARresampler   
image_meanr   	image_stdr   r   r   	do_resize
do_rescaledo_center_cropdo_normalizer   r   r   __static_attributes____classcell__)r   s   @r   r
   r
      sV    !**H&J$IS12DIIJNL#!5 # #r!   r
   N)__doc__image_utilsr   r   r   processing_utilsr   r   video_processing_utilsr   r
   __all__r   r!   r   <module>r5      s-    - Z Z 4 8#- #, "
"r!   