Merge pull request #534 from jackyOO7/patch-5

Create Demo_OpenCV_Simple_GUI.py

Demo_OpenCV_Simple_GUI.py

@@ -0,0 +1,76 @@
+import sys
+if sys.version_info[0] >= 3:
+  import PySimpleGUI as sg
+else:
+  import PySimpleGUI27 as sg
+import cv2
+import numpy as np
+from sys import exit as exit
+
+"""
+Demo program that displays a webcam using OpenCV and applies some very basic image functions
+
+- functions from top to bottom -
+none:       no processing
+threshold:  simple b/w-threshold on the luma channel, slider sets the threshold value
+canny:      edge finding with canny, sliders set the two threshold values for the function => edge sensitivity
+contour:    colour finding in the frame, first slider sets the hue for the colour to find, second the minimum saturation
+            for the object. Found objects are drawn with a red contour.
+blur:       simple Gaussian blur, slider sets the sigma, i.e. the amount of blur smear
+hue:        moves the image hue values by the amount selected on the slider
+enhance:    applies local contrast enhancement on the luma channel to make the image fancier - slider controls fanciness.
+"""
+def main():
+
+  sg.ChangeLookAndFeel('LightGreen')
+
+  # define the window layout
+  layout = [[sg.Text('OpenCV Demo', size=(40, 1), justification='center')],
+            [sg.Image(filename='', key='image')],
+            [sg.Radio('None', 'Radio',  True, size=(10, 1))],
+            [sg.Radio('threshold', 'Radio', size=(10, 1),key='thresh'),sg.Slider((0,255),128,1,orientation='h', size=(40, 15),key='thresh_slider')],
+            [sg.Radio('canny', 'Radio', size=(10, 1), key='canny'),sg.Slider((0, 255), 128, 1, orientation='h', size=(20, 15), key='canny_slider_a'),sg.Slider((0, 255), 128, 1, orientation='h', size=(20, 15), key='canny_slider_b')],
+            [sg.Radio('contour', 'Radio', size=(10, 1), key='contour'),sg.Slider((0, 255), 128, 1, orientation='h', size=(20, 15), key='contour_slider'),sg.Slider((0, 255), 80, 1, orientation='h', size=(20, 15), key='base_slider')],
+            [sg.Radio('blur', 'Radio', size=(10, 1),key='blur'),sg.Slider((1,11),1,1,orientation='h', size=(40, 15),key='blur_slider')],
+            [sg.Radio('hue', 'Radio', size=(10, 1), key='hue'),sg.Slider((0, 225), 0, 1, orientation='h', size=(40, 15), key='hue_slider')],
+            [sg.Radio('enhance', 'Radio', size=(10, 1),key='enhance'),sg.Slider((1,255),128,1,orientation='h', size=(40, 15),key='enhance_slider')],
+            [sg.ReadButton('Exit', size=(10, 1))]]
+
+  # create the window and show it without the plot
+  window = sg.Window('Demo Application - OpenCV Integration',
+                     location=(800,400))
+  window.Layout(layout).Finalize()
+
+  cap = cv2.VideoCapture(0)
+  while True:
+    event, values = window.ReadNonBlocking()
+    if event == 'Exit' or values is None:
+      sys.exit(0)
+    ret, frame = cap.read()
+    if values['thresh']:
+      frame=cv2.cvtColor(frame,cv2.COLOR_BGR2LAB)[:,:,0]
+      _,frame=cv2.threshold(frame,values['thresh_slider'],255,cv2.THRESH_BINARY)
+    if values['canny']:
+      frame=cv2.Canny(frame,values['canny_slider_a'],values['canny_slider_b'])
+    if values['blur']:
+      frame=cv2.GaussianBlur(frame,(21,21),values['blur_slider'])
+    if values['hue']:
+      frame=cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)
+      frame[:,:,0]+=values['hue_slider']
+      frame=cv2.cvtColor(frame,cv2.COLOR_HSV2BGR)
+    if values['enhance']:
+      enh_val=values['enhance_slider']/40
+      clahe=cv2.createCLAHE(clipLimit=enh_val, tileGridSize=(8,8))
+      lab=cv2.cvtColor(frame,cv2.COLOR_BGR2LAB)
+      lab[:,:,0]=clahe.apply(lab[:,:,0])
+      frame=cv2.cvtColor(lab,cv2.COLOR_LAB2BGR)
+    if values['contour']:
+      hue=cv2.cvtColor(frame,cv2.COLOR_BGR2HSV)
+      hue=cv2.GaussianBlur(hue,(21,21),1)
+      hue=cv2.inRange(hue,np.array([values['contour_slider'],values['base_slider'],40]),np.array([values['contour_slider']+30,255,220]))
+      _,cnts,_=cv2.findContours(hue,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
+      cv2.drawContours(frame,cnts,-1,(0,0,255),2)
+    imgbytes=cv2.imencode('.png', frame)[1].tobytes() #ditto
+    window.FindElement('image').Update(data=imgbytes)
+
+main()