Как я могу сохранить модель распознавания лиц? - PullRequest
0 голосов
/ 01 мая 2020

Я работаю над проектом по распознаванию лиц с использованием open cv, теперь меня беспокоит то, как я могу сохранить свою модель распознавания Lbph после того, как она обучена, я уже попробовал функцию распознавателя.write (), но когда я go в этом Таким образом, я получаю часть признания, что имена не определены, как я могу ее решить, любая помощь будет оценена.

часть обучения: '' '

import cv2
import os
import numPy

# preparing data
dataset_path="C:/Users/acer/PycharmProjects/Final_Project/Database"
if len(os.listdir(dataset_path))!=0:
    (images, lables, names, id) = ([], [], {}, 0)
    for (_, dirs, _) in os.walk(dataset_path):
        for subdir in dirs:
            names[id] = subdir
            subjectpath = os.path.join(dataset_path, subdir)
            for filename in os.listdir(subjectpath):
                path = subjectpath + '/' + filename
                lable = id
                images.append(cv2.imread(path, 0))
                lables.append(int(lable))
            id += 1

    # Create a Numpy array from the two lists above
    (images, lables) = [numpy.array(li) for li in [images, lables]]

    # initializing and training Local Binary Patterns Histogram model
    Face_Recognizer = cv2.face.LBPHFaceRecognizer_create()
    print("Training started....")
    Face_Recognizer.train(images, lables)
    Face_Recognizer.write("C:/Users/acer/PycharmProjects/Final_Project/Face_Recognizer.yml")
    print("Training completed")
else:
    print("oops, Dataset is not present! First create the dataset")

' ' '

часть распознавания:

' ''

Face_Recognizer = cv2.face.LBPHFaceRecognizer_create()
Face_Recognizer.read("C:/Users/acer/PycharmProjects/Final_Project/Face_Recognizer.yml")
(width, height) = (300, 300)
webcam = cv2.VideoCapture(0)
fourcc=cv2.VideoWriter_fourcc(*'XVID') # this function is used to store the fourcc code
video_writer_obj=cv2.VideoWriter("E:/Pycharm/data/saved_video.avi",fourcc,20,(640,480))
while True:
    (_, im) = webcam.read()
    gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
    faces = face_cascade.detectMultiScale(gray, 1.3, 5)
    for (x, y, w, h) in faces:
        cv2.rectangle(im, (x, y), (x + w, y + h), (255, 0, 0), 2)
        face = gray[y:y + h, x:x + w]
        face_resize = cv2.resize(face, (width, height))
        # Try to recognize the face
        prediction = Face_Recognizer.predict(face_resize)
        cv2.rectangle(im, (x, y), (x + w, y + h), (0, 255, 0), 3)

        if prediction[1] < 500:

            cv2.putText(im, '% s - %.0f' %
                        (names[prediction[0]], prediction[1]), (x - 10, y - 10),
                        cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 0))
        else:
            cv2.putText(im, 'not recognized',
                        (x - 10, y - 10), cv2.FONT_HERSHEY_PLAIN, 1, (0, 255, 0))
    video_writer_obj.write(im)
    cv2.imshow('OpenCV', im)

    key = cv2.waitKey(10)
    if key == 32:
        break
'''

внесите необходимые изменения в мой код. спасибо

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...