首页 文章

使用opencv和tkinter进行网络摄像头

提问于
浏览
0

我在python中使用 opencvtkinter 编写了一个程序,它可以显示视频网络摄像头,我可以快照 . 我有一个名为快照的文件夹,文件名保存为时间戳,但我想用自己的(用户)而不是时间戳来保存文件名 . 我不知道,我是python和tkinter的初学者 . 你能帮我么?

以下是我的photoboothapp代码:

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):
        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
                btm = tki.Button(self.root, text="name",command=callback)
        btn = tki.Button(self.root, text="Snapshot!",
            command=self.takeSnapshot)

                #btm = tki.Button(self.root, text="name",command=callback)

                #btm=tki.Button(self.root,text="name",command=callback)
        btn.pack(side="bottom", fill="both", expand="yes", padx=10,
            pady=10)
                #btm.pack(side="bottom",fill="both",expand="yes", padx=15,pady=15)
        # 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 Tkinter 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)

                # OpenCV 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()

1 回答

  • 0

    您可以使用tkinter python的Entry小部件 .

    按钮上方添加此行 .

    Text=Entry(self.root)
    Text.pack()
    

    并且在takeSnapshot()函数中,在写入时执行此操作 .

    p=os.path.sep.join((Text.get(), filename))
    

    我已经编辑了您的代码以便更好地理解 .

    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):
            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
            btm = tki.Button(self.root, text="name",command=callback)
            Text=Entry(self.root)
            Text.pack()
            btn = tki.Button(self.root, text="Snapshot!",
                command=self.takeSnapshot)
    
                    #btm = tki.Button(self.root, text="name",command=callback)
    
                    #btm=tki.Button(self.root,text="name",command=callback)
            btn.pack(side="bottom", fill="both", expand="yes", padx=10,
                pady=10)
                    #btm.pack(side="bottom",fill="both",expand="yes", padx=15,pady=15)
            # 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 Tkinter 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)
    
                    # OpenCV 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((Text.get(), 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()
    

相关问题