43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import numpy as np
|
|
import imageio
|
|
from PIL import Image
|
|
|
|
def write_video(file_path, frames, fps, reversed=True, start_frame_dupe_amount=15, last_frame_dupe_amount=30):
|
|
"""
|
|
Writes frames to an mp4 video file
|
|
:param file_path: Path to output video, must end with .mp4
|
|
:param frames: List of PIL.Image objects
|
|
:param fps: Desired frame rate
|
|
:param reversed: if order of images to be reversed (default = True)
|
|
"""
|
|
if reversed == True:
|
|
frames = frames[::-1]
|
|
|
|
# Get dimensions of the frames
|
|
w, h = frames[0].size
|
|
|
|
# Create an imageio video writer
|
|
writer = imageio.get_writer(file_path, fps=fps)
|
|
|
|
# Duplicate the start and end frames
|
|
start_frames = [frames[0]] * start_frame_dupe_amount
|
|
end_frames = [frames[-1]] * last_frame_dupe_amount
|
|
|
|
# Write the duplicated frames to the video writer
|
|
for frame in start_frames:
|
|
# Convert PIL image to numpy array
|
|
np_frame = np.array(frame)
|
|
writer.append_data(np_frame)
|
|
|
|
# Write the frames to the video writer
|
|
for frame in frames:
|
|
np_frame = np.array(frame)
|
|
writer.append_data(np_frame)
|
|
|
|
# Write the duplicated frames to the video writer
|
|
for frame in end_frames:
|
|
np_frame = np.array(frame)
|
|
writer.append_data(np_frame)
|
|
|
|
# Close the video writer
|
|
writer.close() |