Я использую openCV для обучения наборов данных для распознавания лиц. У меня есть 3 картинки для моих тренировочных наборов. Все в коде работает нормально, пока я не выполню код для обучения.
cv2.error: OpenCV(3.4.1) C:\projects\opencv-python\opencv_contrib\modules\face\src\lbph_faces.cpp:359: error: (-210) Empty training data was given. You'll need more than one sample to learn a model. in function cv::face::LBPH::train
Следует отметить, что путь, упомянутый в ошибке, даже не существует на моей машине.
У меня есть два файла Python, один из которых создан, чтобы просто видеть лицо человека на веб-камере и сохранять его в каком-то месте. Ниже приведен код: -
import cv2
import numpy as np
face_cascade = cv2.CascadeClassifier('data\haarcascade_frontalface_alt2.xml')
cap = cv2.VideoCapture(0)
while True:
ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = face_cascade.detectMultiScale(gray, scaleFactor=1.5, minNeighbors=5)
for (x, y, w, h) in faces:
print(x, y, w, h)
roi_gray = gray[y:y+h, x:x+w]
roi_color = frame[y:y+h, x:x+w]
img_item = "amitesh2.png"
color = (255, 0, 0)
stroke = 2
width = x+w
height = y+h
rect = cv2.rectangle(frame, (x, y),(width, height), color, stroke)
cv2.imwrite(img_item, roi_gray)
cv2.imshow('test', frame)
if cv2.waitKey(20) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
Приведенный выше код работает в соответствии с требованием. Проблема в следующем коде, который тренирует и распознает лицо.
import os
import numpy as np
import cv2
import pickle
from PIL import Image
face_cascade = cv2.CascadeClassifier('data\haarcascade_frontalface_alt2.xml')
print(face_cascade)
recog = cv2.face.LBPHFaceRecognizer_create()
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
img_dir = os.path.join(BASE_DIR, "images")
curr_id = 0
label_ids = {}
x_train = []
y_label = []
for root, dirs, files in os.walk(img_dir):
for fil in files:
if fil.endswith("png") or fil.endswith("jpg"):
path = os.path.join(root, fil)
label = os.path.basename(os.path.dirname(root)).lower()
#print(label, path)
if not label in label_ids:
label_ids[label] = curr_id
curr_id += 1
id_ = label_ids[label]
pil_image = Image.open(path).convert("L")
img_array = np.array(pil_image, "uint8")
#print(img_array)
faces = face_cascade.detectMultiScale(img_array, scaleFactor=1.5, minNeighbors=5)
for (x, y, w, h) in faces:
roi = img_array[y:y+h, x:x+w]
x_train.append(roi)
y_label.append(id_)
with open("labels.pkl", 'wb') as f:
pickle.dump(label_ids, f)
recog.train(x_train, np.array(y_label))
recog.save("trainer.yml")
Я искал весь StackOverflow. Я мог видеть сообщения, но не мог найти что-нибудь по моей проблеме. Пожалуйста, дайте мне несколько советов.