Python用dlib和openCV获取人脸的特征值和人脸中文姓名并存入mysql中,在进行人脸匹配时调取mysql中的数据,并在视屏上显示人名
时间: 2024-01-21 18:19:37 浏览: 92
要实现这个功能,你需要先安装dlib和OpenCV库,以及MySQL数据库。然后按照以下步骤进行操作:
1. 使用dlib库中的人脸检测器和特征提取器获取人脸的特征值,并将其存入MySQL数据库中。
```python
import dlib
import cv2
import mysql.connector
# 加载dlib的人脸检测器和特征提取器
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor('shape_predictor_68_face_landmarks.dat')
# 连接MySQL数据库
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
# 获取游标
mycursor = mydb.cursor()
# 读取图片并进行人脸检测和特征提取
img = cv2.imread('test.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
faces = detector(gray)
for face in faces:
landmarks = predictor(gray, face)
features = [landmarks.part(i).x for i in range(68)] + [landmarks.part(i).y for i in range(68)]
# 将人脸特征值和姓名存入MySQL数据库
sql = "INSERT INTO faces (name, features) VALUES (%s, %s)"
val = ("张三", str(features))
mycursor.execute(sql, val)
mydb.commit()
```
2. 在进行人脸匹配时,从MySQL数据库中获取存储的特征值,并使用dlib的距离函数进行匹配,找到最匹配的人脸,并显示其姓名。
```python
# 连接MySQL数据库
mydb = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
# 获取游标
mycursor = mydb.cursor()
# 读取视频流
cap = cv2.VideoCapture(0)
while True:
ret, frame = cap.read()
if not ret:
break
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = detector(gray)
for face in faces:
landmarks = predictor(gray, face)
features = [landmarks.part(i).x for i in range(68)] + [landmarks.part(i).y for i in range(68)]
# 从MySQL数据库中获取所有人脸的特征值
mycursor.execute("SELECT * FROM faces")
results = mycursor.fetchall()
min_distance = float('inf')
best_match = None
# 使用dlib的距离函数进行匹配
for result in results:
name = result[0]
features_db = eval(result[1])
distance = dlib.distance(features, features_db)
if distance < min_distance:
min_distance = distance
best_match = name
# 在视频上显示人名
cv2.rectangle(frame, (face.left(), face.top()), (face.right(), face.bottom()), (0, 255, 0), 2)
cv2.putText(frame, best_match, (face.left(), face.top() - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
cv2.imshow('frame', frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
```
以上就是实现人脸匹配并显示姓名的代码。注意,这里只是一个简单的示例,实际应用中还需要考虑一些问题,比如人脸数据库的管理、人脸匹配的效率等。
阅读全文