opensora/models/ae/videobase/utils/video_utils.py

import torch
import numpy as np

def tensor_to_video(x):
    x = x.detach().cpu()
    x = torch.clamp(x, -1, 1)
    x = (x + 1) / 2
    x = x.permute(1, 0, 2, 3).float().numpy() # c t h w -> 
    x = (255 * x).astype(np.uint8)
    return x