mport torch
from PIL import Image
import numpy as np
import dlib
import cv2
#老规矩 上面是调用需要的库
from model.CNN import CNNnet
#这里是调用自定义的CNN模型
label_name = [
"Engage-confused",
"Engage-engaged",
"Engage-frustrated",
"Not engage-bored",
"Not engage-drowsy",
"Not engage-lookingaway"
]
#这段代码定义了一个包含不同情绪标签的列表。具体来说,列表中包含了以下情绪标签:
#- Engage-confused(专注-困惑)
#- Engage-engaged(专注-投入)
#- Engage-frustrated(专注-沮丧)
#- Not engage-bored(不专注-无聊)
#- Not engage-drowsy(不专注-昏昏欲睡)
#- Not engage-lookingaway(不专注-看向远处)
#这些标签用于表示人脸情绪识别模型预测出的不同情绪类别,便于后续对识别结果进行解释和展示。
detector = dlib.get_frontal_face_detector()
if __name__ == "__main__":
# ------------------------------------