メインコンテンツへスキップ
GitHub source

class Video

W&B にビデオを ログ するためのクラス。

method Video.__init__

__init__(
    data_or_path: Union[str, pathlib.Path, ForwardRef('np.ndarray'), ForwardRef('TextIO'), ForwardRef('BytesIO')],
    caption: Optional[str] = None,
    fps: Optional[int] = None,
    format: Optional[Literal['gif', 'mp4', 'webm', 'ogg']] = None
)
W&B Video オブジェクト を初期化します。 引数:
  • data_or_path: ビデオはファイルへのパスまたは io オブジェクトで初期化できます。また、 numpy テンソルでも初期化可能です。 numpy テンソルは 4 次元または 5 次元である必要があります。次元は (フレーム数, チャンネル, 高さ, 幅) または (バッチ, フレーム数, チャンネル, 高さ, 幅) である必要があります。 numpy 配列または io オブジェクトで初期化する場合、 format 引数でフォーマットを指定する必要があります。
  • caption: 表示用にビデオに関連付けられるキャプション。
  • fps: 生のビデオフレームをエンコードする際に使用するフレームレート。デフォルト 値 は 4 です。この パラメータ は、 data_or_path が文字列またはバイト列の場合は効果がありません。
  • format: ビデオのフォーマット。 numpy 配列または io オブジェクトで初期化する場合に必要です。この パラメータ は、ビデオ データをエンコードする際のフォーマットを決定するために使用されます。受け入れられる 値 は “gif”、 “mp4”、 “webm”、 “ogg” です。 値 が提供されない場合、デフォルトのフォーマットは “gif” になります。
例: numpy 配列をビデオとして ログ する
import numpy as np
import wandb

with wandb.init() as run:
    # 軸は (フレーム数, チャンネル, 高さ, 幅) です
    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)})