opencv实现人脸识别（五） 运用tkinter进行GUI绘制 整合人脸识别模块
生活随笔
收集整理的這篇文章主要介紹了
opencv实现人脸识别（五） 运用tkinter进行GUI绘制 整合人脸识别模块
小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
因?yàn)橹皩W(xué)習(xí)過(guò)tkinter庫(kù)，所以在學(xué)習(xí)了人臉識(shí)別模塊的編寫(xiě)后，
打算繪制一個(gè)簡(jiǎn)單的GUI來(lái)應(yīng)用人臉識(shí)別功能。
主界面如下所示：
簽到打開(kāi)在點(diǎn)開(kāi)后直接進(jìn)行人臉識(shí)別，如果成功則自動(dòng)關(guān)閉視頻窗口。
錄入新的人臉界面：
輸入姓名后打開(kāi)攝像頭，開(kāi)始拍攝鏡頭前的人的照片，然后生成訓(xùn)練文件。
并且可以查詢歷史簽到記錄。
這里需要另外添加的模塊是關(guān)于數(shù)據(jù)庫(kù)的，這里選用了sqlite，所需的功能也很簡(jiǎn)單，
只需要兩個(gè)表，一個(gè)用來(lái)存放用戶姓名，一個(gè)用來(lái)存放簽到記錄。
這里是數(shù)據(jù)庫(kù)模塊的代碼
db.py:
import sqlite3
from datetime import *
class record:
def __init__(self):
# 創(chuàng)建或打開(kāi)一個(gè)數(shù)據(jù)庫(kù)
# check_same_thread 屬性用來(lái)規(guī)避多線程操作數(shù)據(jù)庫(kù)的問(wèn)題
self.conn = sqlite3.connect("recordinfo.db", check_same_thread=False)
# 創(chuàng)建游標(biāo)
self.cursor = self.conn.cursor()
# 建表
self.conn.execute('create table if not exists record_table('
'id integer primary key autoincrement,'
'name varchar(30) ,'
'record_time timestamp)')
self.conn.execute('create table if not exists name_table('
'id integer primary key autoincrement,'
'name varchar(30))')
# 插入數(shù)據(jù)
def insert_record(self, name):
self.conn.execute('insert into record_table values (null, ?, ?)', (name, datetime.now()))
self.conn.commit()
def insert_name(self, name):
self.conn.execute('insert into name_table values (null, ?)', [name])
self.conn.commit()
# 搜索用戶名
def query_name(self):
self.cursor.execute("select name from name_table")
results = self.cursor.fetchall()
name_list = []
for i in results:
i = list(i)
name_list += i
return name_list
def query_record(self):
self.cursor.execute('select * from record_table')
results = self.cursor.fetchall()
return results
def close(self):
self.cursor.close()
self.conn.close()
然后是把之前的拍照模塊，訓(xùn)練模塊整合在一起，綁定到錄入新的人臉的 確定按鈕上
add_face.py:
import os
import cv2
from PIL import Image, ImageTk
import numpy as np
import db
def makeDir():
if not os.path.exists("face_trainer"):
os.mkdir("face_trainer")
if not os.path.exists("FaceData"):
os.mkdir("FaceData")
def getFace(name):
cap = cv2.VideoCapture(0)
face_detector = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
count = 0
while True:
sucess, img = cap.read() # 從攝像頭讀取圖片
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
faces = face_detector.detectMultiScale(gray, 1.3, 5)
for (x, y, w, h) in faces:
cv2.rectangle(img, (x, y), (x+w, y+h), (255, 0, 0))
count += 1
cv2.imwrite("FaceData/User." + name.get() + '.' + str(count) + '.jpg', gray[y: y + h, x: x + w])
cv2.imshow('image', img)
# 保持畫(huà)面的持續(xù)。
k = cv2.waitKey(1)
if k == 27: # 通過(guò)esc鍵退出攝像
break
elif count >= 20: # 得到1000個(gè)樣本后退出攝像
break
cap.release()
cv2.destroyAllWindows()
def getImagesAndLabels(path,detector, usernames):
imagePaths = [os.path.join(path, f) for f in os.listdir(path)]
faceSamples = []
ids = []
for imagePath in imagePaths:
PIL_img = Image.open(imagePath).convert('L')
img_numpy = np.array(PIL_img, 'uint8')
username = os.path.split(imagePath)[-1].split(".")[1]
id = 1
for x in usernames:
if username == x:
break
else:
id += 1
faces = detector.detectMultiScale(img_numpy)
for (x, y, w, h) in faces:
faceSamples.append(img_numpy[y:y + h, x: x + w])
ids.append(id)
return faceSamples, ids
def trainFace(names):
# 人臉數(shù)據(jù)路徑
path = 'FaceData'
recognizer = cv2.face.LBPHFaceRecognizer_create()
detector = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")
faces, ids = getImagesAndLabels(path, detector, names)
recognizer.train(faces, np.array(ids))
recognizer.write(r'face_trainer rainer.yml')
def add_face(name, names):
makeDir()
getFace(name)
trainFace(names)
user = db.record()
user.insert_name(name.get())
然后把識(shí)別人臉的模塊綁定給主界面的簽到按鈕
detect.py:
import cv2
import time
import db
def check( names):
cam = cv2.VideoCapture(0)
recognizer = cv2.face.LBPHFaceRecognizer_create()
recognizer.read('face_trainer/trainer.yml')
cascadePath = "haarcascade_frontalface_default.xml"
faceCascade = cv2.CascadeClassifier(cascadePath)
font = cv2.FONT_HERSHEY_SIMPLEX
minW = 0.1 * cam.get(3)
minH = 0.1 * cam.get(4)
while True:
ret, img = cam.read()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
faces = faceCascade.detectMultiScale(
gray,
scaleFactor=1.2,
minNeighbors=5,
minSize=(int(minW), int(minH))
)
for (x, y, w, h) in faces:
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)
idnum, confidence = recognizer.predict(gray[y:y + h, x:x + w])
if confidence < 100:
username = names[idnum-1]
confidence = "{0}%".format(round(100 - confidence))
cv2.putText(img, str(username), (x + 5, y - 5), font, 1, (0, 0, 255), 1)
cv2.putText(img, str(confidence), (x + 5, y + h - 5), font, 1, (0, 0, 0), 1)
cv2.imshow('camera', img)
time.sleep(2)
db.record().insert_record(username) # 簽到信息插入數(shù)據(jù)庫(kù)
cam.release()
cv2.destroyAllWindows()
return
else:
idnum = "unknown"
confidence = "{0}%".format(round(100 - confidence))
cv2.putText(img, str(idnum), (x + 5, y - 5), font, 1, (0, 0, 255), 1)
cv2.putText(img, str(confidence), (x + 5, y + h - 5), font, 1, (0, 0, 0), 1)
cv2.imshow('camera', img)
k = cv2.waitKey(10)
if k == 27:
break
cam.release()
cv2.destroyAllWindows()
把上述模塊整合到GUI界面中
from tkinter import *
from tkinter import ttk
import add_face
import db
import detect
class APP:
def __init__(self):
self.root = Tk()
self.root.title('FACE')
self.root.geometry('%dx%d' % (400, 300))
# 數(shù)據(jù)庫(kù)實(shí)例創(chuàng)建
self.mydb = db.record()
self.createFirstPage()
# 新錄入的人的姓名
self.name = StringVar()
mainloop()
def createFirstPage(self):
self.page1 = Frame(self.root)
self.page1.grid()
Label(self.page1, height=4, text='人臉識(shí)別系統(tǒng)', font=('粗體', 20)).grid(columnspan=2)
#self.usernames 是 用戶名字組成的列表
self.usernames = []
self.usernames = self.mydb.query_name()
self.button11 = Button(self.page1, width=18, height=2, text="簽到打卡", bg='red', font=("宋", 12),
relief='raise', command = lambda :detect.check( self.usernames))
self.button11.grid(row=1, column=0, padx=25, pady=10)
self.button12 = Button(self.page1, width=18, height=2, text="錄入新的人臉", bg='green', font=("宋", 12),
relief='raise', command = self.createSecondPage)
self.button12.grid(row=1, column=1, padx=25, pady=10)
self.button13 = Button(self.page1, width=18, height=2, text="查詢簽到信息", bg='white', font=("宋", 12),
relief='raise',command = self.checkDataView)
self.button13.grid( row=2, column=0,padx=25, pady=10)
self.button14 = Button(self.page1, width=18, height=2, text="退出系統(tǒng)", bg='gray', font=("宋", 12),
relief='raise',command = self.quitMain)
self.button14.grid(row=2, column=1,padx=25, pady=10)
def createSecondPage(self):
# self.camera = cv2.VideoCapture(0)
self.page1.grid_forget()
self.page2 = Frame(self.root)
self.page2.pack()
Label(self.page2, text='歡迎使用人臉識(shí)別系統(tǒng)', font=('粗體', 20)).pack()
# 輸入姓名的文本框
font1 = ('宋',18)
# self.name = StringVar()
self.text = Entry(self.page2, textvariable=self.name, width=20, font=font1).pack(side=LEFT)
self.name.set('請(qǐng)輸入姓名')
# 確認(rèn)名字的按鈕
self.button21 = Button(self.page2, text='確認(rèn)', bg='white', font=("宋", 12),
relief='raise', command=lambda :add_face.add_face( self.name, self.usernames))
self.button21.pack(side=LEFT, padx=5, pady=10)
# 返回按鈕
self.button22 = Button(self.page2, text="返回", bg='white', font=("宋", 12),
relief='raise',command = self.backFirst)
self.button22.pack(side=LEFT, padx=10, pady=10)
def checkDataView(self):
self.page3 = Frame(self.root)
self.page1.grid_forget()
self.root.geometry('700x360')
self.page3.pack()
Label(self.page3, text='今日簽到信息', bg='white', fg='red', font=('宋體', 25)).pack(side=TOP, fill='x')
# 簽到信息查看視圖
self.checkDate = ttk.Treeview(self.page3, show='headings', column=('sid', 'name', 'check_time'))
self.checkDate.column('sid', width=100, anchor="center")
self.checkDate.column('name', width=200, anchor="center")
self.checkDate.column('check_time', width=300, anchor="center")
self.checkDate.heading('sid', text='簽到序號(hào)')
self.checkDate.heading('name', text='名字')
self.checkDate.heading('check_time', text='簽到時(shí)間')
# 插入數(shù)據(jù)
self.records = self.mydb.query_record()
for i in self.records:
self.checkDate.insert('', 'end', values=i)
# y滾動(dòng)條
yscrollbar = Scrollbar(self.page3, orient=VERTICAL, command=self.checkDate.yview)
self.checkDate.configure(yscrollcommand=yscrollbar.set)
yscrollbar.pack(side=RIGHT, fill=Y)
self.checkDate.pack(expand=1, fill=BOTH)
# 返回按鈕
Button(self.page3, width=20, height=2, text="返回", bg='gray', font=("宋", 12),
relief='raise',command =self.backMain).pack(padx = 20, pady = 20)
def backFirst(self):
self.page2.pack_forget()
self.root.geometry('400x300')
self.page1.grid()
def backMain(self):
self.root.geometry('400x300')
self.page3.pack_forget()
self.page1.grid()
def quitMain(self):
sys.exit(0)
if __name__ == '__main__':
demo = APP()
整合的代碼邏輯還可以更清楚一點(diǎn)，以后有時(shí)間改進(jìn)，
然后列出了部分學(xué)習(xí)過(guò)程中遇到的新知識(shí)和問(wèn)題，以作參考
QA：
1.Python: cv2.waitKey([delay]) → retval
waitKey()函數(shù)的功能是不斷刷新圖像，頻率時(shí)間為delay，單位為ms。
返回值為當(dāng)前鍵盤(pán)按鍵值。
用cv2.waitKey(n) == 27 break
即為 按下esc鍵時(shí)退出當(dāng)前畫(huà)面
2.os.path.join()函數(shù)：
連接兩個(gè)或更多的路徑名組件
os.listdir() 方法
用于返回指定的文件夾包含的文件或文件夾的名字的列表。這個(gè)列表以字母順序
3.Image.open(imagePath).convert('L')
L代表轉(zhuǎn)換到灰度圖
4.os.path.split 把路徑分割成 dirname 和 basename，返回一個(gè)元組
5.ret,frame = cap.read()
cap.read()按幀讀取視頻，ret,frame是獲cap.read()方法的兩個(gè)返回值。其中ret是布爾值，如果讀取幀是正確的則返回True，如果文件讀取到結(jié)尾，它的返回值就為False。frame就是每一幀的圖像，是個(gè)三維矩陣。
6.把StringVar類型變量傳入函數(shù)中時(shí)，用StringVar.get()方法可以得到字符串類型，并進(jìn)行字符串連接等操作
7.tkinter庫(kù)中的組件，在使用command參數(shù)時(shí)，在函數(shù)名前加 lambda： 就可以在函數(shù)名后加函數(shù)的參數(shù)，否則會(huì)報(bào)錯(cuò)。 例如 Button(root, command= lambda: FUNC(a,b)).pack()
8.遇到一個(gè)問(wèn)題，在進(jìn)行一次新的人臉?shù)浫牒螅蜔o(wú)法進(jìn)行簽到，必須退出系統(tǒng)重新進(jìn)入，才能進(jìn)行簽到，報(bào)的錯(cuò)誤是 error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'
經(jīng)過(guò)檢查， 我在寫(xiě)GUI的類時(shí)，一開(kāi)始寫(xiě)的是 self.camera = cv2.VideoCapture(0), 然后將self.camera 調(diào)入綁定的動(dòng)作函數(shù)，而在多個(gè)動(dòng)作函數(shù)中，都有cv2.VideoCapture(0).release的操作。問(wèn)題就在于只獲取了一次攝像頭資源，但釋放了多次。
解決方案：把GUI類中的self.camera刪去，在執(zhí)行的動(dòng)作函數(shù)中，分別定義局部變量來(lái)獲取攝像頭資源，然后函數(shù)結(jié)束時(shí)釋放資源。
9. 報(bào)錯(cuò)：sqlite3.ProgrammingError: Incorrect number of bindings supplied. The current statement uses 1, and there are 3 supplied
在寫(xiě)數(shù)據(jù)庫(kù)里的插入函數(shù)時(shí)，遇到這個(gè)問(wèn)題，例如conn.execute(‘inssert into xxtable values(?)’, a) 假設(shè)我插入a時(shí)輸?shù)氖恰產(chǎn)bc’,就會(huì)提示我 there are 3 supplied。也就是這時(shí)候傳入的是字符串的長(zhǎng)度。
而從下圖可以看出，把a(bǔ) 設(shè)為元組，那么傳的就是長(zhǎng)度為一的元組中的元素
因此改為： conn.execute(‘inssert into xxtable values(?)’, (a, ))
10.數(shù)據(jù)庫(kù)中設(shè)置了兩個(gè)表，一個(gè)記錄用戶姓名，一個(gè)記錄用戶簽到記錄，刪除表中原有記錄，但是由于兩個(gè)表的主鍵id都設(shè)置為autoincrement, 在記錄刪除后，id并沒(méi)有置零。其實(shí)在數(shù)據(jù)庫(kù)中，有一個(gè)表sqlite_sequence，這個(gè)表中會(huì)記錄自增量的值。
因此在刪除記錄后，需要將自增量也清零
UPDATE sqlite_sequence SET seq = 0 WHERE name = ‘TableName’
11.對(duì)于treeview控件的使用，特別是其中的insert函數(shù)， 這里要插入的每條記錄都是由三個(gè)元素組成的，因此需要一個(gè)元組組成的列表。而正好數(shù)據(jù)庫(kù)中游標(biāo)的fetchall()，正好符合要求。
12.數(shù)據(jù)庫(kù)中 建表時(shí)候， 要用到當(dāng)前時(shí)間，可以用 create xxtable (cur_time timestamp ) ,插入數(shù)據(jù)時(shí)則用 datetime.now來(lái)獲取當(dāng)前本地時(shí)間
13. 報(bào)錯(cuò)：IndexError：List index out of range
username = names[idnum-1]
我猜測(cè)是idnum的數(shù)值越界了。 我打印出函數(shù)中idnum的值，發(fā)現(xiàn)有35，而本來(lái)idnum的值對(duì)應(yīng)的是第幾個(gè)錄入人臉數(shù)據(jù)的人，當(dāng)是只錄入了兩個(gè)人，明顯有錯(cuò)誤。idnum的值由cv2.face.LBPHFaceRecognize_create().predict 函數(shù)返回，這個(gè)函數(shù)與訓(xùn)練文件的生成有關(guān)，于是在getImageAndLabel函數(shù)中把idnum值重新對(duì)應(yīng)。
開(kāi)始時(shí)對(duì)應(yīng)關(guān)系錯(cuò)誤，idnum與 表示第幾張圖片的count值對(duì)應(yīng)， 而應(yīng)該改為與表示地幾個(gè)人的id值對(duì)應(yīng)。
總結(jié)
以上是生活随笔為你收集整理的opencv实现人脸识别（五） 运用tkinter进行GUI绘制 整合人脸识别模块的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇： VPS教程：搭建个人云笔记服务器
- 下一篇： 图——关键路径(代码超详细注释)