SaveVideo.py

import cv2
import numpy as np
 
predictions = make_animation(source_image, driving_video, generator, kp_detector, relative=True)


img_array = []
h,w = predictions[0].shape[:2]

out = cv2.VideoWriter('/content/gdrive/My Drive/first-order-motion-model/project.avi',cv2.VideoWriter_fourcc(*'DIVX'), 30, (w,h))
for f in predictions:
    out.write((f[:, :, ::-1]*255).astype(np.uint8))
out.release()