PowerShellで静止画の顔検出をしてみる

OpenCvSharpを使用して、画像から顔を検出してみる。

顔があったらその周りを赤色の四角い枠で囲む。

顔検出には数千枚の画像ファイルから機械学習を行って、人間の顔とは？ってのを修めたデータが必要らしい…

でも、それをやるには時間がかかりすぎるので、Sampleに入っている学習済みのデータ「haarcascade_frontalface_alt.xml」を使用させてもらう。

利用させてもらっておいてどうかとは思うが、試したデータでは今一だった。

どうも余計なものまで顔と認識して検出してくる。もちろん、顔も検出してくれるからいいんだけどね。

いつものDLLと「haarcascade_frontalface_alt.xml」を同じフォルダーにコピーして、検出した元画像を「foo.jpg」として下のコードを実行する。

1. [void][System.Reflection.Assembly]::LoadFrom((Join-Path $pwd OpenCvSharp.dll))
3. #　元データ
4. $iplImage = [OpenCvSharp.Cv]::LoadImage((Join-Path $pwd "foo.jpg"))
6. $color = [OpenCvSharp.CvColor]::Red
7. $scale = 1.04;
8. $scaleFactor = 1.139;
9. $minNeighbors = 2;
11. $srcWidth = $iplImage.Width
12. $srcHeight = $iplImage.Height
13. $srcSize = New-Object OpenCvSharp.CvSize([int]$srcWidth, [int]$srcHeight)
14. $srcIplImage = New-Object OpenCvSharp.IplImage( $srcSize, [OpenCvSharp.BitDepth]::U8, 1)
16. $dstWidth = $iplImage.Width / $scale
17. $dstHeight = $iplImage.Height / $scale
18. $dstSize = New-Object OpenCvSharp.CvSize([int]$dstWidth, [int]$dstHeight)
19. $dst1IplImage = New-Object OpenCvSharp.IplImage( $dstSize, [OpenCvSharp.BitDepth]::U8, 1)
21. #　顔検出用の画像の生成
22. [OpenCvSharp.Cv]::CvtColor($iplImage,$srcIplImage, [OpenCvSharp.ColorConversion]::BgraToGray)
23. [OpenCvSharp.Cv]::Resize($srcIplImage, $dst1IplImage, [OpenCvSharp.Interpolation]::Linear)
24. [OpenCvSharp.Cv]::EqualizeHist($dst1IplImage,$dst1IplImage)
26. $cascade = [OpenCvSharp.CvHaarClassifierCascade]::FromFile((Join-Path $pwd "haarcascade_frontalface_alt.xml"))
28. $storage = New-Object OpenCvSharp.CvMemStorage
30. $storage.Clear()
32. #　顔の検出
33. $faces = [OpenCvSharp.Cv]::HaarDetectObjects($dst1IplImage, $cascade, $storage, $scaleFactor, $minNeighbors, 0, (New-Object OpenCvSharp.CvSize(30, 30)));
35. # 検出された全ての顔位置に四角い枠で囲む
36. for ($i = 0; $i -lt $faces.Total; $i++) {
37. $r = $faces[$i].Rect
38. $pt1 = New-Object OpenCvSharp.CvPoint($r.x, $r.y)
39. $pt2 = new-object OpenCvSharp.CvPoint(($r.x + $r.Width),($r.y + $r.Height ))
40. $iplImage.Rectangle($pt1, $pt2, $color,　3, 8, 0)
41. }
43. # 丸で囲いたい場合はこっち
44. #for ($i = 0; $i -lt $faces.Total; $i++) {
45. # $r = $faces[$i].Rect
46. # $X = [OpenCvSharp.Cv]::Round(($r.X + $r.Width * 0.5) * $Scale)
47. # $Y = [OpenCvSharp.Cv]::Round(($r.Y + $r.Height * 0.5) * $Scale)
48. # $center = New-Object OpenCvSharp.CvPoint($X, $Y)
49. # $radius = [OpenCvSharp.Cv]::Round(($r.Width + $r.Height) * 0.25 * $Scale)
50. # $iplImage.Circle($center, $radius, $color, 3, [OpenCvSharp.LineType]::AntiAlias, 0)
51. #}
53. #　赤枠で囲ったデータを保存する
54. $iplImage.SaveImage((Join-Path $pwd "bar.jpg"))
56. $iplImage.Dispose()
57. $srcIplImage.Dispose()
58. $dst1IplImage.Dispose()