我们进一步完善了这个app01.py,我们优化了界面使其更人性化,下面介绍中包含了原有的功能及新增的功能:
创建和管理文件夹:
create_folder 函数用于创建保存照片和视频的文件夹。
get_next_file_number 函数用于获取文件夹中下一个可用的文件编号。
图像处理:
pil_to_cv 函数用于将PIL图像转换为OpenCV图像。
remove_buttons 函数用于去除界面上的按钮。
add_timestamp 函数用于在图像上添加时间戳。
apply_filter 函数用于应用图像滤镜(灰度和模糊)。
鼠标回调:
mouse_callback 函数用于处理鼠标事件,包括按钮点击、区域选择等。
文件操作:
save_photo 函数用于保存照片。
start_recording 和 stop_recording 函数用于开始和停止录像。
switch_camera 函数用于切换摄像头。
open_photo_folder 函数用于打开照片文件夹。
绘制功能:
draw_buttons 函数用于在图像上绘制按钮。
draw_hints 函数用于在图像上绘制提示信息。
主函数:
main 函数是程序的入口点,负责初始化摄像头、设置鼠标回调、读取和处理图像、显示图像等。
import cv2
import os
import numpy as np
from PIL import Image, ImageDraw, ImageFont
import datetime
import webbrowser
import logging
import threading# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')# 检查并创建保存照片和视频的文件夹
def create_folder(folder_name):if not os.path.exists(folder_name):os.makedirs(folder_name)return folder_name# 获取文件夹中的最大编号
def get_next_file_number(folder_name, file_extension):files = os.listdir(folder_name)files = [f for f in files if f.endswith(file_extension)]if files:numbers = [int(f.split('.')[0]) for f in files]return max(numbers) + 1else:return 1# 将PIL图像转换为OpenCV图像
def pil_to_cv(image):return cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)# 鼠标回调函数
def mouse_callback(event, x, y, flags, param):global next_photo_number, next_video_number, running, recording, out, frame, scale_factor, cam_index, roi, button_hintsif event == cv2.EVENT_LBUTTONDOWN:if 10 <= x <= 100 and 10 <= y <= 50: # 关闭按钮区域running = Falseelif 10 <= x <= 100 and 70 <= y <= 110: # 拍照按钮区域threading.Thread(target=save_photo, args=(frame, next_photo_number)).start()next_photo_number += 1elif 10 <= x <= 100 and 130 <= y <= 170: # 开始/停止录像按钮区域if not recording:start_recording()else:stop_recording()elif 10 <= x <= 100 and 190 <= y <= 230: # 放大按钮区域scale_factor = min(3.0, scale_factor * 2)elif 10 <= x <= 100 and 250 <= y <= 290: # 缩小按钮区域scale_factor = max(1.0, scale_factor / 2)elif 10 <= x <= 100 and 310 <= y <= 350: # 切换摄像头按钮区域switch_camera()elif 10 <= x <= 100 and 370 <= y <= 410: # 查看照片按钮区域open_photo_folder()elif event == cv2.EVENT_RBUTTONDOWN:roi[0], roi[1] = x, yelif event == cv2.EVENT_RBUTTONUP:roi[2], roi[3] = x - roi[