
    謜iݟ                        d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
mZ d dlZd dlmZmZ d dlmZ dd	lmZ dd
lmZmZ ddlmZ ddlmZmZmZ ddlmZm Z  ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, ddl-m.Z. ddl/m0Z0 ddl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:  e(       rd dl;Z; e*       rd dl<m=c m>c m?Z@  e+j                  eB      ZCdZD e&deD       e0d       G d de                    ZE e'eEj                        eE_F        eEj                  j                  8eEj                  j                  j                  ddd      eEj                  _G        yy)    N)Callable)deepcopy)partial)AnyOptional)create_repois_offline_mode)validate_typed_dict   )custom_object_save)BatchFeatureget_size_dict)BaseImageProcessorFast)ChannelDimensionSizeDictvalidate_kwargs)UnpackVideosKwargs)IMAGE_PROCESSOR_NAMEPROCESSOR_NAMEVIDEO_PROCESSOR_NAME
TensorTypeadd_start_docstrings	copy_funcis_torch_availableis_torchcodec_availableis_torchvision_v2_availableloggingsafe_load_json_file)cached_file)requires)	
VideoInputVideoMetadatagroup_videos_by_shapeinfer_channel_dimension_formatis_valid_video
load_videomake_batched_metadatamake_batched_videosreorder_videosa  
    Args:
        do_resize (`bool`, *optional*, defaults to `self.do_resize`):
            Whether to resize the video's (height, width) dimensions to the specified `size`. Can be overridden by the
            `do_resize` parameter in the `preprocess` method.
        size (`dict`, *optional*, defaults to `self.size`):
            Size of the output video after resizing. Can be overridden by the `size` parameter in the `preprocess`
            method.
        size_divisor (`int`, *optional*, defaults to `self.size_divisor`):
            The size by which to make sure both the height and width can be divided.
        default_to_square (`bool`, *optional*, defaults to `self.default_to_square`):
            Whether to default to a square video when resizing, if size is an int.
        resample (`PILImageResampling`, *optional*, defaults to `self.resample`):
            Resampling filter to use if resizing the video. Only has an effect if `do_resize` is set to `True`. Can be
            overridden by the `resample` parameter in the `preprocess` method.
        do_center_crop (`bool`, *optional*, defaults to `self.do_center_crop`):
            Whether to center crop the video to the specified `crop_size`. Can be overridden by `do_center_crop` in the
            `preprocess` method.
        crop_size (`dict[str, int]` *optional*, defaults to `self.crop_size`):
            Size of the output video after applying `center_crop`. Can be overridden by `crop_size` in the `preprocess`
            method.
        do_rescale (`bool`, *optional*, defaults to `self.do_rescale`):
            Whether to rescale the video by the specified scale `rescale_factor`. Can be overridden by the
            `do_rescale` parameter in the `preprocess` method.
        rescale_factor (`int` or `float`, *optional*, defaults to `self.rescale_factor`):
            Scale factor to use if rescaling the video. Only has an effect if `do_rescale` is set to `True`. Can be
            overridden by the `rescale_factor` parameter in the `preprocess` method.
        do_normalize (`bool`, *optional*, defaults to `self.do_normalize`):
            Whether to normalize the video. Can be overridden by the `do_normalize` parameter in the `preprocess`
            method. Can be overridden by the `do_normalize` parameter in the `preprocess` method.
        image_mean (`float` or `list[float]`, *optional*, defaults to `self.image_mean`):
            Mean to use if normalizing the video. This is a float or list of floats the length of the number of
            channels in the video. Can be overridden by the `image_mean` parameter in the `preprocess` method. Can be
            overridden by the `image_mean` parameter in the `preprocess` method.
        image_std (`float` or `list[float]`, *optional*, defaults to `self.image_std`):
            Standard deviation to use if normalizing the video. This is a float or list of floats the length of the
            number of channels in the video. Can be overridden by the `image_std` parameter in the `preprocess` method.
            Can be overridden by the `image_std` parameter in the `preprocess` method.
        do_convert_rgb (`bool`, *optional*, defaults to `self.image_std`):
            Whether to convert the video to RGB.
        video_metadata (`VideoMetadata`, *optional*):
            Metadata of the video containing information about total duration, fps and total number of frames.
        do_sample_frames (`int`, *optional*, defaults to `self.do_sample_frames`):
            Whether to sample frames from the video before processing or to process the whole video.
        num_frames (`int`, *optional*, defaults to `self.num_frames`):
            Maximum number of frames to sample when `do_sample_frames=True`.
        fps (`int` or `float`, *optional*, defaults to `self.fps`):
            Target frames to sample per second when `do_sample_frames=True`.
        return_tensors (`str` or `TensorType`, *optional*):
            Returns stacked tensors if set to `pt, otherwise returns a list of tensors.
        data_format (`ChannelDimension` or `str`, *optional*, defaults to `ChannelDimension.FIRST`):
            The channel dimension format for the output video. Can be one of:
            - `"channels_first"` or `ChannelDimension.FIRST`: video in (num_channels, height, width) format.
            - `"channels_last"` or `ChannelDimension.LAST`: video in (height, width, num_channels) format.
            - Unset: Use the channel dimension format of the input video.
        input_data_format (`ChannelDimension` or `str`, *optional*):
            The channel dimension format for the input video. If unset, the channel dimension format is inferred
            from the input video. Can be one of:
            - `"channels_first"` or `ChannelDimension.FIRST`: video in (num_channels, height, width) format.
            - `"channels_last"` or `ChannelDimension.LAST`: video in (height, width, num_channels) format.
            - `"none"` or `ChannelDimension.NONE`: video in (height, width) format.
        device (`torch.device`, *optional*):
            The device to process the videos on. If unset, the device is inferred from the input videos.
        return_metadata (`bool`, *optional*):
            Whether to return video metadata or not.
        z!Constructs a base VideoProcessor.)visiontorchvision)backendsc                       e Zd ZdZdZdZdZdZdZdZ	dZ
dZdZdZdZdZdZdZdZdZdZdZeZdgZdee   ddf fdZdefd	Zd
ddefdZ	 	 d?dede dz  de e!z  dz  fdZ"	 	 d?dedee#z  de$dz  de%dz  de&d   f
dZ'	 	 d?dede(e)z  dz  de(dz  de&d   fdZ* e+e,      dedee   defd       Z-	 d@de&d   de$de$de.de/d   de$d e.d!e$d"e!d#e$d$e!e&e!   z  dz  d%e!e&e!   z  dz  d&e(e0z  dz  defd'Z1e2	 	 	 	 	 dAd(e(e3jh                  z  d)e(e3jh                  z  dz  d*e$d+e$d,e(e$z  dz  d-e(fd.       Z5dBd/e(e3jh                  z  d0e$fd1Z6e2d(e(e3jh                  z  de7e#e(e8f   e#e(e8f   f   fd2       Z9e2d3e#e(e8f   fd4       Z:de#e(e8f   fd5Z;de(fd6Z<d7e(e3jh                  z  fd8Z=d9 Z>e2d:e(e3jh                  z  fd;       Z?e2dCd<       Z@d@d=e(e&e(   z  e&e&e(      z  fd>ZA xZBS )DBaseVideoProcessorNTgp?Fpixel_values_videoskwargsreturnc                    t         |           |j                  dd        |j                         D ]  \  }}	 t	        | ||        |j                  d| j                        }|'t        ||j                  d| j                              nd | _        |j                  d| j                        }|t        |d	      nd | _        t        | j                  j                  j                               | _        | j                   D ]E  }|j#                  |      t	        | |||          %t	        | |t%        t'        | |d                    G y # t
        $ r%}t        j                  d| d| d|         |d }~ww xY w)
Nprocessor_classz
Can't set z with value z for sizedefault_to_square)r5   r6   	crop_size)
param_name)super__init__popitemssetattrAttributeErrorloggererrorr5   r   r6   r7   listvalid_kwargs__annotations__keysmodel_valid_processing_keysgetr   getattr)selfr1   keyvalueerrr5   r7   	__class__s          _/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/video_processing_utils.pyr:   zBaseVideoProcessor.__init__   se   

$d+ !,,. 	JCc5)	 zz&$)),  tvzzBUW[WmWm7no 		
 JJ{DNN;	MVMby[Ihl ,00A0A0Q0Q0V0V0X+Y(33 	GCzz#*c6#;/c8GD#t,D#EF		G! " z#l5'tfMN	s   E	E; E66E;c                 (     | j                   |fi |S N)
preprocess)rH   videosr1   s      rM   __call__zBaseVideoProcessor.__call__   s    tv000    videoztorch.Tensorc                    t        j                  |      }|j                  d   dk(  s|ddddddf   dk  j                         s|S |ddddddf   dz  }d|ddddddf   z
  dz  |ddddddf   |dddddddf   z  z   }|S )z
        Converts a video to RGB format.

        Args:
            video (`"torch.Tensor"`):
                The video to convert.

        Returns:
            `torch.Tensor`: The converted video.
           .N   g     o@r   )tvFgrayscale_to_rgbshapeany)rH   rT   alphas      rM   convert_to_rgbz!BaseVideoProcessor.convert_to_rgb   s     $$U+;;r?ac1al(;c(A'F'F'HL c1al#e+U3a?++s2U3a?5KeTWY[Z[Y[]^`aTaNb5bbrS   metadata
num_framesfpsc                    ||t        d      ||n| j                  }||n| j                  }|j                  }|6|4||j                  t        d      t	        ||j                  z  |z        }||kD  rt        d| d| d      |*t        j                  d|||z        j	                         }|S t        j                  d|      j	                         }|S )a%  
        Default sampling function which uniformly samples the desired number of frames between 0 and total number of frames.
        If `fps` is passed along with metadata, `fps` frames per second are sampled uniformty. Arguments `num_frames`
        and `fps` are mutually exclusive.

        Args:
            metadata (`VideoMetadata`):
                Metadata of the video containing information about total duration, fps and total number of frames.
            num_frames (`int`, *optional*):
                Maximum number of frames to sample. Defaults to `self.num_frames`.
            fps (`int` or `float`, *optional*):
                Target frames to sample per second. Defaults to `self.fps`.

        Returns:
            np.ndarray:
                Indices to sample video frames.
        zc`num_frames`, `fps`, and `sample_indices_fn` are mutually exclusive arguments, please use only one!zAsked to sample `fps` frames per second but no video metadata was provided which is required when sampling with `fps`. Please pass in `VideoMetadata` object or use a fixed `num_frames` per input videoz(Video can't be sampled. The `num_frames=z` exceeds `total_num_frames=z`. r   )
ValueErrorr`   ra   total_num_framesinttorcharange)rH   r_   r`   ra   r1   rd   indicess          rM   sample_framesz BaseVideoProcessor.sample_frames   s   0 ?z5u  $.#9Zt
_c$((#44 #/8<<#7 h  -<sBCJ((::,Fbcsbttwx  !ll1&68H:8UVZZ\G  ll1&67;;=GrS   rQ   video_metadatado_sample_framessample_indices_fnc                 V   t        |      }t        ||      }t        |d         rW|rUg }g }t        ||      D ]:  \  }} ||      }	|	|_        |j                  ||	          |j                  |       < |}|}||fS t        |d         st        |d   t              rg| j                  |      D 
cg c]:  }
t        j                  |
D cg c]  }t        j                  |       c}d      < }}
}|rt        d      ||fS | j                  ||      \  }}||fS c c}w c c}}
w )zB
        Decode input videos and sample frames if needed.
        )rj   r   )r_   )dimzUSampling frames from a list of images is not supported! Set `do_sample_frames=False`.rl   )r)   r(   r&   zipframes_indicesappend
isinstancerA   fetch_imagesrf   stackrY   pil_to_tensorrc   fetch_videos)rH   rQ   rj   rk   rl   sampled_videossampled_metadatarT   r_   rh   imagesimages               rM   _decode_and_sample_videosz,BaseVideoProcessor._decode_and_sample_videos  sX    %V,.vnU &)$)9N!#&v~#> 2x+X>*1'%%eGn5 ''1	2
 $F-N ~%%  q	*&)T* #'"3"3F"; KKv Ne!2!25!9 NTUV  $$o  ~%% *.):):6Uf):)g&~%% !Os   3D%D 'D% D%input_data_formatdevicec                 ^   g }|D ]  }t        |t        j                        r#t        j                  |      j                         }|t        |      }|t        j                  k(  r"|j                  dddd      j                         }||j                  |      }|j                  |        |S )z:
        Prepare the input videos for processing.
        r   rW   r      )rs   npndarrayrf   
from_numpy
contiguousr%   r   LASTpermutetorr   )rH   rQ   r}   r~   processed_videosrT   s         rM   _prepare_input_videosz(BaseVideoProcessor._prepare_input_videosF  s      	+E%,((/::< !($B5$I! $4$9$99aAq1<<>!(##E*!	+"  rS   c           	         t        |j                         t        | j                  j                  j                               dgz          t        | j                  |       | j                  j                  D ]  }|j                  |t        | |d              ! |j                  d      }|j                  d      }|j                  d      }|j                  d      }|rt        | j                  fi |nd }| j                  ||||      \  }}| j                  |||      } | j                  di |} | j                  di | |j                  d	       |j                  d
      }	 | j                  dd|i|}
|	r||
d<   |
S )Nreturn_tensors)captured_kwargsvalid_processor_keysr}   rk   r~   rj   )rj   rk   rl   )rQ   r}   r~   data_formatreturn_metadatarQ    )r   rD   rA   rB   rC   r
   
setdefaultrG   r;   r   ri   r|   r   _further_process_kwargs_validate_preprocess_kwargs_preprocess)rH   rQ   r1   
kwarg_namer}   rk   r~   rj   rl   r   preprocessed_videoss              rM   rP   zBaseVideoProcessor.preprocessc  s    	"KKM!%d&7&7&G&G&L&L&N!OScRd!d	
 	D--v6 ++;; 	KJj'$
D*IJ	K #JJ':;!::&89H%$45EUGD$6$6A&A[_!%!?!?)-/	 "@ "
 ++6M^gm+n---77(((262 	

=! **%67.d..GfGG4B 01""rS   do_convert_rgb	do_resizer5   interpolationztvF.InterpolationModedo_center_cropr7   
do_rescalerescale_factordo_normalize
image_mean	image_stdr   c           	         t        |      \  }}i }|j                         D ]3  \  }}|r| j                  |      }|r| j                  |||      }|||<   5 t	        ||      }t        |      \  }}i }|j                         D ]4  \  }}|r| j                  ||      }| j                  |||	|
||      }|||<   6 t	        ||      }t        d|i|      S )N)r5   r   r0   )datatensor_type)r$   r<   r^   resizer*   center_croprescale_and_normalizer   )rH   rQ   r   r   r5   r   r   r7   r   r   r   r   r   r   r1   grouped_videosgrouped_videos_indexresized_videos_groupedr[   stacked_videosresized_videosprocessed_videos_groupedr   s                          rM   r   zBaseVideoProcessor._preprocess  s   $ 0EV/L,,!#%3%9%9%; 	;!E>!%!4!4^!D!%^$Vc!d,:"5)	; ((>@TU 0E^/T,,#% %3%9%9%; 	=!E>!%!1!1.)!L!77
NL*V_N /=$U+	= **BDXY"79I!JXfggrS   pretrained_model_name_or_path	cache_dirforce_downloadlocal_files_onlytokenrevisionc                     ||d<   ||d<   ||d<   ||d<   |||d<    | j                   |fi |\  }} | j                  |fi |S )aR  
        Instantiate a type of [`~video_processing_utils.VideoProcessorBase`] from an video processor.

        Args:
            pretrained_model_name_or_path (`str` or `os.PathLike`):
                This can be either:

                - a string, the *model id* of a pretrained video hosted inside a model repo on
                  huggingface.co.
                - a path to a *directory* containing a video processor file saved using the
                  [`~video_processing_utils.VideoProcessorBase.save_pretrained`] method, e.g.,
                  `./my_model_directory/`.
                - a path or url to a saved video processor JSON *file*, e.g.,
                  `./my_model_directory/video_preprocessor_config.json`.
            cache_dir (`str` or `os.PathLike`, *optional*):
                Path to a directory in which a downloaded pretrained model video processor should be cached if the
                standard cache should not be used.
            force_download (`bool`, *optional*, defaults to `False`):
                Whether or not to force to (re-)download the video processor files and override the cached versions if
                they exist.
            proxies (`dict[str, str]`, *optional*):
                A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128',
                'http://hostname': 'foo.bar:4012'}.` The proxies are used on each request.
            token (`str` or `bool`, *optional*):
                The token to use as HTTP bearer authorization for remote files. If `True`, or not specified, will use
                the token generated when running `hf auth login` (stored in `~/.huggingface`).
            revision (`str`, *optional*, defaults to `"main"`):
                The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a
                git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any
                identifier allowed by git.


                <Tip>

                To test a pull request you made on the Hub, you can pass `revision="refs/pr/<pr_number>"`.

                </Tip>

            return_unused_kwargs (`bool`, *optional*, defaults to `False`):
                If `False`, then this function returns just the final video processor object. If `True`, then this
                functions returns a `Tuple(video_processor, unused_kwargs)` where *unused_kwargs* is a dictionary
                consisting of the key/value pairs whose keys are not video processor attributes: i.e., the part of
                `kwargs` which has not been used to update `video_processor` and is otherwise ignored.
            subfolder (`str`, *optional*, defaults to `""`):
                In case the relevant files are located inside a subfolder of the model repo on huggingface.co, you can
                specify the folder name here.
            kwargs (`dict[str, Any]`, *optional*):
                The values in kwargs of any keys which are video processor attributes will be used to override the
                loaded values. Behavior concerning key/value pairs whose keys are *not* video processor attributes is
                controlled by the `return_unused_kwargs` keyword parameter.

        Returns:
            A video processor of type [`~video_processing_utils.ImagVideoProcessorBase`].

        Examples:

        ```python
        # We can't instantiate directly the base class *VideoProcessorBase* so let's show the examples on a
        # derived class: *LlavaOnevisionVideoProcessor*
        video_processor = LlavaOnevisionVideoProcessor.from_pretrained(
            "llava-hf/llava-onevision-qwen2-0.5b-ov-hf"
        )  # Download video_processing_config from huggingface.co and cache.
        video_processor = LlavaOnevisionVideoProcessor.from_pretrained(
            "./test/saved_model/"
        )  # E.g. video processor (or model) was saved using *save_pretrained('./test/saved_model/')*
        video_processor = LlavaOnevisionVideoProcessor.from_pretrained("./test/saved_model/video_preprocessor_config.json")
        video_processor = LlavaOnevisionVideoProcessor.from_pretrained(
            "llava-hf/llava-onevision-qwen2-0.5b-ov-hf", do_normalize=False, foo=False
        )
        assert video_processor.do_normalize is False
        video_processor, unused_kwargs = LlavaOnevisionVideoProcessor.from_pretrained(
            "llava-hf/llava-onevision-qwen2-0.5b-ov-hf", do_normalize=False, foo=False, return_unused_kwargs=True
        )
        assert video_processor.do_normalize is False
        assert unused_kwargs == {"foo": False}
        ```r   r   r   r   r   )get_video_processor_dict	from_dict)	clsr   r   r   r   r   r   r1   video_processor_dicts	            rM   from_pretrainedz"BaseVideoProcessor.from_pretrained  st    n ({#1 %5!"%z#F7O'Cs'C'CDa'lek'l$fs}}1<V<<rS   save_directorypush_to_hubc           	         t         j                  j                  |      rt        d| d      t        j                  |d       |rw|j                  dd      }|j                  d|j                  t         j                  j                        d         }t        |fd	di|j                  }| j                  |      }| j                  t        | || 
       t         j                  j                  |t              }| j                  |       t         j#                  d|        |r%| j%                  ||j'                  d             |gS )aq  
        Save an video processor object to the directory `save_directory`, so that it can be re-loaded using the
        [`~video_processing_utils.VideoProcessorBase.from_pretrained`] class method.

        Args:
            save_directory (`str` or `os.PathLike`):
                Directory where the video processor JSON file will be saved (will be created if it does not exist).
            push_to_hub (`bool`, *optional*, defaults to `False`):
                Whether or not to push your model to the Hugging Face model hub after saving it. You can specify the
                repository you want to push to with `repo_id` (will default to the name of `save_directory` in your
                namespace).
            kwargs (`dict[str, Any]`, *optional*):
                Additional key word arguments passed along to the [`~utils.PushToHubMixin.push_to_hub`] method.
        zProvided path (z#) should be a directory, not a fileT)exist_okcommit_messageNrepo_idr   )configzVideo processor saved in r   )r   r   )ospathisfileAssertionErrormakedirsr;   splitsepr   r   _get_files_timestamps_auto_classr   joinr   to_json_filer?   info_upload_modified_filesrF   )rH   r   r   r1   r   r   files_timestampsoutput_video_processor_files           rM   save_pretrainedz"BaseVideoProcessor.save_pretrained"  s/    77>>.) ?>2BBe!fgg
NT2#ZZ(8$?NjjN,@,@,Mb,QRG!'CDCFCKKG#99.I 't^DA ')ggll>CW&X#56/0K/LMN'' -jj) (  ,,,rS   c                 J   |j                  dd      }|j                  dd      }|j                  dd      }|j                  dd      }|j                  dd      }|j                  dd      }|j                  d	d
      }	|j                  dd      }
|j                  dd      }d|d}|
|
|d<   t               r|st        j                  d       d}t	        |      }t
        j                  j                  |      }t
        j                  j                  |      r|}d}d}nXt        }	 t        |t        ||||||||	d      }|t        fD cg c]  }t        ||||||||||	d      x}	 | }}|r|d   nd}d}|t        |      }d|v r|d   }||t        |      }|t        d| d| d d      |rt        j                  d|        ||fS t        j                  d d|        ||fS c c}w # t        $ r  t        $ r t        d| d| d| d      w xY w)a  
        From a `pretrained_model_name_or_path`, resolve to a dictionary of parameters, to be used for instantiating a
        video processor of type [`~video_processing_utils.VideoProcessorBase`] using `from_dict`.

        Parameters:
            pretrained_model_name_or_path (`str` or `os.PathLike`):
                The identifier of the pre-trained checkpoint from which we want the dictionary of parameters.
            subfolder (`str`, *optional*, defaults to `""`):
                In case the relevant files are located inside a subfolder of the model repo on huggingface.co, you can
                specify the folder name here.

        Returns:
            `tuple[Dict, Dict]`: The dictionary(ies) that will be used to instantiate the video processor object.
        r   Nr   Fproxiesr   r   r   	subfolder _from_pipeline
_from_autovideo processor)	file_typefrom_auto_classusing_pipelinez+Offline mode: forcing local_files_only=TrueT)
filenamer   r   r   r   r   
user_agentr   r   %_raise_exceptions_for_missing_entriesr   z Can't load video processor for 'z'. If you were trying to load it from 'https://huggingface.co/models', make sure you don't have a local directory with the same name. Otherwise, make sure 'z2' is the correct path to a directory containing a z filevideo_processorzloading configuration file z from cache at )r;   r	   r?   r   strr   r   isdirr   r   r    r   r   OSError	Exceptionr   )r   r   r1   r   r   r   r   r   r   r   from_pipeliner   r   is_localresolved_video_processor_fileresolved_processor_filevideo_processor_filer   resolved_fileresolved_video_processor_filesr   processor_dicts                         rM   r   z+BaseVideoProcessor.get_video_processor_dictR  s   $ JJ{D1	$4e<**Y-

7D)!::&8%@::j$/JJ{B/	

#3T: **\59#4Y
$+8J'(%5KKEF#(+,I(J%77==!>?77>>78,I)&*#H#7 2 +61+'#1#%5)%':?+' &:;O$P2 )49%-&/+9$+-="''1%-&/BG*     "2. 2* :X215]a .&  $".01HIN N2'56G'H$(49M9U#67T#U '23P2Q R55R4S T++?*@G  KK56S5TUV $V++	 KK-.B-C?SpRqr $V++2.    67T6U V99V8W X//C.DEK s   )$G: "G5/G: 5G: :(H"r   c                    |j                         }|j                  dd      }d|v rd|v r|j                  d      |d<   d|v rd|v r|j                  d      |d<    | di |}g }|j                         D ]0  \  }}t        ||      st	        |||       |j                  |       2 |D ]  }|j                  |d        t        j                  d|        |r||fS |S )a  
        Instantiates a type of [`~video_processing_utils.VideoProcessorBase`] from a Python dictionary of parameters.

        Args:
            video_processor_dict (`dict[str, Any]`):
                Dictionary that will be used to instantiate the video processor object. Such a dictionary can be
                retrieved from a pretrained checkpoint by leveraging the
                [`~video_processing_utils.VideoProcessorBase.to_dict`] method.
            kwargs (`dict[str, Any]`):
                Additional parameters from which to initialize the video processor object.

        Returns:
            [`~video_processing_utils.VideoProcessorBase`]: The video processor object instantiated from those
            parameters.
        return_unused_kwargsFr5   r7   NzVideo processor r   )copyr;   r<   hasattrr=   rr   r?   r   )r   r   r1   r   r   	to_removerI   rJ   s           rM   r   zBaseVideoProcessor.from_dict  s   "  488:%zz*@%H
 V*> >+1::f+= (& [4H%H06

;0G -5 45 	 ,,. 	&JC,e4  %	&  	"CJJsD!	" 	&&789"F**""rS   c                 6   t        | j                        }i }|j                         D ]1  \  }}|%t        t	        |       |d      }|dk7  s$|'|||<   -|||<   3 |j                  dd       |j                  dd       | j                  j                  |d<   |S )z
        Serializes this instance to a Python dictionary.

        Returns:
            `dict[str, Any]`: Dictionary of all the attributes that make up this video processor instance.
        N	NOT_FOUNDrE   _valid_kwargs_namesvideo_processor_type)r   __dict__r<   rG   typer;   rL   __name__)rH   outputfiltered_dictrI   rJ   class_defaults         rM   to_dictzBaseVideoProcessor.to_dict  s     $--( ,,. 	+JC} 'T
C E K/M4M).M#&%*c"	+ 	7>/6040G0G,-rS   c                     | j                         }|j                         D ]3  \  }}t        |t        j                        s!|j                         ||<   5 t        j                  |dd      dz   S )z
        Serializes this instance to a JSON string.

        Returns:
            `str`: String containing all the attributes that make up this feature_extractor instance in JSON format.
        r   T)indent	sort_keys
)r   r<   rs   r   r   tolistjsondumps)rH   
dictionaryrI   rJ   s       rM   to_json_stringz!BaseVideoProcessor.to_json_string  sb     \\^
$**, 	1JC%,"',,.
3	1 zz*Q$?$FFrS   json_file_pathc                     t        |dd      5 }|j                  | j                                ddd       y# 1 sw Y   yxY w)z
        Save this instance to a JSON file.

        Args:
            json_file_path (`str` or `os.PathLike`):
                Path to the JSON file in which this image_processor instance's parameters will be saved.
        wutf-8encodingN)openwriter  )rH   r  writers      rM   r   zBaseVideoProcessor.to_json_file$  s<     .#8 	0FLL,,./	0 	0 	0s	    8Ac                 T    | j                   j                   d| j                          S )N )rL   r   r  )rH   s    rM   __repr__zBaseVideoProcessor.__repr__/  s(    ..))*!D,?,?,A+BCCrS   	json_filec                     t        |dd      5 }|j                         }ddd       t        j                        } | di |S # 1 sw Y   &xY w)a  
        Instantiates a video processor of type [`~video_processing_utils.VideoProcessorBase`] from the path to a JSON
        file of parameters.

        Args:
            json_file (`str` or `os.PathLike`):
                Path to the JSON file containing the parameters.

        Returns:
            A video processor of type [`~video_processing_utils.VideoProcessorBase`]: The video_processor object
            instantiated from that JSON file.
        rr  r  Nr   )r	  readr   loads)r   r  readertextr   s        rM   from_json_filez!BaseVideoProcessor.from_json_file2  sP     )S73 	!v;;=D	!#zz$/*)**	! 	!s   AAc                     t        |t              s|j                  }ddlmc m} t        ||      st        | d      || _        y)a	  
        Register this class with a given auto class. This should only be used for custom video processors as the ones
        in the library are already mapped with `AutoVideoProcessor `.

        <Tip warning={true}>

        This API is experimental and may have some slight breaking changes in the next releases.

        </Tip>

        Args:
            auto_class (`str` or `type`, *optional*, defaults to `"AutoVideoProcessor "`):
                The auto class to register this new video processor with.
        r   Nz is not a valid auto class.)	rs   r   r   transformers.models.automodelsautor   rc   r   )r   
auto_classauto_modules      rM   register_for_auto_classz*BaseVideoProcessor.register_for_auto_classE  sC      *c*#,,J66{J/
|+FGHH$rS   video_url_or_urlsc                     d}t               st        j                  d       d}t        |t              r0t	        t        |D cg c]  }| j                  ||       c}       S t        |||      S c c}w )z
        Convert a single or a list of urls into the corresponding `np.array` objects.

        If a single url is passed, the return value will be a single object. If a list is passed a list of objects is
        returned.
        
torchcodecz`torchcodec` is not installed and cannot be used to decode the video by default. Falling back to `torchvision`. Note that `torchvision` decoding is deprecated and will be removed in future versions. r,   ro   )backendrl   )r   warningswarnrs   rA   rp   rw   r'   )rH   r  rl   r!  xs        rM   rw   zBaseVideoProcessor.fetch_videos_  sx     &(MMI $G'.ars\]d//EV/Wstuu/Teff ts   A2)NNrO   )NFFNmain)F)AutoVideoProcessor)Cr   
__module____qualname__r   resampler   r   r5   size_divisorr6   r7   r   r   r   r   r   r   rk   ra   r`   rj   r   r   rB   model_input_namesr   r:   r   rR   r"   r^   r#   re   floatri   dictboolr   rA   r|   r   r   r   r   BASE_VIDEO_PROCESSOR_DOCSTRINGrP   r   r   r   r   classmethodr   PathLiker   r   tupler   r   r   r   r  r   r  r  r  rw   __classcell__)rL   s   @rM   r/   r/      sA    KHJIDLIINJNLN
CJNOL./G!5 G$ G>1L 1 
8 "&"&	33 $J3 5[4	3r )--1&&&& &,&& +	&&
 $d?&& 
n	&&V <@!	   !11D8  d
	 
 
n	 : &*#*# &*# 
	*#*#t 37+h^$+h +h 	+h
 +h   78+h +h +h +h +h +h DK'$.+h 4;&-+h j(4/+h  
!+hZ  /3$!&#'`='*R[['8`= $t+`= 	`=
 `= TzD `= `= `=D.-cBKK.? .-d .-` {,,/"++,={,	tCH~tCH~-	.{, {,z *#T#s(^ *# *#Xc3h 0G G	03+< 	0D +sR[['8 + +$ % %2gcDIoT#Y.O grS   r/   r   r&  zvideo processor file)objectobject_classobject_files)Ir   r   r"  collections.abcr   r   r   	functoolsr   typingr   r   numpyr   huggingface_hubr   r	   huggingface_hub.dataclassesr
   dynamic_module_utilsr   image_processing_utilsr   r   image_processing_utils_fastr   image_utilsr   r   r   processing_utilsr   r   utilsr   r   r   r   r   r   r   r   r   r   r   	utils.hubr    utils.import_utilsr!   video_utilsr"   r#   r$   r%   r&   r'   r(   r)   r*   rf   $torchvision.transforms.v2.functional
transformsv2
functionalrY   
get_loggerr   r?   r/  r/   r   __doc__formatr   rS   rM   <module>rM     sC    	  $      8 ; 4 @ 
 3    # (
 
 
  66 
		H	%A" H '" 
,-[g/ [g .	
[g| "++=+I+I!J  !!))5-?-K-K-S-S-Z-Z /CRh .[ .""* 6rS   