Hugging Face's logo

Diffusers documentation

Processador de Vídeo

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.43.0).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Processador de Vídeo

O VideoProcessor fornece uma API unificada para pipelines de vídeo para preparar entradas para codificação VAE e pós-processar saídas após a decodificação. A classe herda de VaeImageProcessor, portanto, inclui transformações como redimensionamento, normalização e conversão entre PIL Image, PyTorch e arrays NumPy.

VideoProcessor

diffusers.VideoProcessor.preprocess_video

< >

( videoheight: int | None = Nonewidth: int | None = None**kwargs ) → torch.Tensor of shape (batch_size, num_channels, num_frames, height, width)

Parâmetros

  • video (list[PIL.Image], list[list[PIL.Image]], torch.Tensor, np.array, list[torch.Tensor], list[np.array]) — O vídeo de entrada. Pode ser um dos seguintes:
    • lista das imagens PIL.
    • lista de listas de imagens PIL.
    • Tensores Torch 4D (forma esperada para cada tensor (num_frames, num_channels, height, width)).
    • Arrays NumPy 4D (forma esperada para cada array (num_frames, height, width, num_channels)).
    • lista de tensores Torch 4D (forma esperada para cada tensor (num_frames, num_channels, height, width)).
    • lista de arrays NumPy 4D (forma esperada para cada array (num_frames, height, width, num_channels)).
    • Arrays NumPy 5D: forma esperada para cada array (batch_size, num_frames, height, width, num_channels).
    • Tensores Torch 5D: forma esperada para cada array (batch_size, num_frames, num_channels, height, width).
  • height (int, opcional, padrão None) — A altura em quadros pré-processados do vídeo. Se None, usará o get_default_height_width() para obter a altura padrão.
  • width (int, optional, defaults to None) -- The width in preprocessed frames of the video. If None, will use get_default_height_width() to get the default width.

Retorna

torch.Tensor de forma (batch_size, num_channels, num_frames, height, width)

Um tensor 5D contendo os vídeos em lote com canais na frente.

Pré-processa o(s) vídeo(s) de entrada. Os argumentos de palavra-chave serão encaminhados para VaeImageProcessor.preprocess.

diffusers.VideoProcessor.postprocess_video

< >

( video: Tensoroutput_type: str = 'np'**kwargs )

Parâmetros

  • video (torch.Tensor) — O vídeo como um tensor.
  • output_type (str, padrão "np") — Tipo de saída do tensor video pós-processado.

Converte um tensor de vídeo em uma lista de quadros para exportação. Os argumentos de palavra-chave serão encaminhados para VaeImageProcessor.postprocess.

Update on GitHub

←Processador de Imagem VAE