如何在Tkinter GUI中显示从硬盘浏览视频?

时间:2018-10-15 06:00:26

标签: python tkinter filedialog

我对Tkinter概念不熟悉,所以我试图制作一个简单的GUI,在其中传输流网络摄像头。除此之外,我还尝试拍摄快照,并使用浏览按钮从硬盘上选择一个视频文件,并在需要时在面板上显示。

这是我的代码:

from PIL import Image, ImageTk
import tkinter as tk
from tkinter import filedialog, messagebox
import argparse
import datetime
import cv2
import os

class Application:
    def __init__(self, output_path = "./"):
        """ Initialize application which uses OpenCV + Tkinter. It displays
            a video stream in a Tkinter window and stores current snapshot on disk """
        self.vs = cv2.VideoCapture(0) # capture video frames, 0 is your default video camera
        self.output_path = output_path  # store output path
        self.current_image = None  # current image from the camera

        self.root = tk.Tk()  # initialize root window
        self.root.title("PyImageSearch PhotoBooth")  # set window title
        # self.destructor function gets fired when the window is closed
        self.root.protocol('WM_DELETE_WINDOW', self.destructor)

        self.panel = tk.Label(self.root)  # initialize image panel
        self.panel.pack(padx=10, pady=10)

        # create a button, that when pressed, will take the current frame and save it to file
        btn = tk.Button(self.root, text="Snapshot!", command=self.take_snapshot)
        btn.pack(fill="both", expand=True, padx=10, pady=10)

        tk.Button(self.root, text="Browse", command=self.loadtemplate, width=10).pack()
        # start a self.video_loop that constantly pools the video sensor
        # for the most recently read frame
        self.video_loop()

    def video_loop(self):
        """ Get frame from the video stream and show it in Tkinter """
        ok, frame = self.vs.read()  # read frame from video stream
        if ok:  # frame captured without any errors
            cv2image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGBA)  # convert colors from BGR to RGBA
            self.current_image = Image.fromarray(cv2image)  # convert image for PIL
            imgtk = ImageTk.PhotoImage(image=self.current_image)  # convert image for tkinter
            self.panel.imgtk = imgtk  # anchor imgtk so it does not be deleted by garbage-collector
            self.panel.config(image=imgtk)  # show the image
        self.root.after(30, self.video_loop)  # call the same function after 30 milliseconds

    def loadtemplate(self):
        filename = tk.filedialog.askopenfilename(filetypes=(("Template files", "*.tplate")
                                                           , ("HTML files", "*.html;*.htm")
                                                           , ("All files", "*.*")))
        if filename:
            try:
                self.root.settings["template"].set(filename)
            except:
                tk.messagebox.showerror("Open Source File", "Failed to read file \n'%s'" % filename)
                return

    def take_snapshot(self):
        """ Take snapshot and save it to the file """
        ts = datetime.datetime.now() # grab the current timestamp
        filename = "{}.jpg".format(ts.strftime("%Y-%m-%d_%H-%M-%S"))  # construct filename
        p = os.path.join(self.output_path, filename)  # construct output path
        self.current_image.save(p, "JPEG")  # save image as jpeg file
        print("[INFO] saved {}".format(filename))

    def destructor(self):
        """ Destroy the root object and release all resources """
        print("[INFO] closing...")
        self.root.destroy()
        self.vs.release()  # release web camera
        cv2.destroyAllWindows()  # it is not mandatory in this application

# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-o", "--output", default="./",
    help="path to output directory to store snapshots (default: current folder")
args = vars(ap.parse_args())

# start the app
print("[INFO] starting...")
pba = Application(args["output"])
pba.root.mainloop()

在此代码中,我可以获取快照,但无法在#self.panel中找到显示浏览视频的方法。

有人可以帮我吗?

1 个答案:

答案 0 :(得分:1)

您可以使用相同的VideoCapture类从网络摄像头源或文件源捕获视频帧,方法是根据设备名传递设备源的设备索引(整数)或构造函数中的文件名(字符串)。 documentation