o
    ¹ÈÏiˆ  ã                   @   s^   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„ deƒZ
G dd„ de	ƒZdgZd	S )
z&Fast Video processor class for VJEPA2.é   )ÚIMAGENET_DEFAULT_MEANÚIMAGENET_DEFAULT_STDÚPILImageResampling)ÚUnpackÚVideosKwargs)ÚBaseVideoProcessorc                   @   s   e Zd ZdS )ÚVJEPA2VideoProcessorInitKwargsN)Ú__name__Ú
__module__Ú__qualname__© r   r   úf/home/ubuntu/.local/lib/python3.10/site-packages/transformers/models/vjepa2/video_processing_vjepa2.pyr      s    r   c                       s^   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eZdgZdee f‡ fdd„Z‡  ZS )	ÚVJEPA2VideoProcessorÚshortest_edgeg’$I’$Ir@é   TÚpixel_values_videosÚkwargsc                    sj   |  dd¡}t|tƒst|tƒrd|vrtdƒ‚|d }t|d d ƒ}d|i|d< tƒ jdi |¤Ž d S )	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__r   r   r   &   s   
zVJEPA2VideoProcessor.__init__)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   Ú__classcell__r   r   r    r   r      s    r   N)Ú__doc__Úimage_utilsr   r   r   Úprocessing_utilsr   r   Úvideo_processing_utilsr   r   r   Ú__all__r   r   r   r   Ú<module>   s   
