[python]webカメラの画像を表示し、任意のタイミングで画像を保存する方法

pip install opencv-python
pip install Pillow

import sys
import cv2
import os
import datetime
import tkinter as tk
from PIL import Image, ImageTk
import numpy as np

outfol = 'C:/' #ここに画像を出力保存したいフォルダパスを記述してください

def imwrite(filename, img, params=None):
    try:
        ext = os.path.splitext(filename)[1]
        result, n = cv2.imencode(ext, img, params)

        if result:
            with open(filename, mode='w+b') as f:
                n.tofile(f)
            return True
        else:
            return False
    except Exception as e:
        print(e)
        return False

class Application(tk.Frame):
    def __init__(self, master=None):
        super().__init__(master)
        self.master = master
        self.master.title(u"撮影")
        self.master.geometry("960x640")  # ウィンドウサイズ X1920/2 Y1080/2+100
        self.master.resizable(width=False, height=False)
        self.canvas = tk.Canvas(master, width=960, height=540, bg="white")  # キャンバスサイズをHDの半分に設定
        self.canvas.pack(side=tk.TOP, fill=tk.BOTH, expand=True)
        
        # ラベルの初期化
        self.label1_text = tk.StringVar()
        self.label1_text.set("")  # 初期値を空に設定
        # ラベルを作成
        self.label1 = tk.Label(self.master, textvariable=self.label1_text)
        self.label1.pack(side=tk.TOP)
        
        self.capStart()
        self.create_widgets()

    def update(self):
        global p_img
        global p_frame
        global p_org
        ret, p_frame = cap.read()
        if ret:
            # ウェブカメラから取得したフレームをHDサイズの半分にリサイズ
            p_org=p_frame
            p_frame = cv2.resize(p_frame, (960, 540))
            p_img = ImageTk.PhotoImage(Image.fromarray(cv2.cvtColor(p_frame, cv2.COLOR_BGR2RGB)))
            self.canvas.create_image(960 // 2, 540 // 2, image=p_img)
        else:
            print("u-Fail")
            self.label1_text.set("カメラからの映像取得に失敗しました")
            
        self.master.after(50, self.update)

    def button1_clicked(self, event):
        strtime = '{0:%Y%m%d%H%M%S}'.format(datetime.datetime.now())
        imwrite(outfol + '/' + strtime + '.jpg', p_org)
        self.label1_text.set("画像を保存しました: " + strtime + ".jpg")

    def create_widgets(self):
        button1 = tk.Button(self.master, text=str(u"画像保存"))
        button1.bind("<1>", self.button1_clicked)
        button1.pack(side=tk.TOP)
        self.update()

    def capStart(self):
        print('camera-ON')
        try:
            global cap
            # ウェブカメラの解像度を設定
            cap = cv2.VideoCapture(0)
            cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
            cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)
        except Exception as e:
            print("error:", e)
            cap.release()
            cv2.destroyAllWindows()

def main():
    root = tk.Tk()
    app = Application(master=root)
    app.mainloop()

if __name__ == "__main__":
    main()