简易的拍照录像使用python,opencv写的

import cv2
import tkinter as tk
from tkinter import messagebox
from datetime import datetime
import threading
import io

class CameraApp:
    def __init__(self, master):
        self.master = master
        self.master.title("图像采集")

        self.capture = cv2.VideoCapture(0)
        self.is_recording = False
        self.video_writer = None

        # 创建拍照和拍视频按钮
        self.btn_capture = tk.Button(master, text="拍照",width=50, command=self.take_photo)
        self.btn_capture.pack(pady=10)

        self.btn_record = tk.Button(master, text="拍视频", width=50,command=self.toggle_recording)
        self.btn_record.pack(pady=10)

        self.master.protocol("WM_DELETE_WINDOW", self.on_closing)

        # 创建一个标签用于显示摄像头画面
        self.canvas = tk.Canvas(master, width=640, height=480)
        self.canvas.pack()
        self.update_frame()

    def update_frame(self):
        ret, frame = self.capture.read()
        if ret:
            # 转换颜色格式从 BGR 到 RGB
            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            # 将图像转换为 PhotoImage 并在画布上显示
            self.photo = tk.PhotoImage(data=self.cv_to_img(frame))
            self.canvas.create_image(0, 0, image=self.photo, anchor=tk.NW)
        self.master.after(10, self.update_frame)

    def cv_to_img(self, img):
        """将 OpenCV 图像转换为 PhotoImage 格式"""
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        pil_img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
        pil_img = cv2.imencode('.png', pil_img)[1].tobytes()
        return pil_img

    def take_photo(self):
        ret, frame = self.capture.read()
        if ret:
            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
            filename = f"photo_{timestamp}.jpg"
            cv2.imwrite(filename, frame)
            messagebox.showinfo("信息", f"照片已保存为 {filename}")

    def toggle_recording(self):
        if self.is_recording:
            self.stop_recording()
        else:
            self.start_recording()

    def start_recording(self):
        self.is_recording = True
        self.btn_record.config(text="完成")

        # Start a separate thread for video recording
        threading.Thread(target=self.record_video).start()

    def record_video(self):
        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
        filename = f"video_{timestamp}.mp4"
        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
        self.video_writer = cv2.VideoWriter(filename, fourcc, 20.0, (640, 480))

        while self.is_recording:
            ret, frame = self.capture.read()
            if ret:
                self.video_writer.write(frame)

        self.video_writer.release()
        messagebox.showinfo("信息", f"视频已保存为 {filename}")
        self.btn_record.config(text="拍视频")

    def stop_recording(self):
        self.is_recording = False

    def on_closing(self):
        self.capture.release()
        if self.is_recording:
            self.stop_recording()
        self.master.destroy()

if __name__ == "__main__":
    root = tk.Tk()
    app = CameraApp(root)
    root.mainloop()