
    qi                     T    d Z ddlmZmZmZ ddlmZmZ ddlm	Z	  G d de	      Z
dgZy)z&Fast Video processor class for VJEPA2.   )IMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STDPILImageResampling)UnpackVideosKwargs)BaseVideoProcessorc                   r     e Zd Zej                  ZeZeZ	d e
d      iZdZdZdZdZdZdee   f fdZ xZS )VJEPA2VideoProcessorshortest_edgeg$I$Ir@   Tkwargsc                     |j                  dd      }t        |t              s$t        |t              rd|vrt	        d      |d   }t        |dz  dz        }d|i|d<   t        |   di | y )	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       d/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/vjepa2/video_processing_vjepa2.pyr   zVJEPA2VideoProcessor.__init__    su    JJ{C0	)S)i.()2K !cdd!(+I)c/C/0);7v"6"    )__name__
__module____qualname__r   BILINEARresampler   
image_meanr   	image_stdr   r   r   	do_resize
do_rescaledo_center_cropdo_normalizer   r   r   __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>r1      s-    - Z Z 4 8#- #, "
"r   