Задача определения лиц на фото, пожалуй, самая распространенная задача в компьютерном зрении. Набор удобных функций для этого есть в библиотеке opencv. В этой заметке я поделюсь простым примером нахождения лица на изображении.
Для опытов возьмем одну известную картину.
Также нам нужен обученный каскадный классификатор Хаара. Скачать можно тут. Библиотека предоставляет возможность обучать свои классификаторы, но это тема для отдельной большой статьи.
Приступим. Сделаем импорт, загрузим картинку и классификатор.
1 2 3 | import cv2 image = cv2.imread('mona.jpg') clf = cv2.CascadeClassifier('haarcascade_frontalface_alt.xml') |
Выполним определение лиц. Делается это одной командой.
1 2 3 4 5 6 7 | faces = clf.detectMultiScale( image, scaleFactor=1.1, minNeighbors=4, minSize=(30, 30), flags=cv2.cv.CV_HAAR_SCALE_IMAGE ) |
Вот и все. Чтобы увидеть, корректно ли алгоритм обнаружил лицо, выведем квадрат на картинке.
1 2 3 4 | for (x, y, w, h) in faces: cv2.rectangle(image, (x, y), (x+w, y+h), (0, 0, 255), 2) cv2.imshow("Face", image) cv2.waitKey(0) |
Запустив код, вы можете убедиться, что лицо найдено.
Теперь можно поэкспериментировать на своих фотографиях.
