read_video.py

from tqdm.auto import tqdm
import cv2
import torch

def read_video(file):
    capture = cv2.VideoCapture(file)
    fps = capture.get(cv2.CAP_PROP_FPS)
    n_frames = capture.get(cv2.CAP_PROP_FRAME_COUNT)

    frames = []

    for _ in tqdm(np.arange(n_frames)):
        success, image_cv = capture.read()

        if not success:
            break

        frame = cv2.cvtColor(image_cv, cv2.COLOR_BGR2RGB)
        frame = torch.tensor(frame/255, dtype=torch.float32)
        frames.append(frame.permute(2,0,1)[None])

    return frames, fps, n_frames

# pip install moviepy --upgrade
from moviepy.video.io.ImageSequenceClip import ImageSequenceClip
fps=25.
out_video_path = 'video.mp4'

frames = [(f*255).astype(np.uint8) for f in images]

clip = clip =ImageSequenceClip(frames, fps=fps)
clip.write_videofile(out_video_path, fps=fps)