tkinterで作ったGUI上で動画を再生したい

前提・実現したいこと

tkinterで作ったGUI上で動画を再生したいと考えています。

発生している問題・エラーメッセージ

cv2.imshowはtkinterで使えないようで、canvas上で動画が再生されればよいと思ったのですが、その処理がうまくわかりません。また、再生・一時停止・クリック時の時間データを取得する3つのボタンもつけたいと考えています。

該当のソースコード

python
1```import sys
2import tkinter
3from tkinter import *
4from tkinter import ttk
5from tkinter.font import Font
6from PIL import Image, ImageTk
7import cv2
8
9
10
11def button_click():
12    txt.insert('end', "取得したビデオの現在時間\n")
13    txt.see('end')
14    
15#def start_click():
16    
17
18# GUI設定
19#
20root = tkinter.Tk()
21root.title(u"TkinterのCanvasを使ってみる")
22root.geometry("1280x720+125+25")
23root.columnconfigure(0, weight=1)
24root.rowconfigure(0, weight=1)
25
26# Frame
27frame1 = ttk.Frame(root, padding=10)
28frame1.rowconfigure(1, weight=1)
29frame1.columnconfigure(0, weight=1)
30frame1.grid(sticky=(N,W,S,E))
31
32#Button
33button1 = ttk.Button(frame1, text='OK', command=button_click)
34button1.place(x=1100, y=650)
35
36# Text
37f = Font(family='Helvetica', size=11)
38v1 = StringVar()
39txt = Text(frame1,width=10,height=30)
40txt.configure(font=f)
41txt.place(x=1100, y=60)
42
43# Scrollbar
44scrollbar = ttk.Scrollbar(
45    frame1, 
46    orient=VERTICAL, 
47    command=txt.yview)
48txt['yscrollcommand'] = scrollbar.set
49scrollbar.place(x=1200, y=60,height=525)
50
51
52
53#Canvas
54canvas = tkinter.Canvas(root, width = 800, height = 450)#Canvasの作成
55#canvas.create_rectangle(0, 0, 1600, 900, fill = 'green')#塗りつぶし
56canvas.place(x=100, y=100)#Canvasの配置
57
58
59
60root.mainloop()
61
62### 試したこと
63
64canvas.create_imageに動画データを入れようとしましたがうまくできませんでした。
65
66### 補足情報（FW/ツールのバージョンなど）
67
68ここにより詳細な情報を記載してください。

BeatStar

2019/10/25 06:22

質問のコード、「第三者として」読んでみてください。読みやすいですか? 読みやすいようにしてください。編集で"<コード>"(または"<code>") となっているボタンを押して、それを使って書いてください。

行動規範の内容に同意します

回答1件

こんな感じでいかがでしょう。

python
1#! /usr/bin/python3
2import tkinter
3import cv2
4import PIL.Image, PIL.ImageTk
5import time
6
7class App:
8    def __init__(self, window, window_title, video_source=0):
9        self.window = window
10        self.window.title(window_title)
11        self.video_source = video_source
12        self.vid = MyVideoCapture(self.video_source)
13        self.canvas = tkinter.Canvas(window, width = self.vid.width, height = self.vid.height)
14        self.canvas.pack()
15        self.btn_snapshot=tkinter.Button(window, text="Snapshot", width=50, command=self.snapshot)
16        self.btn_snapshot.pack(anchor=tkinter.CENTER, expand=True)
17        self.delay = 15
18        self.update()
19        self.window.mainloop()
20    def snapshot(self):
21        ret, frame = self.vid.get_frame()
22        if ret:
23            cv2.imwrite("frame-" + time.strftime("%d-%m-%Y-%H-%M-%S") + ".jpg", cv2.cvtColor(frame, cv2.COLOR_RGB2BGR))
24    def update(self):
25        ret, frame = self.vid.get_frame()
26        if ret:
27            self.photo = PIL.ImageTk.PhotoImage(image = PIL.Image.fromarray(frame))
28            self.canvas.create_image(0, 0, image = self.photo, anchor = tkinter.NW)
29
30        self.window.after(self.delay, self.update)
31class MyVideoCapture:
32    def __init__(self, video_source=0):
33        # Open the video source
34        self.vid = cv2.VideoCapture(video_source)
35        if not self.vid.isOpened():
36            raise ValueError("Unable to open video source", video_source)
37
38        self.width = self.vid.get(cv2.CAP_PROP_FRAME_WIDTH)
39        self.height = self.vid.get(cv2.CAP_PROP_FRAME_HEIGHT)
40    def get_frame(self):
41        if self.vid.isOpened():
42            ret, frame = self.vid.read()
43            if ret:
44                return (ret, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
45            else:
46                return (ret, None)
47        else:
48            return (ret, None)
49    def __del__(self):
50        if self.vid.isOpened():
51            self.vid.release()
52App(tkinter.Tk(), "Tkinter and OpenCV")