Shortcuts

write_video

torchvision.io.write_video(filename: str, video_array: Tensor, fps: float, video_codec: str = 'libx264', options: Optional[Dict[str, Any]] = None, audio_array: Optional[Tensor] = None, audio_fps: Optional[float] = None, audio_codec: Optional[str] = None, audio_options: Optional[Dict[str, Any]] = None) None[source]

将[T, H, W, C]格式的4D张量写入视频文件

警告

在不久的将来,我们计划将PyTorch的视频解码功能集中在torchcodec项目中。我们鼓励您尝试并提供反馈,因为torchvision的视频解码器最终将被弃用。

Parameters:
  • filename (str) – 视频将保存的路径

  • video_array (Tensor[T, H, W, C]) – 包含单个帧的张量,作为 [T, H, W, C] 格式的 uint8 张量

  • fps (Number) – 视频每秒帧数

  • video_codec (str) – 视频编解码器的名称,例如“libx264”、“h264”等。

  • options (Dict) – 包含要传递给PyAV视频流的选项的字典

  • audio_array (Tensor[C, N]) – 包含音频的张量,其中C是通道数,N是样本数

  • audio_fps (Number) – 音频采样率,通常为44100或48000

  • audio_codec (str) – 音频编解码器的名称,例如“mp3”、“aac”等。

  • audio_options (Dict) – 包含要传递给PyAV音频流的选项的字典