Vladimir Goncharov
39 linhas
1.1 KiB
PHP
39 linhas
1.1 KiB
PHP
<?php
|
|
|
|
use CV\Scalar, CV\Size;
|
|
use function CV\{imread, imwrite, rectangle};
|
|
|
|
|
|
$src = imread("images/faces.jpg");
|
|
$size = $src->size(); // 2000x500
|
|
|
|
$minSide = min($size->width, $size->height);
|
|
$divider = $minSide / 300;
|
|
\CV\resize($src, $resized, new Size($size->width / $divider, $size->height / $divider)); // 1200x300
|
|
|
|
$blob = \CV\DNN\blobFromImage($resized, 1, new Size(), new Scalar(104, 177, 123), true, false);
|
|
|
|
$net = \CV\DNN\readNetFromCaffe('models/ssd/res10_300x300_ssd_deploy.prototxt', 'models/ssd/res10_300x300_ssd_iter_140000.caffemodel');
|
|
|
|
$net->setInput($blob, "");
|
|
|
|
$r = $net->forward();
|
|
|
|
var_export($r->shape);
|
|
|
|
$scalar = new Scalar(0, 0, 255);
|
|
for ($i = 0; $i < $r->shape[2]; $i++) {
|
|
$confidence = $r->atIdx([0,0,$i,2]);
|
|
if ($confidence > 0.9) {
|
|
var_export($confidence);echo "\n";
|
|
$startX = $r->atIdx([0,0,$i,3]) * $src->cols;
|
|
$startY = $r->atIdx([0,0,$i,4]) * $src->rows;
|
|
$endX = $r->atIdx([0,0,$i,5]) * $src->cols;
|
|
$endY = $r->atIdx([0,0,$i,6]) * $src->rows;
|
|
|
|
rectangle($src, $startX, $startY, $endX, $endY, $scalar, 3);
|
|
}
|
|
}
|
|
|
|
imwrite("results/_detect_face_by_dnn_ssd.jpg", $src);
|