outputs a video of a training example

output_training_ex.py

import os
import cv2
import glob

# Set image dir of your choice
imageDir = r'dataset\nervous_frames\_0\\'

# Read in files and sort
filenames = os.listdir(imageDir)
filenames.sort(key= lambda x: int(x[:-4]))
img_array=[]

# Initiate loop over filenames
for filename in filenames:
    img = cv2.imread(imageDir+filename)
    height, width, layers = img.shape
    size = (width,height)
    
    # Store every image array you want to write
    img_array.append(img)
   
# initialize writer and write video
out = cv2.VideoWriter('out.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)

for i in range(len(img_array)):
    out.write(img_array[i])
out.release()