| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845 |
- # Copyright 2025 The HuggingFace Inc. team.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import json
- import os
- import warnings
- from collections.abc import Callable
- from functools import partial
- from typing import Any
- import numpy as np
- from huggingface_hub import create_repo, is_offline_mode
- from huggingface_hub.dataclasses import validate_typed_dict
- from .dynamic_module_utils import custom_object_save
- from .image_processing_backends import TorchvisionBackend
- from .image_processing_utils import BatchFeature
- from .image_utils import (
- ChannelDimension,
- SizeDict,
- is_vision_available,
- validate_kwargs,
- )
- from .processing_utils import Unpack, VideosKwargs
- from .utils import (
- 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,
- )
- from .utils.hub import cached_file
- from .utils.import_utils import requires
- from .video_utils import (
- VideoInput,
- VideoMetadata,
- group_videos_by_shape,
- infer_channel_dimension_format,
- is_valid_video,
- load_video,
- make_batched_metadata,
- make_batched_videos,
- reorder_videos,
- )
- if is_torch_available():
- import torch
- if is_torchvision_v2_available():
- import torchvision.transforms.v2.functional as tvF
- if is_vision_available():
- from .image_utils import PILImageResampling
- logger = logging.get_logger(__name__)
- BASE_VIDEO_PROCESSOR_DOCSTRING = r"""
- 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.
- """
- @add_start_docstrings(
- "Constructs a base VideoProcessor.",
- BASE_VIDEO_PROCESSOR_DOCSTRING,
- )
- @requires(backends=("vision", "torchvision"))
- class BaseVideoProcessor(TorchvisionBackend):
- _auto_class = None
- resample = None
- image_mean = None
- image_std = None
- size = None
- size_divisor = None
- default_to_square = True
- crop_size = None
- do_resize = None
- do_center_crop = None
- do_rescale = None
- rescale_factor = 1 / 255
- do_normalize = None
- do_convert_rgb = None
- do_sample_frames = None
- fps = None
- num_frames = None
- video_metadata = None
- return_metadata = False
- valid_kwargs = VideosKwargs
- model_input_names = ["pixel_values_videos"]
- def __init__(self, **kwargs: Unpack[VideosKwargs]) -> None:
- super().__init__(**kwargs)
- def __call__(self, videos, **kwargs) -> BatchFeature:
- return self.preprocess(videos, **kwargs)
- def convert_to_rgb(
- self,
- video: "torch.Tensor",
- ) -> VideoInput:
- """
- Converts a video to RGB format.
- Args:
- video (`"torch.Tensor"`):
- The video to convert.
- Returns:
- `torch.Tensor`: The converted video.
- """
- video = tvF.grayscale_to_rgb(video)
- if video.shape[-3] == 3 or not (video[..., 3, :, :] < 255).any():
- return video
- # There is a transparency layer, blend it with a white background.
- # Calculate the alpha proportion for blending.
- alpha = video[..., 3, :, :] / 255.0
- video = (1 - alpha[..., None, :, :]) * 255 + alpha[..., None, :, :] * video[..., :3, :, :]
- return video
- def sample_frames(
- self,
- metadata: VideoMetadata,
- num_frames: int | None = None,
- fps: int | float | None = None,
- **kwargs,
- ):
- """
- 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.
- """
- if fps is not None and num_frames is not None:
- raise ValueError(
- "`num_frames`, `fps`, and `sample_indices_fn` are mutually exclusive arguments, please use only one!"
- )
- num_frames = num_frames if num_frames is not None else self.num_frames
- fps = fps if fps is not None else self.fps
- total_num_frames = metadata.total_num_frames
- # If num_frames is not given but fps is, calculate num_frames from fps
- if num_frames is None and fps is not None:
- if metadata is None or metadata.fps is None:
- raise ValueError(
- "Asked 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 video"
- )
- num_frames = int(total_num_frames / metadata.fps * fps)
- if num_frames > total_num_frames:
- raise ValueError(
- f"Video can't be sampled. The `num_frames={num_frames}` exceeds `total_num_frames={total_num_frames}`. "
- )
- if num_frames is not None:
- indices = torch.arange(0, total_num_frames, total_num_frames / num_frames).int()
- else:
- indices = torch.arange(0, total_num_frames).int()
- return indices
- def _decode_and_sample_videos(
- self,
- videos: VideoInput,
- video_metadata: VideoMetadata | dict,
- do_sample_frames: bool | None = None,
- sample_indices_fn: Callable | None = None,
- ) -> list["torch.Tensor"]:
- """
- Decode input videos and sample frames if needed.
- """
- videos = make_batched_videos(videos)
- video_metadata = make_batched_metadata(videos, video_metadata=video_metadata)
- # Only sample frames if an array video is passed, otherwise first decode -> then sample
- if is_valid_video(videos[0]) and do_sample_frames:
- sampled_videos = []
- sampled_metadata = []
- for video, metadata in zip(videos, video_metadata):
- indices = sample_indices_fn(metadata=metadata)
- metadata.frames_indices = indices
- sampled_videos.append(video[indices])
- sampled_metadata.append(metadata)
- videos = sampled_videos
- video_metadata = sampled_metadata
- elif not is_valid_video(videos[0]):
- if isinstance(videos[0], list):
- # Videos sometimes are passed as a list of image URLs, especially through templates
- videos = [
- torch.stack([tvF.pil_to_tensor(image) for image in images], dim=0)
- for images in self.fetch_images(videos)
- ]
- if do_sample_frames:
- raise ValueError(
- "Sampling frames from a list of images is not supported! Set `do_sample_frames=False`."
- )
- else:
- videos, video_metadata = self.fetch_videos(videos, sample_indices_fn=sample_indices_fn)
- return videos, video_metadata
- def _prepare_input_videos(
- self,
- videos: VideoInput,
- input_data_format: str | ChannelDimension | None = None,
- device: str | None = None,
- ) -> list["torch.Tensor"]:
- """
- Prepare the input videos for processing.
- """
- processed_videos = []
- for video in videos:
- # `make_batched_videos` always returns a 4D array per video
- if isinstance(video, np.ndarray):
- # not using tvF.to_tensor as it doesn't handle (C, H, W) numpy arrays
- video = torch.from_numpy(video).contiguous()
- # Infer the channel dimension format if not provided
- if input_data_format is None:
- input_data_format = infer_channel_dimension_format(video)
- if input_data_format == ChannelDimension.LAST:
- video = video.permute(0, 3, 1, 2).contiguous()
- if device is not None:
- video = video.to(device)
- processed_videos.append(video)
- return processed_videos
- @add_start_docstrings(
- BASE_VIDEO_PROCESSOR_DOCSTRING,
- )
- def preprocess(
- self,
- videos: VideoInput,
- **kwargs: Unpack[VideosKwargs],
- ) -> BatchFeature:
- validate_kwargs(
- captured_kwargs=kwargs.keys(),
- valid_processor_keys=list(self.valid_kwargs.__annotations__.keys()) + ["return_tensors"],
- )
- # Perform type validation on received kwargs
- validate_typed_dict(self.valid_kwargs, kwargs)
- # Set default kwargs from self. This ensures that if a kwarg is not provided
- # by the user, it gets its default value from the instance, or is set to None.
- for kwarg_name in self.valid_kwargs.__annotations__:
- kwargs.setdefault(kwarg_name, getattr(self, kwarg_name, None))
- input_data_format = kwargs.pop("input_data_format")
- do_sample_frames = kwargs.pop("do_sample_frames")
- device = kwargs.pop("device")
- video_metadata = kwargs.pop("video_metadata")
- sample_indices_fn = partial(self.sample_frames, **kwargs) if do_sample_frames else None
- videos, video_metadata = self._decode_and_sample_videos(
- videos,
- video_metadata=video_metadata,
- do_sample_frames=do_sample_frames,
- sample_indices_fn=sample_indices_fn,
- )
- videos = self._prepare_input_videos(videos=videos, input_data_format=input_data_format, device=device)
- kwargs = self._standardize_kwargs(**kwargs)
- self._validate_preprocess_kwargs(**kwargs)
- # Pop kwargs that are not needed in _preprocess
- kwargs.pop("data_format")
- return_metadata = kwargs.pop("return_metadata")
- preprocessed_videos = self._preprocess(videos=videos, **kwargs)
- if return_metadata:
- preprocessed_videos["video_metadata"] = video_metadata
- return preprocessed_videos
- def _preprocess(
- self,
- videos: list["torch.Tensor"],
- do_convert_rgb: bool,
- do_resize: bool,
- size: SizeDict,
- resample: "PILImageResampling | tvF.InterpolationMode | int | None",
- do_center_crop: bool,
- crop_size: SizeDict,
- do_rescale: bool,
- rescale_factor: float,
- do_normalize: bool,
- image_mean: float | list[float] | None,
- image_std: float | list[float] | None,
- return_tensors: str | TensorType | None = None,
- **kwargs,
- ) -> BatchFeature:
- # Group videos by size for batched resizing
- grouped_videos, grouped_videos_index = group_videos_by_shape(videos)
- resized_videos_grouped = {}
- for shape, stacked_videos in grouped_videos.items():
- if do_convert_rgb:
- stacked_videos = self.convert_to_rgb(stacked_videos)
- if do_resize:
- stacked_videos = self.resize(stacked_videos, size=size, resample=resample)
- resized_videos_grouped[shape] = stacked_videos
- resized_videos = reorder_videos(resized_videos_grouped, grouped_videos_index)
- # Group videos by size for further processing
- # Needed in case do_resize is False, or resize returns videos with different sizes
- grouped_videos, grouped_videos_index = group_videos_by_shape(resized_videos)
- processed_videos_grouped = {}
- for shape, stacked_videos in grouped_videos.items():
- if do_center_crop:
- stacked_videos = self.center_crop(stacked_videos, crop_size)
- # Fused rescale and normalize
- stacked_videos = self.rescale_and_normalize(
- stacked_videos, do_rescale, rescale_factor, do_normalize, image_mean, image_std
- )
- processed_videos_grouped[shape] = stacked_videos
- processed_videos = reorder_videos(processed_videos_grouped, grouped_videos_index)
- return BatchFeature(data={"pixel_values_videos": processed_videos}, tensor_type=return_tensors)
- @classmethod
- def from_pretrained(
- cls,
- pretrained_model_name_or_path: str | os.PathLike,
- cache_dir: str | os.PathLike | None = None,
- force_download: bool = False,
- local_files_only: bool = False,
- token: str | bool | None = None,
- revision: str = "main",
- **kwargs,
- ):
- r"""
- 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 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}
- ```"""
- kwargs["cache_dir"] = cache_dir
- kwargs["force_download"] = force_download
- kwargs["local_files_only"] = local_files_only
- kwargs["revision"] = revision
- if token is not None:
- kwargs["token"] = token
- video_processor_dict, kwargs = cls.get_video_processor_dict(pretrained_model_name_or_path, **kwargs)
- return cls.from_dict(video_processor_dict, **kwargs)
- def save_pretrained(self, save_directory: str | os.PathLike, push_to_hub: bool = False, **kwargs):
- """
- 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.
- """
- if os.path.isfile(save_directory):
- raise AssertionError(f"Provided path ({save_directory}) should be a directory, not a file")
- os.makedirs(save_directory, exist_ok=True)
- if push_to_hub:
- commit_message = kwargs.pop("commit_message", None)
- repo_id = kwargs.pop("repo_id", save_directory.split(os.path.sep)[-1])
- repo_id = create_repo(repo_id, exist_ok=True, **kwargs).repo_id
- files_timestamps = self._get_files_timestamps(save_directory)
- # If we have a custom config, we copy the file defining it in the folder and set the attributes so it can be
- # loaded from the Hub.
- if self._auto_class is not None:
- custom_object_save(self, save_directory, config=self)
- # If we save using the predefined names, we can load using `from_pretrained`
- output_video_processor_file = os.path.join(save_directory, VIDEO_PROCESSOR_NAME)
- self.to_json_file(output_video_processor_file)
- logger.info(f"Video processor saved in {output_video_processor_file}")
- if push_to_hub:
- self._upload_modified_files(
- save_directory,
- repo_id,
- files_timestamps,
- commit_message=commit_message,
- token=kwargs.get("token"),
- )
- return [output_video_processor_file]
- @classmethod
- def get_video_processor_dict(
- cls, pretrained_model_name_or_path: str | os.PathLike, **kwargs
- ) -> tuple[dict[str, Any], dict[str, Any]]:
- """
- 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.
- """
- cache_dir = kwargs.pop("cache_dir", None)
- force_download = kwargs.pop("force_download", False)
- proxies = kwargs.pop("proxies", None)
- token = kwargs.pop("token", None)
- local_files_only = kwargs.pop("local_files_only", False)
- revision = kwargs.pop("revision", None)
- subfolder = kwargs.pop("subfolder", "")
- from_pipeline = kwargs.pop("_from_pipeline", None)
- from_auto_class = kwargs.pop("_from_auto", False)
- user_agent = {"file_type": "video processor", "from_auto_class": from_auto_class}
- if from_pipeline is not None:
- user_agent["using_pipeline"] = from_pipeline
- if is_offline_mode() and not local_files_only:
- logger.info("Offline mode: forcing local_files_only=True")
- local_files_only = True
- pretrained_model_name_or_path = str(pretrained_model_name_or_path)
- is_local = os.path.isdir(pretrained_model_name_or_path)
- if os.path.isfile(pretrained_model_name_or_path):
- resolved_video_processor_file = pretrained_model_name_or_path
- resolved_processor_file = None
- is_local = True
- else:
- video_processor_file = VIDEO_PROCESSOR_NAME
- try:
- # Try to load with a new config name first and if not successful try with the old file name
- # NOTE: we save all processor configs as nested dict in PROCESSOR_NAME from v5, which is the standard
- resolved_processor_file = cached_file(
- pretrained_model_name_or_path,
- filename=PROCESSOR_NAME,
- cache_dir=cache_dir,
- force_download=force_download,
- proxies=proxies,
- local_files_only=local_files_only,
- token=token,
- user_agent=user_agent,
- revision=revision,
- subfolder=subfolder,
- _raise_exceptions_for_missing_entries=False,
- )
- resolved_video_processor_files = [
- resolved_file
- for filename in [video_processor_file, IMAGE_PROCESSOR_NAME]
- if (
- resolved_file := cached_file(
- pretrained_model_name_or_path,
- filename=filename,
- cache_dir=cache_dir,
- force_download=force_download,
- proxies=proxies,
- local_files_only=local_files_only,
- token=token,
- user_agent=user_agent,
- revision=revision,
- subfolder=subfolder,
- _raise_exceptions_for_missing_entries=False,
- )
- )
- is not None
- ]
- resolved_video_processor_file = (
- resolved_video_processor_files[0] if resolved_video_processor_files else None
- )
- except OSError:
- # Raise any OS error raise by `cached_file`. It will have a helpful error message adapted to
- # the original exception.
- raise
- except Exception:
- # For any other exception, we throw a generic error.
- raise OSError(
- f"Can't load video processor for '{pretrained_model_name_or_path}'. If you were trying to load"
- " it from 'https://huggingface.co/models', make sure you don't have a local directory with the"
- f" same name. Otherwise, make sure '{pretrained_model_name_or_path}' is the correct path to a"
- f" directory containing a {video_processor_file} file"
- )
- # Load video_processor dict. Priority goes as (nested config if found -> video processor config -> image processor config)
- # We are downloading both configs because almost all models have a `processor_config.json` but
- # not all of these are nested. We need to check if it was saved recebtly as nested or if it is legacy style
- video_processor_dict = None
- if resolved_processor_file is not None:
- processor_dict = safe_load_json_file(resolved_processor_file)
- if "video_processor" in processor_dict:
- video_processor_dict = processor_dict["video_processor"]
- if resolved_video_processor_file is not None and video_processor_dict is None:
- video_processor_dict = safe_load_json_file(resolved_video_processor_file)
- if video_processor_dict is None:
- raise OSError(
- f"Can't load video processor for '{pretrained_model_name_or_path}'. If you were trying to load"
- " it from 'https://huggingface.co/models', make sure you don't have a local directory with the"
- f" same name. Otherwise, make sure '{pretrained_model_name_or_path}' is the correct path to a"
- f" directory containing a {video_processor_file} file"
- )
- if is_local:
- logger.info(f"loading configuration file {resolved_video_processor_file}")
- else:
- logger.info(
- f"loading configuration file {video_processor_file} from cache at {resolved_video_processor_file}"
- )
- return video_processor_dict, kwargs
- @classmethod
- def from_dict(cls, video_processor_dict: dict[str, Any], **kwargs):
- """
- 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.
- """
- video_processor_dict = video_processor_dict.copy()
- return_unused_kwargs = kwargs.pop("return_unused_kwargs", False)
- video_processor_dict.update({k: v for k, v in kwargs.items() if k in cls.valid_kwargs.__annotations__})
- video_processor = cls(**video_processor_dict)
- # Apply extra kwargs to instance (BC for remote code, e.g. phi4_multimodal)
- extra_keys = []
- for key in reversed(list(kwargs.keys())):
- if hasattr(video_processor, key) and key not in cls.valid_kwargs.__annotations__:
- setattr(video_processor, key, kwargs.pop(key, None))
- extra_keys.append(key)
- if extra_keys:
- logger.warning_once(
- f"Image processor {cls.__name__}: kwargs {extra_keys} were applied for backward compatibility. "
- f"To avoid this warning, add them to valid_kwargs: create a custom TypedDict extending "
- f"ImagesKwargs with these keys and set it as the `valid_kwargs` class attribute."
- )
- logger.info(f"Video processor {video_processor}")
- if return_unused_kwargs:
- return video_processor, kwargs
- else:
- return video_processor
- def to_dict(self) -> dict[str, Any]:
- """
- Serializes this instance to a Python dictionary.
- Returns:
- `dict[str, Any]`: Dictionary of all the attributes that make up this video processor instance.
- """
- filtered_dict = super().to_dict()
- filtered_dict.pop("image_processor_type", None)
- filtered_dict["video_processor_type"] = self.__class__.__name__
- return filtered_dict
- def to_json_string(self) -> str:
- """
- Serializes this instance to a JSON string.
- Returns:
- `str`: String containing all the attributes that make up this feature_extractor instance in JSON format.
- """
- dictionary = self.to_dict()
- for key, value in dictionary.items():
- if isinstance(value, np.ndarray):
- dictionary[key] = value.tolist()
- return json.dumps(dictionary, indent=2, sort_keys=True) + "\n"
- def to_json_file(self, json_file_path: str | os.PathLike):
- """
- 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.
- """
- with open(json_file_path, "w", encoding="utf-8") as writer:
- writer.write(self.to_json_string())
- def __repr__(self):
- return f"{self.__class__.__name__} {self.to_json_string()}"
- @classmethod
- def from_json_file(cls, json_file: str | os.PathLike):
- """
- 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.
- """
- with open(json_file, "r", encoding="utf-8") as reader:
- text = reader.read()
- video_processor_dict = json.loads(text)
- return cls(**video_processor_dict)
- @classmethod
- def register_for_auto_class(cls, auto_class="AutoVideoProcessor"):
- """
- 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.
- """
- if not isinstance(auto_class, str):
- auto_class = auto_class.__name__
- import transformers.models.auto as auto_module
- if not hasattr(auto_module, auto_class):
- raise ValueError(f"{auto_class} is not a valid auto class.")
- cls._auto_class = auto_class
- def fetch_videos(self, video_url_or_urls: str | list[str] | list[list[str]], sample_indices_fn=None):
- """
- 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.
- """
- backend = "torchcodec"
- if not is_torchcodec_available():
- warnings.warn(
- "`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. "
- )
- backend = "torchvision"
- if isinstance(video_url_or_urls, list):
- return list(zip(*[self.fetch_videos(x, sample_indices_fn=sample_indices_fn) for x in video_url_or_urls]))
- else:
- return load_video(video_url_or_urls, backend=backend, sample_indices_fn=sample_indices_fn)
- BaseVideoProcessor.push_to_hub = copy_func(BaseVideoProcessor.push_to_hub)
- if BaseVideoProcessor.push_to_hub.__doc__ is not None:
- BaseVideoProcessor.push_to_hub.__doc__ = BaseVideoProcessor.push_to_hub.__doc__.format(
- object="video processor", object_class="AutoVideoProcessor", object_files="video processor file"
- )
|