Vladimir Goncharov
68 linhas
2.3 KiB
PHP
68 linhas
2.3 KiB
PHP
<?php
|
|
|
|
use CV\Face\LBPHFaceRecognizer, CV\CascadeClassifier, CV\Scalar, CV\Point;
|
|
use function CV\{imread, cvtColor, equalizeHist};
|
|
use const CV\{COLOR_BGR2GRAY};
|
|
|
|
// face by lbpcascade_frontalface
|
|
$faceClassifier = new CascadeClassifier();
|
|
$faceClassifier->load('models/lbpcascades/lbpcascade_frontalface.xml');
|
|
|
|
$faceRecognizer = LBPHFaceRecognizer::create();
|
|
|
|
$labels = ['unknown', 'me', 'angelina'];
|
|
|
|
// me
|
|
$src = imread("images/faces.jpg");
|
|
$gray = cvtColor($src, COLOR_BGR2GRAY);
|
|
$faceClassifier->detectMultiScale($gray, $faces);
|
|
//var_export($faces);
|
|
equalizeHist($gray, $gray);
|
|
$faceImages = $faceLabels = [];
|
|
foreach ($faces as $k => $face) {
|
|
$faceImages[] = $gray->getImageROI($face); // face coordinates to image
|
|
$faceLabels[] = 1; // me
|
|
//cv\imwrite("results/recognize_face_by_lbph_me$k.jpg", $gray->getImageROI($face));
|
|
}
|
|
$faceRecognizer->train($faceImages, $faceLabels);
|
|
|
|
// angelina
|
|
$src = imread("images/angelina_faces.png");
|
|
$gray = cvtColor($src, COLOR_BGR2GRAY);
|
|
$faceClassifier->detectMultiScale($gray, $faces);
|
|
//var_export($faces);
|
|
equalizeHist($gray, $gray);
|
|
$faceImages = $faceLabels = [];
|
|
foreach ($faces as $k => $face) {
|
|
$faceImages[] = $gray->getImageROI($face); // face coordinates to image
|
|
$faceLabels[] = 2; // Angelina
|
|
//cv\imwrite("results/recognize_face_by_lbph_angelina$k.jpg", $gray->getImageROI($face));
|
|
}
|
|
$faceRecognizer->update($faceImages, $faceLabels);
|
|
|
|
//$faceRecognizer->write('results/lbph_model.xml');
|
|
//$faceRecognizer->read('results/lbph_model.xml');
|
|
|
|
// test image
|
|
$src = imread("images/angelina_and_me.png");
|
|
$gray = cvtColor($src, COLOR_BGR2GRAY);
|
|
$faceClassifier->detectMultiScale($gray, $faces);
|
|
//var_export($faces);
|
|
equalizeHist($gray, $gray);
|
|
foreach ($faces as $face) {
|
|
$faceImage = $gray->getImageROI($face);
|
|
|
|
//predict
|
|
$faceLabel = $faceRecognizer->predict($faceImage, $faceConfidence);
|
|
echo "{$faceLabel}, {$faceConfidence}\n";
|
|
|
|
$scalar = new \CV\Scalar(0, 0, 255);
|
|
\CV\rectangleByRect($src, $face, $scalar, 2);
|
|
|
|
$text = $labels[$faceLabel];
|
|
\CV\rectangle($src, $face->x, $face->y, $face->x + ($faceLabel == 1 ? 50 : 130), $face->y - 30, new Scalar(255,255,255), -2);
|
|
\CV\putText($src, "$text", new Point($face->x, $face->y - 2), 0, 1.5, new Scalar(), 2);
|
|
}
|
|
|
|
cv\imwrite("results/_recognize_face_by_lbph.jpg", $src);
|