这段代码使用MindSpore模型进行实时人脸识别,识别结果会显示在摄像头画面上。该代码片段展示了如何使用MindSpore模型进行实时人脸识别,并使用OpenCV库进行图像处理和显示。

cap = cv2.VideoCapture(0)
stop = False
while not stop:
    success, img = cap.read()
    subjects = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '10', '11', '12', '13', '14', '15', '16', '17',
                '18', '19', '20', '21', '22', '23', '24', '25', '26', '27', '28', '29', '30', '31', '32', '33']
    # 生成图像的副本,这样就能保留原始图像
    img1 = img.copy()
    # 检测人脸
    # 将测试图像转换为灰度图像,因为opencv人脸检测器需要灰度图像
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    # 检测多尺度图像,返回值是一张脸部区域信息的列表(x,y,宽,高)
    rect = face_cascade.detectMultiScale(img, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30),
                                         flags=cv2.CASCADE_SCALE_IMAGE)
    # 如果未检测到面部
    if len(rect) == 0:
        txt = 'no face!'
        cv2.putText(img1, txt, (10, 20), cv2.FONT_HERSHEY_COMPLEX, 1, (128, 128, 0), 2)
    if not rect is None:
        for (x, y, w, h) in rect:
            face = gray[y:y + w, x:x + h].astype(np.float32)  # 数值转换
            face = cv2.resize(face, (100, 100))
            face = Tensor(face)
            cv2.rectangle(img1, (x, y), (x + w, y + h), (0, 255, 0), 2)  # 画出矩形框
            output =network(face)
            predicted_class = np.argmax(output.asnumpy(),axis=1)
            #if min_d < 200000000000:
            cv2.putText(img1, predicted_class, (x, y), cv2.FONT_HERSHEY_COMPLEX, 1, (128, 128, 0), 2)
            #else:
             #   label = 'unknown'
              #  cv2.putText(img1, label, (x, y), cv2.FONT_HERSHEY_COMPLEX, 1, (128, 128, 0), 2)
    cv2.imshow('img', img1)
    if (cv2.waitKey(1) & 0xFF == ord('q')):  # 按下q程序结束
            stop = True
            cv2.destroyAllWindows()  # 释放窗口

错误解决:

ValueError: mindspore\core\utils\check_convert_utils.cc:367 CheckInteger] The primitive[Conv2D]'s x shape size must be equal to 4, but got 2.

这个错误提示表示输入到Conv2D算子的输入数据形状不正确,需要将输入数据的形状转换为4维,例如(batch_size, channel, height, width)。

  1. 检查模型和输入数据形状: 检查代码中使用的MindSpore模型和输入数据的形状是否匹配。如果不匹配,需要进行相应的形状转换。
  2. 加载模型: 确保MindSpore模型已经被加载,并且可以正常运行。
  3. 预测结果显示: 可以尝试使用模型的predict函数对输入数据进行预测,然后将预测结果显示在图像上。

参考资源:

通过调整代码,确保模型和输入数据形状匹配,并正确加载和调用模型,就可以实现实时人脸识别功能。

使用MindSpore模型实时人脸识别系统

原文地址: https://www.cveoy.top/t/topic/jqxQ 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录