
    3j                     D    S r SSKJrJrJr  SSKJr   " S S\5      rS/rg)z Video processor class for Vivit.   )IMAGENET_STANDARD_MEANIMAGENET_STANDARD_STDPILImageResampling)BaseVideoProcessorc                      ^  \ rS rSr\R
                  r\r\	r
SS0rSrSSS.rSrSrSrSrSrSrSrS	/rS
 rU 4S jrSrU =r$ )VivitVideoProcessor   shortest_edge   F   )heightwidthTg?pixel_valuesc                     U(       a  UR                  5       U-  S-
  nU(       a   U R                  UR                  5       XV5      nU$ )N   )float	normalize)selfimages
do_rescalerescale_factordo_normalize
image_mean	image_stds          j/home/wildlama/miniconda3/lib/python3.13/site-packages/transformers/models/vivit/video_processing_vivit.pyrescale_and_normalize)VivitVideoProcessor.rescale_and_normalize$   s8    \\^n4q8F^^FLLNJJF    c                 R   > [         TU ]  " U40 UD6nUR                  S5      US'   U$ )Npixel_values_videosr   )super
preprocesspop)r   videoskwargsbatch	__class__s       r   r"   VivitVideoProcessor.preprocess,   s0    "64V4 %		*? @nr    )__name__
__module____qualname____firstlineno__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"   __static_attributes____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   r   <module>r>      s*    ' \ \ 8, < !
!r   