view-video.py

import cv2
import local_common as cm
import os
import sys

vid_path = "out-video.avi"
csv_path = "out-key.csv"

font = cv2.FONT_HERSHEY_SIMPLEX

if len(sys.argv) == 3:
    vid_path = sys.argv[1]
    csv_path = sys.argv[2]
elif len(sys.argv) == 2:
    epoch_id = int(sys.argv[1])
    vid_path = 'epochs/out-video-{}.avi'.format(epoch_id)
    csv_path = 'epochs/out-key-{}.csv'.format(epoch_id)

assert os.path.isfile(vid_path)
assert os.path.isfile(csv_path)

frame_count = cm.frame_count(vid_path)
cap = cv2.VideoCapture(vid_path)
imgs = []
rows = cm.fetch_csv_data(csv_path)
assert frame_count == len(rows)
for i in range(frame_count):
    ret, img = cap.read()
    text = 'frame: {}, wheel: {}'.format(i, rows[i]['wheel'])
    cv2.putText(img, text, (50,50), font, 1, (255,255,255))
    imgs.append(img)

frame_id = 0
while True:
    cv2.imshow('frame', imgs[frame_id])
    ch = cv2.waitKey(0) # wait 
    if ch == ord('q'):
        break
    elif ch == ord('l'):
        frame_id = min(frame_id + 1, frame_count -1 )
    elif ch == ord('j'):
        frame_id = max(frame_id - 1, 0)