added barebones GUI interface

Author: hilletc7

opticalFlow/farnebackOpticalFlow.py

@@ -5,17 +5,43 @@
 Right now, in the result visualization, the intensity of a pixel's motion will change both it's color and magnitude.
 Brighter pixels have more motion.
 The output visualization is stored in the same location as the input video with the name <input_vid_filename>_FB_FLOW.mp4
-
 The idea is that perhaps the data about how certain pixels/features are moving across the screen could be used to figure out how the player camera / aim was changing.
 """
 
+from tkinter import *
+from tkinter import filedialog
+from tkinter.messagebox import showinfo
 import numpy as np
 import cv2 as cv
 
+# GUI FILE BROWSER------------------------------------------------------------
+
+window = Tk()
+window.geometry('300x150') # sets the size of the GUI window
+window.title('Select a Video File') # creates a title for the window
+
+# function allowing you to find/select video in GUI
+def get_file_path():
+    global file_path
+    # Open and return file path
+    file_path = filedialog.askopenfilename(title = "Select a Video File", filetypes = (("mp4", "*.mp4"), ("mov files", "*.mov") ,("wmv", "*.wmv"), ("avi", "*.avi")))
+    showinfo(title='Selected File', message=file_path)
+    
+# function allowing you to select the output path in the GUI
+def output():
+    global outpath
+    outpath = filedialog.asksaveasfilename(filetypes=[("mp4", '*.mp4')])
+    window.destroy()
+
+# Creating a button to search for the input file and to select the output destinatio and file name
+b1 = Button(window, text = 'Open a File', command = get_file_path).pack()
+b2 = Button(window, text = 'Save File Name', command = output).pack()
+window.mainloop()
+
 # PARAMETERS--------------------------------
 
 # path to input video file
-vidpath = r""
+vidpath = file_path
 
 # do you want to save the output video?
 savevid = True
@@ -44,11 +70,15 @@
 # create black result image
 hsv_img = np.zeros_like(old_frame)
 hsv_img[...,1] = 255
+# get features from first frame
+print(f"\nRunning farneback Optical Flow on: {vidpath}")
 
 # if saving video
 if savevid:
     # path to save output video
-    savepath = vidpath.split('.')[0] + '_FB_FLOW' + '.mp4'
+    filename = outpath
+    savepath = filename + '_FB_FLOW' + '.mp4'
+    print(f"Saving Output video to: {savepath}")
 
     # get shape of video frames
     height, width, channels = old_frame.shape
@@ -57,33 +87,36 @@
     fourcc = cv.VideoWriter_fourcc(*'mp4v')
     videoOut = cv.VideoWriter(savepath, fourcc, fps, (width, height))
 
+
 # PROCESS VIDEO ---------------------------
 while(True):
     # get frame and convert to grayscale
     _, new_frame = cap.read()
-    new_frame_gray = cv.cvtColor(new_frame, cv.COLOR_BGR2GRAY)
+    if _:
+      new_frame_gray = cv.cvtColor(new_frame, cv.COLOR_BGR2GRAY)
 
     # do Farneback optical flow
-    flow = cv.calcOpticalFlowFarneback(old_frame_gray, new_frame_gray, None, pyr_scale, levels, winsize, iterations, poly_n, poly_sigma, flags)
+      flow = cv.calcOpticalFlowFarneback(old_frame_gray, new_frame_gray, None, pyr_scale, levels, winsize, iterations, poly_n, poly_sigma, flags)
 
     # conversion
-    mag, ang = cv.cartToPolar(flow[...,0], flow[...,1])
+      mag, ang = cv.cartToPolar(flow[...,0], flow[...,1])
 
     # draw onto the result image - color is determined by direction, brightness is by magnitude of motion
     #hsv_img[...,0] = ang*180/np.pi/2
     #hsv_img[...,2] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
 
     # color and brightness by magnitude
-    hsv_img[...,0] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
-    hsv_img[...,1] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
-    hsv_img[...,2] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
+      hsv_img[...,0] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
+      hsv_img[...,1] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
+      hsv_img[...,2] = cv.normalize(mag, None, 0, 255, cv.NORM_MINMAX)
 
-    bgr_img = cv.cvtColor(hsv_img, cv.COLOR_HSV2BGR)
+      bgr_img = cv.cvtColor(hsv_img, cv.COLOR_HSV2BGR)
 
     #  show the image and break out if ESC pressed
-    cv.imshow('Farneback Optical Flow', bgr_img)
-    k = cv.waitKey(30) & 0xff
-    if k == 27:
+      cv.imshow('Farneback Optical Flow', bgr_img)
+      k = cv.waitKey(30) & 0xff
+    else:
+        k == 27
         break
 
     # write frames to new output video
@@ -95,4 +128,7 @@
 
 # cleanup
 videoOut.release()
-cv.destroyAllWindows()
+cv.destroyAllWindows()
+
+# after video is finished
+print('\nComplete!\n')