Hugging Face's logo

Diffusers documentation

Procesador de video

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

and get access to the augmented documentation experience

to get started

Procesador de video

El VideoProcessor ofrece una API unificada para los pipelines de video con el fin de preparar las entradas para la codificación con VAE y realizar el posprocesado de las salidas una vez que se han decodificado. La clase hereda de VaeImageProcessor, por lo que incluye transformaciones como el redimensionado, la normalización y la conversión entre imágenes PIL, PyTorch y matrices de 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]) — El video de entrada. Puede ser uno de los siguientes:
    • lista de las imágenes PIL.
    • lista de listas de imágenes PIL.
    • Tensores Torch 4D (forma esperada para cada tensor (num_frames, num_channels, height, width)).
    • Arreglos NumPy de 4D (forma esperada para cada arreglo (num_frames, height, width, num_channels)).
    • lista de tensores Torch de 4D (forma esperada para cada tensor (num_frames, num_channels, height, width)).
    • lista de matrices NumPy de 4D (forma esperada para cada matriz (num_frames, height, width, num_channels)).
    • Arreglos NumPy de 5D: forma esperada para cada arreglo (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, valor predeterminado None) — La altura en los fotogramas preprocesados del video. Si es None, se usará get_default_height_width() para obtener la altura predeterminada.
  • 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.

Devuelve

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

Un tensor de 5D que contiene los videos en lotes con los canales al inicio.

Preprocesa el video de entrada. Los argumentos de palabra clave se reenviarán a VaeImageProcessor.preprocess.

diffusers.VideoProcessor.postprocess_video

< >

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

Parámetros

  • video (torch.Tensor) — El video como tensor.
  • output_type (str, por defecto "np") — Tipo de salida del tensor video después del postprocesado.

Convierte un tensor de video en una lista de fotogramas para su exportación. Los argumentos de palabra clave se reenviarán a VaeImageProcessor.postprocess.

Update on GitHub

←Procesador de imágenes VAE