Create video from png files

vid-create.py

#!/usr/local/bin/python3

#Needs to install ffmpeg
#Run: python vid-create.py
#ffmpeg -i input.avi output.mp4

#pip install opencv-python
##########

import cv2
import argparse
import os
import functools
from functools import cmp_to_key

#Function to check if string can be cast to int
def isnum (num):
    try:
        int(num)
        return True
    except:
        return False

#Numerically sorts filenames
def image_sort (x,y):
    x = int(x.split(".")[0])
    y = int(y.split(".")[0])
    return x-y

# Construct the argument parser and parse the arguments
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument("-e", "--extension", required=False, default='png', help="Extension name. default is 'png'.")
arg_parser.add_argument("-o", "--output", required=False, default='output.mp4', help="Output video file.")
arg_parser.add_argument("-d", "--directory", required=False, default='.', help="Specify image directory.")
arg_parser.add_argument("-fps", "--framerate", required=False, default='10', help="Set the video framerate.")
arg_parser.add_argument("-s", "--sort", required=False, default='numeric', help="Determines the type of file-order sort that will be used.")
arg_parser.add_argument("-t", "--time", required=False, default='none', help="Sets the framerate so that the video length matches the time in seconds.")
arg_parser.add_argument("-v", "--visual", required=False, default='false', help="If 'true' then will display preview window.")
arg_parser.add_argument("-safe", "--safe", required=False, default='true', help="If 'false' then will try to render all images, not just consistenly-sized ones.")
args = vars(arg_parser.parse_args())

# Arguments
dir_path = args['directory']
ext = args['extension']
output = args['output']
framerate = args['framerate']
sort_type = args['sort']
time = args['time']
visual = args['visual']

#Flips 'visual' to a bool
visual = visual == "true"
safe = args['safe'] == "true"

#Sets the framerate to argument, or defaults to 10
if not isnum(framerate):
    framerate = 10
else:
    framerate = int(framerate)

#Get the files from directory
images = []
for f in os.listdir(dir_path):
    if f.endswith(ext):
        images.append(f)

#Sort the files found in the directory
if sort_type == "numeric":
    int_name = images[0].split(".")[0]
    if isnum(int_name):
        images = sorted(images, key=cmp_to_key(image_sort))
    else:
        print("Failed to sort numerically, switching to alphabetic sort")
        images.sort()
elif sort_type == "alphabetic":
    images.sort()

#Change framerate to fit the time in seconds if a time has been specified.
#Overrides the -fps arg
if isnum(time):
    framerate = int(len(images) / int(time))
    print("Adjusting framerate to " + str(framerate))

# Determine the width and height from the first image
image_path = os.path.join(dir_path, images[0])
frame = cv2.imread(image_path)

if visual:
    cv2.imshow('video',frame)
regular_size = os.path.getsize(image_path)
height, width, channels = frame.shape
print('height:', height)

# Define the codec and create VideoWriter object
#fourcc = cv2.VideoWriter_fourcc(*'mp4v') # Be sure to use lower case
fourcc = cv2.cv.CV_FOURCC(*'XVID')
out = cv2.VideoWriter('video.avi', fourcc, framerate, (width, height))

for n, image in enumerate(images):
    image_path = os.path.join(dir_path, image)
    image_size = os.path.getsize(image_path)
    if image_size < regular_size / 1.5 and safe:
        print("Cancelled: " + image)
        continue

    frame = cv2.imread(image_path)
    font = cv2.FONT_HERSHEY_SIMPLEX
    cv2.putText(frame, image, (230, 50), font, 0.8, (35, 35, 35), 2, cv2.CV_AA)
    out.write(frame) # Write out frame to video
    if visual:
        cv2.imshow('video', frame)

    if (cv2.waitKey(1) & 0xFF) == ord('q'): # Hit `q` to exit
        break
    if n%100 == 0:
        print("path: " + image_path)
        print("Frame " + str(n))

# Release everything if job is finished
#out.release()
#cv2.destroyAllWindows()

print("The output video is video.avi")