mlaga97
/
PySimpleGUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #538 from MikeTheWatchGuy/master 

Pulling over new OpenCV Webcam Demo
This commit is contained in:
MikeTheWatchGuy 2018-10-22 10:59:39 -04:00 committed by GitHub
parent e96b914680 eacf900075
commit 057d1f6d4b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 127 additions and 54 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
Demo_LED_Indicators.py
View File

@ -1,54 +1,51 @@
#!/usr/bin/env python #!/usr/bin/env python
import sys import sys
if sys.version_info[0] >= 3: if sys.version_info[0] >= 3:
import PySimpleGUI as sg import PySimpleGUI as sg
else: else:
import PySimpleGUI27 as sg import PySimpleGUI27 as sg
import time import time
import random import random
""" """
Demo program showing how to create your own "LED Indicators" Demo program showing how to create your own "LED Indicators"
The LEDIndicator function acts like a new Element that is directly placed in a window's layout The LEDIndicator function acts like a new Element that is directly placed in a window's layout
After the Window is created, use the SetLED function to access the LED and set the color After the Window is created, use the SetLED function to access the LED and set the color
""" """
def LEDIndicator(key=None, radius=30): def LEDIndicator(key=None, radius=30):
return sg.Graph(canvas_size=(radius, radius), return sg.Graph(canvas_size=(radius, radius),
graph_bottom_left=(-radius, -radius), graph_bottom_left=(-radius, -radius),
graph_top_right=(radius, radius), graph_top_right=(radius, radius),
pad=(0, 0), key=key) pad=(0, 0), key=key)
def SetLED(window, key, color): def SetLED(window, key, color):
graph = window.FindElement(key) graph = window.FindElement(key)
graph.Erase() graph.Erase()
graph.DrawCircle((0, 0), 12, fill_color=color, line_color=color) graph.DrawCircle((0, 0), 12, fill_color=color, line_color=color)
layout = [[sg.Text('My LED Status Indicators', size=(20,1))], layout = [[sg.Text('My LED Status Indicators', size=(20,1))],
[sg.Text('CPU Use'), LEDIndicator('_cpu_')], [sg.Text('CPU Use'), LEDIndicator('_cpu_')],
[sg.Text('RAM'), LEDIndicator('_ram_')], [sg.Text('RAM'), LEDIndicator('_ram_')],
[sg.Text('Temperature'), LEDIndicator('_temp_')], [sg.Text('Temperature'), LEDIndicator('_temp_')],
[sg.Text('Server 1'), LEDIndicator('_server1_')], [sg.Text('Server 1'), LEDIndicator('_server1_')],
[sg.RButton('Exit')]] [sg.RButton('Exit')]]
window = sg.Window('My new window', default_element_size=(12, 1), auto_size_text=False).Layout(layout).Finalize() window = sg.Window('My new window', default_element_size=(12, 1), auto_size_text=False).Layout(layout).Finalize()
i = 0 i = 0
while True: # Event Loop while True: # Event Loop
event, value = window.ReadNonBlocking() event, value = window.Read(timeout=400)
if event == 'Exit': if event == 'Exit' or event is None:
window.CloseNonBlocking() break
break if value is None:
if value is None: break
break i += 1
i += 1 SetLED(window, '_cpu_', 'green' if random.randint(1, 10) > 5 else 'red')
SetLED(window, '_cpu_', 'green' if random.randint(1, 10) > 5 else 'red') SetLED(window, '_ram_', 'green' if random.randint(1, 10) > 5 else 'red')
SetLED(window, '_ram_', 'green' if random.randint(1, 10) > 5 else 'red') SetLED(window, '_temp_', 'green' if random.randint(1, 10) > 5 else 'red')
SetLED(window, '_temp_', 'green' if random.randint(1, 10) > 5 else 'red') SetLED(window, '_server1_', 'green' if random.randint(1, 10) > 5 else 'red')
SetLED(window, '_server1_', 'green' if random.randint(1, 10) > 5 else 'red')
time.sleep(.400)

76
Demo_OpenCV_Simple_GUI.py Normal file
View File

@ -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()