pyav保存视频
目录
imageio替代pyav
imageio替代pyav
import imageio
import numpy as np
import torch
# 创建一个随机的图像张量，形状为 (N, C, H, W)
# 这里 N = 30（帧数），C = 3（通道数），H = 64（高度），W = 64（宽度）
N, C, H, W = 30, 3, 64, 64
tensor_data = torch.rand(N, C, H, W) # 随机生成数据
# 将张量转换为 NumPy 数组，并调整维度为 (N, H, W, C)
tensor_data = tensor_data.permute(0, 2, 3, 1).numpy()
# 设置输出视频文件路径和帧率
images_path = 'output_video.mp4'
fps = 8
# 使用 imageio 写入视频
with imageio.get_writer(images_path, fps=fps, codec='h264', ffmpeg_log_level='error') as writer:
for frame in tensor_data:
writer.append_data((frame * 255).astype(np.uint8)) # 将数据转换为 uint8 类型
print(f"视频已保存至 {images_path}")