image_to_video.py

import cv2
import argparse
import os

parser = argparse.ArgumentParser()
parser.add_argument(
    '-dir',
    '-directory',
    required=False,
    default='ut',
    help='specify location of image folder'
)
parser.add_argument(
    '-n',
    '-name',
    required=False,
    default='merged',
    help='specify location of video'
)
parser.add_argument(
    '-f',
    '-framerate',
    type=int,
    default=30,
    help='specify the frame rate'
)
args = parser.parse_args()

path = args.dir
video_name = args.n + ".mp4"
frame_rate = args.f
images = os.listdir(path)
images = [os.path.join(path, image) for image in images]
images.sort(key=lambda x: os.path.getmtime(x))

img_path = images[-1]
frame = cv2.imread(img_path)
cv2.imshow('video',frame)

h, w = frame.shape[:2]
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
writer = cv2.VideoWriter(video_name, fourcc, frame_rate, (w, h))

for img in images:
    frame = cv2.imread(img)
    writer.write(frame)
    # cv2.imshow('video',frame)
    if (cv2.waitKey(1) & 0xFF) == ord('q'):
        break
writer.release()
cv2.destroyAllWindows()