天天看点

用python给朋友一个监控软件

作者:字节巫师

先看运行效果

用python给朋友一个监控软件

一、技术要点:

1、用tk做软件界面;

2、用CV2完成视频采集;

注意CV2来自opencv-python,版本号与python版本要注意

我的python3.11,opencv-python4.7.0

pip install opencv-python==4.7.0

二、源码

import cv2

import tkinter as tk

from PIL import ImageTk, Image

import os

from datetime import datetime

cap = cv2.VideoCapture(0)

out = None

is_recording = False

def take_photo():

# 读取视频帧

ret, frame = cap.read()

if ret:

# 生成文件名

now = datetime.now()

timestamp = now.strftime("%Y-%m-%d_%H-%M-%S-%f")

img_name = "images/img_{}.png".format(timestamp)

# 保存照片

cv2.imwrite(img_name, frame)

log_text.insert(tk.END, "拍照成功:{}\n".format(img_name))

# 显示照片在第二个Label上

image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

image = Image.fromarray(image)

image = image.resize((video_label.winfo_width() // 4, video_label.winfo_height() // 4))

photo_image = ImageTk.PhotoImage(image)

photo_label.config(image=photo_image)

photo_label.image = photo_image

# 显示照片在第一个Label上

frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

frame = Image.fromarray(frame)

frame = ImageTk.PhotoImage(frame)

video_label.config(image=frame)

video_label.image = frame

def record_video():

global out, record_btn, stop_btn,video_name

# 创建保存视频的文件夹

if not os.path.exists('movies'):

os.makedirs('movies')

# 录制视频

now = datetime.now()

timestamp = now.strftime("%Y-%m-%d_%H-%M-%S-%f")

video_name = "movies/video_{}.avi".format(timestamp)

is_recording = True

fps = 20.0

fourcc = cv2.VideoWriter_fourcc(*'XVID')

out = cv2.VideoWriter(video_name, fourcc, fps, (640, 480))

out.write(frame)

log_text.insert(tk.END, "开始录制视频:{}\n".format(video_name))

# 更新按钮状态

record_btn.config(state=tk.DISABLED)

stop_btn.config(state=tk.NORMAL)

def stop_video():

global out, record_btn, stop_btn

# 停止录制视频

is_recording = False

out.release()

cv2.destroyAllWindows()

log_text.insert(tk.END, "录制完成\n")

# 更新按钮状态

record_btn.config(state=tk.NORMAL)

stop_btn.config(state=tk.DISABLED)

def save_log():

log_content = log_text.get("1.0", tk.END)

with open("log.txt", "a") as file:

file.write(log_content)

log_text.insert(tk.END, "日志保存成功\n")

def show_video():

global video_label, photo_label, record_btn, stop_btn, log_text,out

# 创建GUI窗口

window = tk.Tk()

window.title("摄像头监控系统")

# 创建用于显示视频的标签

video_label = tk.Label(window)

video_label.pack(padx=10, pady=10)

# 创建用于显示拍照照片的标签

photo_label = tk.Label(window)

photo_label.place(x=video_label.winfo_width() - video_label.winfo_width() // 4,

y=video_label.winfo_height() - video_label.winfo_height() // 4,

width=video_label.winfo_width() // 4, height=video_label.winfo_height() // 4)

# 创建按钮容器

btn_frame = tk.Frame(window)

btn_frame.pack(padx=10, pady=10)

# 创建拍照按钮

photo_btn = tk.Button(btn_frame, text="拍照", command=take_photo)

photo_btn.pack(side=tk.LEFT, padx=10)

# 创建录像按钮

record_btn = tk.Button(btn_frame, text="开始录像", command=record_video)

record_btn.pack(side=tk.LEFT, padx=10)

# 创建停止录像按钮

stop_btn = tk.Button(btn_frame, text="停止录像", command=stop_video, state=tk.DISABLED)

stop_btn.pack(side=tk.LEFT, padx=10)

# 创建滚动条框架

scrollbar_frame = tk.Frame(window)

scrollbar_frame.pack(padx=10, pady=10)

# 创建滚动条

scrollbar = tk.Scrollbar(scrollbar_frame)

scrollbar.pack(side=tk.RIGHT, fill=tk.Y)

# 创建日志文本框

log_text = tk.Text(scrollbar_frame, height=10, width=80)

log_text.pack(side=tk.LEFT, fill=tk.BOTH, padx=5, pady=5)

# 绑定滚动条和文本框

scrollbar.config(command=log_text.yview)

log_text.config(yscrollcommand=scrollbar.set)

# 创建保存日志按钮

save_btn = tk.Button(window, text="保存日志", command=save_log)

save_btn.pack(padx=10, pady=10)

def update_video():

# 读取视频帧

global frame

ret, frame = cap.read()

if ret:

# 将OpenCV图像转换为Tkinter图像

image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

image = Image.fromarray(image)

image = ImageTk.PhotoImage(image)

# 更新标签上的图像

video_label.config(image=image)

video_label.image = image

if is_recording:

# 写入视频帧到视频文件

width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))

height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

fps = 20.0

fourcc = cv2.VideoWriter_fourcc(*'XVID')

out = cv2.VideoWriter(video_name, fourcc, fps, (width, height))

out.write(frame)

# 循环调用更新视频帧

video_label.after(10, update_video)

# 启动更新视频帧的函数

update_video()

# 运行GUI窗口

window.mainloop()

# 运行摄像头监控系统

show_video()