
    ^j                     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 Vivit.   )IMAGENET_STANDARD_MEANIMAGENET_STANDARD_STDPILImageResampling)BaseVideoProcessorc                   |     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d	 Z fd
Z xZS )VivitVideoProcessorshortest_edge   F   )heightwidthTg?pixel_valuesc                 |    |r|j                         |z  dz
  }|r!| j                  |j                         ||      }|S )N   )float	normalize)selfimages
do_rescalerescale_factordo_normalize
image_mean	image_stds          {/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/models/vivit/video_processing_vivit.pyrescale_and_normalizez)VivitVideoProcessor.rescale_and_normalize$   s:    \\^n4q8F^^FLLNJ	JF    c                 P    t        |   |fi |}|j                  d      |d<   |S )Npixel_values_videosr   )super
preprocesspop)r   videoskwargsbatch	__class__s       r   r    zVivitVideoProcessor.preprocess,   s0    "64V4 %		*? @nr   )__name__
__module____qualname__r   BILINEARresampler   r   r   r   sizedefault_to_square	crop_size	do_resizedo_center_cropr   r   r   do_convert_rgbdo_sample_framesmodel_input_namesr   r    __classcell__)r%   s   @r   r   r      sn    !**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>r9      s*    ' \ \ 8, < !
!r   