我正在编写一个面部识别程序,但我一直收到这个错误,我很困惑我在网上看不到其他示例,人们在转换为 UMat 时包含范围
Traceback (most recent call last):
File "test.py", line 48, in <module>
test_photos()
File "test.py", line 40, in test_photos
face, rect = detect_face(test_photo)
File "test.py", line 15, in detect_face
imgUMat = cv2.UMat(img)
TypeError: UMat() missing required argument 'ranges' (pos 2)
我的代码是
def detect_face(img):
imgUMat = cv2.UMat(img)
gray = cv2.cvtColor(imgUMat, cv2.COLOR_BGR2GRAY)
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
faces = face_cascade.detectMultiScale(gray, scaleFactor=1.2, minNeighbors=5)
if (len(faces)==0):
return None, None
(x, y, w, h) = faces[0]
gray = gray.get()
return gray[y:y+h,x:x+w], faces[0]
def prepare_training_data():
faces = []
labels = []
for img in photo_name_list: #a collection of file locations as strings
image = cv2.imread(img)
face, rect = detect_face(image)
if face is not None:
faces.append(face)
labels.append(me)
return faces, labels
def test_photos():
face_recognizer = cv2.face.LBPHFaceRecognizer_create()
faces, labels = prepare_training_data()
face_recognizer.train(np.array(faces), np.array(labels))
face, rect = detect_face(test_photo)
label = face_recognizer.predict(face)
if label == me:
print("it's me")
else:
print("it's not me")
test_photos()
如果我不使用 UMat() 则会收到此错误:
Traceback (most recent call last):
File "test.py", line 48, in <module>
test_photos()
File "test.py", line 40, in test_photos
face, rect = detect_face(test_photo)
File "test.py", line 16, in detect_face
gray = cv2.cvtColor(imgUMat, cv2.COLOR_BGR2GRAY)
TypeError: Expected cv::UMat for argument 'src'
我正在使用 OpenCV 4.0.0,老实说,我只是很困惑,因为据我所知,没有人必须使用 UMat 才能使用 cvtColor(),更不用说在 UMat() 中使用范围了。任何帮助将不胜感激。