请帮我写一段python代码实现人脸表情识别。
时间: 2024-03-03 20:25:53 浏览: 133
好的,以下是一个简单的人脸表情识别的 Python 代码:
```python
import cv2
import numpy as np
from tensorflow.keras.models import load_model
# 加载模型
model = load_model('emotion_detection_model.h5')
# 建立情感类别的标签
EMOTIONS = ['Angry', 'Disgust', 'Fear', 'Happy', 'Neutral', 'Sad', 'Surprise']
# 加载表情检测器
face_detector = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
# 读取摄像头
cap = cv2.VideoCapture(0)
while True:
# 捕获帧
ret, frame = cap.read()
# 转换为灰度图像
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 检测人脸
faces = face_detector.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))
# 对每个检测到的人脸进行情感分类
for (x, y, w, h) in faces:
# 提取人脸图像
face_roi = gray[y:y+h, x:x+w]
face_roi = cv2.resize(face_roi, (48, 48))
face_roi = face_roi.astype('float') / 255.0
face_roi = np.expand_dims(face_roi, axis=0)
face_roi = np.expand_dims(face_roi, axis=-1)
# 进行情感分类
preds = model.predict(face_roi)[0]
emotion = EMOTIONS[preds.argmax()]
# 在检测到的人脸周围绘制矩形和情感标签
cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
cv2.putText(frame, emotion, (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.45, (0, 255, 0), 2)
# 显示结果
cv2.imshow('Emotion Detection', frame)
# 按下 'q' 键退出循环
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放所有资源
cap.release()
cv2.destroyAllWindows()
```
请注意,这里使用了一个预训练的情感分类模型,需要先下载并保存在本地。此外,人脸表情识别是一个相对简单的任务,因此此代码中使用的模型并不涉及到像 ChatGPT 这样的语言模型。如果你还有其他问题或需要进一步的帮助,请告诉我。
阅读全文