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

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()
相关推荐
ZPC821019 分钟前
OpenCV—颜色识别
人工智能·opencv·计算机视觉
Mr.简锋25 分钟前
vs2022搭建opencv开发环境
人工智能·opencv·计算机视觉
weixin_4432906931 分钟前
【论文阅读】InstructPix2Pix: Learning to Follow Image Editing Instructions
论文阅读·人工智能·计算机视觉
平头哥在等你41 分钟前
Python中的正则表达式教程
python·正则表达式
Best_Me071 小时前
如何在Pycharm的终端里进入自己的环境
ide·python·pycharm
ai产品老杨1 小时前
部署神经网络时计算图的优化方法
人工智能·深度学习·神经网络·安全·机器学习·开源
火山引擎边缘云1 小时前
创新实践:基于边缘智能+扣子的智能轮椅 AIoT 解决方案
人工智能·llm·边缘计算
fanxbl9571 小时前
深入探索离散 Hopfield 神经网络
人工智能·神经网络
TaoYuan__1 小时前
深度学习概览
人工智能·深度学习
云起无垠2 小时前
第74期 | GPTSecurity周报
人工智能·安全·网络安全