utils.py

## Defines utility functions

import numpy as np
import cv2

# cv2.cv.CV_FOURCC(*'avc1')
AVC1_CODEC = 828601953

# Utility functions
def display_img(img, title):
    cv2.imshow(title, img)
    cv2.waitKey(0)
    return

def read_img(path):
    img = cv2.imread(path, 0)
    return img

def write_img_array_to_video(img_arr, fps, video_path):
    num_frames = len(img_arr)
    print(num_frames)
    height,width,layers = img_arr[0].shape
    print("Height", height)
    print("Width", width)
    print("Layers", layers)

    video = cv2.VideoWriter(video_path,cv2.VideoWriter_fourcc(*'MJPG'),fps,(width,height))

    for i in range(num_frames):
        video.write(img_arr[i])
    video.release()
    cv2.destroyAllWindows()
    return video

# [(13, 159), (12, 159), (14, 150), (12, 218), (13, 218), (12, 150), (12, 184), (15, 142), (55, 258), (478, 394), (130, 188), (13, 184), (12, 181), (13, 175), (12, 142), (55, 260), (122, 191), (51, 259), (480, 397), (481, 396), (12, 223), (12, 207), (137, 218), (51, 261), (481, 398), (276, 525), (481, 397), (13, 207), (482, 398), (53, 125), (189, 240), (482, 396), (13, 223), (483, 398), (46, 125), (122, 212), (482, 397), (193, 235), (480, 397), (396, 567), (483, 396), (481, 394), (114, 211), (483, 400), (484, 400), (30, 129), (29, 131), (69, 180), (485, 400), (484, 398)]
# [(13, 159), (12, 159), (14, 150), (12, 218), (13, 218), (12, 150), (12, 184), (15, 142), (55, 258), (479, 396), (133, 189), (13, 184), (12, 181), (13, 175), (12, 142), (55, 260), (124, 191), (51, 259), (481, 397), (481, 396), (12, 223), (12, 207), (133, 210), (51, 261), (481, 398), (276, 525), (482, 397), (13, 207), (482, 398), (53, 125), (189, 236), (482, 396), (13, 223), (483, 398), (46, 125), (122, 211), (483, 397), (189, 233), (481, 395), (396, 567), (483, 396), (481, 394), (114, 211), (481, 399), (482, 399), (30, 129), (29, 131), (69, 180), (483, 399), (484, 398)]

# [(13, 160, 317719.39865449222), (12, 160, 304116.78355076991), (14, 224, 289199.24145546299), (14, 218, 281879.15474771522), (55, 262, 280789.85818725626), (12, 185, 278768.06377962232), (13, 150, 263040.41247421323), (55, 259, 261136.38952423813), (12, 224, 257943.05044618886), (13, 185, 255851.94050954262), (12, 218, 252470.1189210072), (12, 150, 251528.73462657284), (189, 237, 246109.07460305188), (14, 169, 243682.4779405208), (479, 397, 240616.02394474586), (481, 397, 231654.83742682115), (277, 525, 230266.49775520258), (12, 181, 228392.33778256187), (482, 397, 220019.39755743326), (133, 189, 219462.46027778156), (16, 142, 215974.79732300149), (481, 398, 213127.92548179833), (482, 398, 212665.03228916042), (51, 261, 210984.18519426059), (483, 398, 205596.08920690604), (51, 258, 205033.27557878138), (189, 233, 203993.92750789211), (121, 212, 203626.11201317093), (12, 195, 201586.4550127277), (483, 397, 200453.32810995704), (124, 191, 197746.11927596063), (54, 125, 196576.87902675185), (12, 142, 194582.15688379627), (481, 396, 194398.89062769298), (484, 398, 188164.86228907306), (481, 399, 187440.08040481652), (482, 396, 187021.63312091833), (482, 399, 186775.63475386216), (195, 237, 185297.63732952485), (481, 400, 183840.45199291638), (483, 399, 183401.06889570752), (482, 400, 181597.36008854286), (396, 567, 181520.34240808664), (116, 214, 181406.70657702861), (484, 397, 180513.93712358637), (14, 207, 180021.49047027651), (483, 400, 178231.18853109822), (484, 399, 175832.0191821777), (484, 400, 175801.31720121493), (481, 401, 174455.31459572635)]
# [(13, 160, 317719.39865449222), (12, 160, 304116.78355076991), (14, 224, 289199.24145546299), (14, 218, 281879.15474771522), (55, 262, 280789.85818725626), (12, 185, 278768.06377962232), (13, 150, 263040.41247421323), (55, 259, 261136.38952423813), (12, 224, 257943.05044618886), (13, 185, 255851.94050954262), (12, 218, 252470.1189210072), (12, 150, 251528.73462657284), (189, 237, 246109.07460305188), (14, 169, 243682.4779405208), (479, 397, 240616.02394474586), (481, 397, 231654.83742682115), (277, 525, 230266.49775520258), (12, 181, 228392.33778256187), (482, 397, 220019.39755743326), (133, 189, 219462.46027778156), (16, 142, 215974.79732300149), (481, 398, 213127.92548179833), (482, 398, 212665.03228916042), (51, 261, 210984.18519426059), (483, 398, 205596.08920690604), (51, 258, 205033.27557878138), (189, 233, 203993.92750789211), (121, 212, 203626.11201317093), (12, 195, 201586.4550127277), (483, 397, 200453.32810995704), (124, 191, 197746.11927596063), (54, 125, 196576.87902675185), (12, 142, 194582.15688379627), (481, 396, 194398.89062769298), (484, 398, 188164.86228907306), (481, 399, 187440.08040481652), (482, 396, 187021.63312091833), (482, 399, 186775.63475386216), (195, 237, 185297.63732952485), (481, 400, 183840.45199291638), (483, 399, 183401.06889570752), (482, 400, 181597.36008854286), (396, 567, 181520.34240808664), (116, 214, 181406.70657702861), (484, 397, 180513.93712358637), (14, 207, 180021.49047027651), (483, 400, 178231.18853109822), (484, 399, 175832.0191821777), (484, 400, 175801.31720121493), (481, 401, 174455.31459572635)]