메인 콘텐츠로 건너뛰기

class Video

W&B에 비디오를 로깅하는 클래스입니다.

방법 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
)
W&B 비디오 객체를 초기화합니다. 매개변수:
  • data_or_path: Video는 파일 경로 또는 io 객체로 초기화할 수 있습니다. Video는 numpy 텐서로도 초기화할 수 있습니다. numpy 텐서는 반드시 4차원 또는 5차원이어야 합니다. 차원은 (프레임 수, 채널, 높이, 너비) 또는 (배치, 프레임 수, 채널, 높이, 너비)여야 합니다. numpy 배열이나 io 객체로 초기화할 때는 format 인자로 형식을 지정해야 합니다.
  • caption: 비디오와 함께 표시할 캡션입니다.
  • fps: 원시 비디오 프레임을 인코딩할 때 사용할 프레임 속도입니다. 기본값은 4입니다. 이 매개변수는 data_or_path가 문자열이나 bytes인 경우에는 영향을 주지 않습니다.
  • 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)})