Files
jonstreeter-ComfyUI-Referen…/utils/~frame2video.py
T
2020-12-01 14:06:29 +08:00

27 lines
691 B
Python

import os
import cv2
video_name = "video.mp4"
frame_folder = "../../results/seq1_new/"
images = [
img for img in os.listdir(frame_folder) if img.endswith(".jpg") or img.endswith(".bmp") or img.endswith(".png")
]
images.sort()
frame = cv2.imread(os.path.join(frame_folder, images[0]))
height, width, layers = frame.shape
video = cv2.VideoWriter("video.avi", cv2.VideoWriter_fourcc("X", "V", "I", "D"), 15, (width, height))
for image in images:
video.write(cv2.imread(os.path.join(frame_folder, image)))
cv2.destroyAllWindows()
video.release()
import imageio
frames = [imageio.imread(os.path.join(frame_folder, image)) for image in images]
imageio.mimsave("movie.gif", frames)