Passer au contenu principal

​
classe Video

Une classe permettant d’enregistrer des vidéos dans W&B.

​
méthode Video.__init__

__init__(
    data_or_path: 'str | pathlib.Path | np.ndarray | TextIO | BytesIO',
    caption: 'str | None' = None,
    fps: 'int | None' = None,
    format: "Literal['gif', 'mp4', 'webm', 'ogg'] | None" = None
)
Initialise un objet W&B vidéo. Arguments :
  • data_or_path : Vidéo peut être initialisé à partir d’un chemin de fichier ou d’un objet io. vidéo peut également être initialisé avec un tenseur NumPy. Le tenseur NumPy doit comporter 4 ou 5 dimensions. Les dimensions doivent être (nombre d’images, canal, hauteur, largeur) ou (lot, nombre d’images, canal, hauteur, largeur). Le paramètre format doit être spécifié avec l’argument format lors de l’initialisation à partir d’un tableau NumPy ou d’un objet io.
  • caption : Légende associée à la vidéo pour l’affichage.
  • fps : La fréquence d’images à utiliser lors de l’encodage des images vidéo brutes. La valeur par défaut est 4. Ce paramètre n’a aucun effet lorsque data_or_path est une chaîne de caractères ou des octets.
  • format : Format de la vidéo, requis en cas d’initialisation à partir d’un tableau NumPy ou d’un objet io. Ce paramètre est utilisé pour déterminer le format à utiliser lors de l’encodage des données vidéo. Les valeurs acceptées sont “gif”, “mp4”, “webm” ou “ogg”. Si aucune valeur n’est fournie, le format par défaut est “gif”.
Exemples : Enregistrer un tableau NumPy en tant que vidéo 
import numpy as np
import wandb

with wandb.init() as run:
    # les axes sont (nombre d’images, canal, hauteur, largeur)
    frames = np.random.randint(
         low=0, high=256, size=(10, 3, 100, 100), dtype=np.uint8
    )
    run.log({"video": wandb.Video(frames, format="mp4", fps=4)})