FilesExpand file tree

 master

/

photoboothapp.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

106 lines (90 loc) · 3.65 KB

 master

/

photoboothapp.py

Copy path

Top

File metadata and controls

• Code
• Blame

106 lines (90 loc) · 3.65 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

# -*- coding: utf-8 -*-

# @Time : 2018/2/8 16:10

# @Author : play4fun

# @File : photoboothapp.py

# @Software: PyCharm

"""

photoboothapp.py:

"""

# import the necessary packages

from __future__ import print_function

from PIL import Image

from PIL import ImageTk

import tkinter as tki

import threading

import datetime

import imutils

import cv2

import os

class PhotoBoothApp:

def __init__(self, vs, outputPath):

# store the video stream object and output path, then initialize

# the most recently read frame, thread for reading frames, and

# the thread stop event

self.vs = vs

self.outputPath = outputPath

self.frame = None

self.thread = None

self.stopEvent = None

# initialize the root window and image panel

self.root = tki.Tk()

self.panel = None

# create a button, that when pressed, will take the current

# frame and save it to file

btn = tki.Button(self.root, text="Snapshot!",

command=self.takeSnapshot)

btn.pack(side="bottom", fill="both", expand="yes", padx=10,

pady=10)

# start a thread that constantly pools the video sensor for

# the most recently read frame

self.stopEvent = threading.Event()

self.thread = threading.Thread(target=self.videoLoop, args=())

self.thread.start()

# set a callback to handle when the window is closed

self.root.wm_title("PyImageSearch PhotoBooth")

self.root.wm_protocol("WM_DELETE_WINDOW", self.onClose)

def videoLoop(self):

# DISCLAIMER:

# I'm not a GUI developer, nor do I even pretend to be. This

# try/except statement is a pretty ugly hack to get around

# a RunTime error that throws due to threading

try:

# keep looping over frames until we are instructed to stop

while not self.stopEvent.is_set():

# grab the frame from the video stream and resize it to

# have a maximum width of 300 pixels

self.frame = self.vs.read()

self.frame = imutils.resize(self.frame, width=300)

# represents images in BGR order; however PIL

# represents images in RGB order, so we need to swap

# the channels, then convert to PIL and ImageTk format

image = cv2.cvtColor(self.frame, cv2.COLOR_BGR2RGB)

image = Image.fromarray(image)

image = ImageTk.PhotoImage(image)

# if the panel is not None, we need to initialize it

if self.panel is None:

self.panel = tki.Label(image=image)

self.panel.image = image

self.panel.pack(side="left", padx=10, pady=10)

# otherwise, simply update the panel

else:

self.panel.configure(image=image)

self.panel.image = image

except RuntimeError as e:

print("[INFO] caught a RuntimeError")

def takeSnapshot(self):

# grab the current timestamp and use it to construct the

# output path

ts = datetime.datetime.now()

filename = "{}.jpg".format(ts.strftime("%Y-%m-%d_%H-%M-%S"))

p = os.path.sep.join((self.outputPath, filename))

# save the file

cv2.imwrite(p, self.frame.copy())

print("[INFO] saved {}".format(filename))

def onClose(self):

# set the stop event, cleanup the camera, and allow the rest of

# the quit process to continue

print("[INFO] closing...")

self.stopEvent.set()

self.vs.stop()

self.root.quit()