
    qi.                     D    d Z ddlmZmZmZ ddlmZ  G d de      ZdgZy)z#Video processor class for VideoMAE.   )IMAGENET_STANDARD_MEANIMAGENET_STANDARD_STDPILImageResampling)BaseVideoProcessorc                   v     e Zd Zej                  ZeZeZ	ddiZ
dZdddZdZdZdZdZdZdZdZdgZ fdZ xZS )	VideoMAEVideoProcessorshortest_edge   F)heightwidthTgp?pixel_valuesc                 P    t        |   |fi |}|j                  d      |d<   |S )Npixel_values_videosr   )super
preprocesspop)selfvideoskwargsbatch	__class__s       h/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/videomae/video_processing_videomae.pyr   z!VideoMAEVideoProcessor.preprocess$   s0    "64V4 %		*? @n    )__name__
__module____qualname__r   BILINEARresampler   
image_meanr   	image_stdsizedefault_to_square	crop_size	do_resizedo_center_crop
do_rescalerescale_factordo_normalizedo_convert_rgbdo_sample_framesmodel_input_namesr   __classcell__)r   s   @r   r   r      si    !**H'J%IS!D-IINJNLN'( r   r   N)	__doc__image_utilsr   r   r   video_processing_utilsr   r   __all__ r   r   <module>r2      s*    * \ \ 8/ , $
$r   