Skip to content

Commit 93735f7

Browse files
committed
fix detectMultiScale
1 parent e214e59 commit 93735f7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

coderbot/cv/image.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,7 @@ class Image():
3939
_aruco_dict = cv2.aruco.Dictionary_get(cv2.aruco.DICT_ARUCO_ORIGINAL)
4040
_aruco_parameters = cv2.aruco.DetectorParameters_create()
4141

42-
#_face_cascade = cv2.CascadeClassifier('/usr/local/share/OpenCV/haarcascades/haarcascade_frontalface_default.xml')
43-
_face_cascade = cv2.CascadeClassifier('/usr/share/opencv/lbpcascades/lbpcascade_frontalface.xml')
42+
_face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
4443

4544
def __init__(self, array):
4645
self._data = array
@@ -85,7 +84,8 @@ def get_transform(cls, image_size_x):
8584
return tx
8685

8786
def find_faces(self):
88-
faces = self._face_cascade.detectMultiScale(self._data)
87+
gray = cv2.cvtColor(self._data, cv2.COLOR_BGR2GRAY)
88+
faces = self._face_cascade.detectMultiScale(gray)
8989
return faces
9090

9191
def filter_color(self, color):

0 commit comments

Comments
 (0)