人脸识别系统代码--照片识别

1.导包

Tkinter用于创建GUI,PIL用于图像处理,cv2用于OpenCV库,subprocess用于运行其他Python脚本。

复制代码
import tkinter as tk
from tkinter import filedialog
from PIL import Image, ImageTk
from PIL.Image import Resampling
import cv2
import subprocess

2.创建主窗口并设置标题和大小

复制代码
win = tk.Tk()
win.title("人脸属性")
win.geometry("800x400")

3.加载预训练的人脸检测模型和性别、年龄识别模型。

复制代码
# 加载预训练的人脸检测模型
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 加载预训练的性别和年龄识别模型
gender_net = cv2.dnn.readNetFromCaffe('deploy_gender.prototxt', 'gender_net.caffemodel')
age_net = cv2.dnn.readNetFromCaffe('deploy_age.prototxt', 'age_net.caffemodel')

4.定义性别和年龄的标签列表

复制代码
gender_list = ['男', '女']
age_list = ['(0-5)', '(5-10)', '(15-20)', '(20-25)',  '(25-30)', '(60-70)', '(45-55)', '(55-65)', '(65-80)', '(80-100)']

5.定义标量,用来储存用户选择图片的路径

复制代码
selected_image1_path = None
selected_image2_path = None

6.定义字体样式

复制代码
my_font = ('Arial', 18, 'normal')

7.创建画布

复制代码
blue = tk.Canvas(win, width=350, height=200, bg='lightblue')
blue.place(x=30, y=120)

pink = tk.Canvas(win, width=350, height=200, bg='light pink')
pink.place(x=420, y=120)

8.定义image1函数

用于处理第一个画布上的人脸识别。函数内部会打开文件选择对话框,加载图像,调整大小,转换为灰度图像进行人脸检测,然后使用深度学习模型预测性别和年龄,并在画布上显示结果。

复制代码
def image1():
    # 调用全局变量
    global selected_image1_path, photo1, img1
    # 打开文件选择对话框
    selected_image1_path = filedialog.askopenfilename()
    if selected_image1_path:  # 检查用户是否选择了文件
        # 加载图像
        image= cv2.imread(selected_image1_path)
        # 调整图像大小
        image = cv2.resize(image, (130, 170))
        # 将 OpenCV 图像转换为 PIL 图像
        pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
        # 将PIL图像对象转换为Tkinter图像对象
        photo1 = ImageTk.PhotoImage(pil_image)
        # 在画布上显示图像
        blue.create_image(80, 105, image=photo1)  # 调整坐标以适应画布

        # 转换为灰度图像
        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        # 检测人脸
        faces = face_cascade.detectMultiScale(gray, 1.1, 4)
        if len(faces) > 0:
            # 遍历检测到的人脸
            for (x, y, w, h) in faces:
                # 从原始图像中裁剪人脸区域
                face_img = image[y:y + h, x:x + w].copy()

                # 预处理人脸图像以适应神经网络输入
                blob = cv2.dnn.blobFromImage(face_img, 1,
                                             (227, 227), (78.4263377603,
                                                          87.7689143744, 114.895847746), swapRB=False)
                # 预测性别
                gender_net.setInput(blob)
                gender_preds = gender_net.forward()
                gender = gender_list[gender_preds[0].argmax()]

                # 预测年龄
                age_net.setInput(blob)
                age_preds = age_net.forward()
                age = age_list[age_preds[0].argmax()]

                tk.Label(blue, text=f'性别:{gender}', bg='lightblue', font=my_font).place(x=160, y=70)
                tk.Label(blue, text=f'年龄:{age}', bg='lightblue', font=my_font).place(x=160, y=120)
        else:
            tk.Label(blue,text='未检测到人脸', bg='lightblue', font=my_font).place(x=160, y=70)

9. 定义image2函数

image1类似,用于处理第二个画布上的人脸识别。

复制代码
def image2():
    # 调用全局变量
    global selected_image2_path, photo2
    # 打开文件选择对话框
    selected_image2_path = filedialog.askopenfilename()
    if selected_image2_path:  # 检查用户是否选择了文件
        # 加载图像
        image = cv2.imread(selected_image2_path)
        # 调整图像大小
        image = cv2.resize(image, (130, 170))
        # 将 OpenCV 图像转换为 PIL 图像
        pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
        # 将PIL图像对象转换为Tkinter图像对象
        photo2 = ImageTk.PhotoImage(pil_image)
        # 在画布上显示图像
        pink.create_image(80, 105, image=photo2)  # 调整坐标以适应画布

        # 转换为灰度图像
        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        # 检测人脸
        faces = face_cascade.detectMultiScale(gray, 1.1, 4)
        if len(faces) > 0:
            # 遍历检测到的人脸
            for (x, y, w, h) in faces:
                # 从原始图像中裁剪人脸区域
                face_img = image[y:y + h, x:x + w].copy()

                # 预处理人脸图像以适应神经网络输入
                blob = cv2.dnn.blobFromImage(face_img, 1,
                                             (227, 227), (78.4263377603,
                                                          87.7689143744, 114.895847746), swapRB=False)
                # 预测性别
                gender_net.setInput(blob)
                gender_preds = gender_net.forward()
                gender = gender_list[gender_preds[0].argmax()]

                # 预测年龄
                age_net.setInput(blob)
                age_preds = age_net.forward()
                age = age_list[age_preds[0].argmax()]

                tk.Label(pink, text=f'性别:{gender}', bg='lightpink', font=my_font).place(x=160, y=70)
                tk.Label(pink, text=f'年龄:{age}', bg='lightpink', font=my_font).place(x=160, y=120)
        else:
            tk.Label(pink, text='未检测到人脸', bg='lightpink', font=my_font).place(x=160, y=70)

10.定义close函数

用于关闭当前窗口并打开另一个Python脚本

复制代码
def close():
    subprocess.Popen(["python", "属性判断.py"])
    win.destroy()

11.加载按钮图像

并创建三个按钮,分别用于选择图像1、选择图像2和关闭窗口

复制代码
image = Image.open("img/F11.gif")  # 加载一张图片
photo7 = ImageTk.PhotoImage(image)
tk.Button(win, image=photo7, command=image1).place(x=100, y=30)

image = Image.open("img/F12.gif")  # 加载一张图片
photo8 = ImageTk.PhotoImage(image)
tk.Button(win, image=photo8, command=image2).place(x=490, y=30)

image = Image.open("img/T.gif")  # 加载一张图片
photo9 = ImageTk.PhotoImage(image)
tk.Button(win, image=photo9,command=close).place(x=300, y=350)

12.退出系统

复制代码
win.mainloop()
相关推荐
艾莉丝努力练剑40 分钟前
【AI大模型接入SDK】Provider分析与实现
c++·人工智能·学习·面试·大模型·llm
磁场转动100万匹41 分钟前
PyTorch 手写数字识别实战:从数据加载到模型训练(零基础详解版)
人工智能·pytorch·python
TaoMetrix7 小时前
TaoMetrix:AI 硬件创业公司如何做好原型制造
人工智能·制造
青 春 记 忆7 小时前
零基础入门python30:Flask个人账本从空目录运行与阶段验收
python·flask·后端开发
智圣新创017 小时前
面向多级组织协同场景 高校第二课堂一站式管理中枢落地全场景实操答疑
大数据·人工智能
cd_949217217 小时前
AI纹理和Substance Painter手绘纹理哪个更适合游戏资产制作?
人工智能·游戏·substance painter
sali-tec7 小时前
C# 基于OpenCv的视觉工作流-章106-差值追踪
图像处理·人工智能·opencv·算法·计算机视觉
Shockang7 小时前
用 AI 打造高品质 Web 应用
人工智能
l1258657 小时前
# LangGraph Memory机制深度解析:短期记忆与长期记忆的工程实践
前端·人工智能·python·langchain·bootstrap
手写码匠7 小时前
Dify 多 Agent 工具权限与安全沙箱实战:让智能体“有能力,但不越权“
人工智能·深度学习·算法·aigc