学校里有门图像处理的课程最终需要提交一个图像处理系统,
正好之前对于opencv有些了解,就简单的写一个人脸识别小程序吧
效果图如下
笔者IDE使用Pycharm,GUI编程直接使用内置的tkinter
环境:
python3.6
opencv4.1
首先导入需要使用的各个库
代码语言:javascript复制#-*- coding: utf-8 -*-
import sys
import importlib
import cv2
import tkinter as tk
import tkinter.messagebox
from tkinter import filedialog
之后我们需要做一个路径选择函数,因为毕竟不能每次识别而去手动改代码内的地址
而这个函数我们稍后会绑定至一个button方便使用
代码语言:javascript复制def selectPath():
global path_
path_ = filedialog.askopenfilename()
path.set(path_)
path = tk.StringVar()
最关键的人脸识别函数
其中所使用到的训练参数数据下载地址:https://github.com/opencv/opencv/tree/master/data/haarcascades
并且xml文件需要放到项目目录下
代码语言:javascript复制def imgface():
try:
# github获取训练好的人脸的参数数据
face_cascade = cv2.CascadeClassifier(r'./haarcascade_frontalface_default.xml')
# 读取图片
image = cv2.imread(path_)
#转化为灰度图
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# 探测图片中的人脸
faces = face_cascade.detectMultiScale(
gray,
scaleFactor=1.15,
minNeighbors=5,
minSize=(5, 5),
flags=cv2.IMREAD_GRAYSCALE)
if(len(faces)==0):
tkinter.messagebox.showerror('错误', '未识别出人脸,请选择更为清晰的图片')
print("find {0} faces!".format(len(faces)))
# faces中的四个量分别为左上角的横坐标、纵坐标、宽度、长度
for (x, y, w, h) in faces:
cv2.rectangle(image,(x,y),(x w,y w),(255,245,0),1)
if (len(faces) > 0):
cv2.imshow("find {0} faces!".format(len(faces)), image)
cv2.waitKey(0)
except:
tkinter.messagebox.showerror('错误', '请选择正确的图片文件!')
之后进行GUI编程:
代码语言:javascript复制importlib.reload(sys)
window = tk.Tk()
window.title('人脸识别小程序')
fm1 = tk.Frame(window)
fm2 = tk.Frame(window)
fm3 = tk.Frame(window)
def selectPath():
global path_
path_ = filedialog.askopenfilename()
path.set(path_)
path = tk.StringVar()
Ltop=tk.Label(fm1,text="请选择图片路径")
B1=tk.Button(fm2, text = "路径选择", command = selectPath)
E1=tk.Entry(fm2, textvariable = path,bd=5)
B2=tk.Button(fm2, text = "确定", command =imgface)
Lbot=tk.Label(fm3,text="学号:1622107031xx 姓名:istw")
Ltop.pack(side = tk.TOP)
B1.pack(side=tk.LEFT)
E1.pack(side = tk.LEFT)
B2.pack(side=tk.LEFT)
Lbot.pack(side = tk.BOTTOM)
fm1.pack(side=tk.TOP)
fm2.pack(side=tk.TOP)
fm3.pack(side=tk.TOP)
sw = window.winfo_screenwidth()
#得到屏幕宽度
sh = window.winfo_screenheight()
#得到屏幕高度
ww = 300
wh = 100
#窗口宽高为100
x = (sw-ww) / 2
y = (sh-wh) / 3
window.geometry("%dx%d %d %d" %(ww,wh,x,y))
window.mainloop()
怎么样,一个简单的有交互界面的小程序就写出来了,只不过因为运用的人脸训练数据是官方提供的好多年前的,所以识别精度并不会太准
还有一个问题就是关于.py文件如何向别人展示的问题,因为不是每个人的电脑中都有py环境的,并且各个版本也不兼容,所以我们为了方便展示,有时候会运用些方法将其转换为exe文件,这个以后会讲到。