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

OpenCvSharpを使用して、Webカメラでキャプチャーしながら顔検出をしてみる。

基本的には静止画の時と同じで、その部分のコードを動画のキャプチャーに埋め込んだだけ。

1. [void][System.Reflection.Assembly]::LoadFrom((Join-Path $pwd OpenCvSharp.dll))
3. #顔を検出するためにHaar分類器のカスケードを用いる
5. $color = [OpenCvSharp.CvColor]::Red
6. $scale = 1.04;
7. $scaleFactor = 1.139;
8. $minNeighbors = 2;
10. $camera = [OpenCvSharp.CvCapture]::FromCamera(0)
11. $window = New-Object OpenCvSharp.CvWindow("SampleCapture")
13. $srcWidth = $camera.FrameWidth
14. $srcHeight = $camera.FrameHeight
15. $srcSize = New-Object OpenCvSharp.CvSize([int]$srcWidth, [int]$srcHeight)
16. $srcIplImage = New-Object OpenCvSharp.IplImage( $srcSize, [OpenCvSharp.BitDepth]::U8, 1)
18. $dstWidth = $camera.FrameWidth / $scale
19. $dstHeight = $camera.FrameHeight / $scale
20. $dstSize = New-Object OpenCvSharp.CvSize([int]$dstWidth, [int]$dstHeight)
21. $dstIplImage = New-Object OpenCvSharp.IplImage( $dstSize, [OpenCvSharp.BitDepth]::U8, 1)
23. while ( [OpenCvSharp.CvWindow]::WaitKey(10) -lt 0 ) {
24. $iplImage = $camera.QueryFrame()
25. #顔検出用の画像の生成
26. [OpenCvSharp.Cv]::CvtColor($iplImage,$srcIplImage, [OpenCvSharp.ColorConversion]::BgraToGray)
27. [OpenCvSharp.Cv]::Resize($srcIplImage, $dstIplImage, [OpenCvSharp.Interpolation]::Linear)
28. [OpenCvSharp.Cv]::EqualizeHist($dstIplImage,$dstIplImage)
30. $cascade = [OpenCvSharp.CvHaarClassifierCascade]::FromFile((Join-Path $pwd "haarcascade_frontalface_alt.xml"))
32. $storage = New-Object OpenCvSharp.CvMemStorage
33. $storage.Clear()
35. # 顔の検出
36. $faces = [OpenCvSharp.Cv]::HaarDetectObjects($dstIplImage, $cascade, $storage, $scaleFactor, $minNeighbors, 0, (New-Object OpenCvSharp.CvSize(30, 30)));
38. Write-Host $faces.Total
39. # 検出された全ての顔位置に四角い枠を描画する
40. for ($i = 0; $i -lt $faces.Total; $i++) {
41. $r = $faces[$i].Rect
42. $pt1 = New-Object OpenCvSharp.CvPoint($r.x, $r.y)
43. $pt2 = new-object OpenCvSharp.CvPoint(($r.x + $r.Width),($r.y + $r.Height ))
44. $iplImage.Rectangle($pt1, $pt2, $color,　3, 8, 0)
45. }
46. # 丸で囲む
47. # for ($i = 0; $i -lt $faces.Total; $i++) {
48. # $r = $faces[$i].Rect
49. # $X = [OpenCvSharp.Cv]::Round(($r.X + $r.Width * 0.5) * $Scale)
50. # $Y = [OpenCvSharp.Cv]::Round(($r.Y + $r.Height * 0.5) * $Scale)
51. # $center = New-Object OpenCvSharp.CvPoint($X, $Y)
52. # $radius = [OpenCvSharp.Cv]::Round(($r.Width + $r.Height) * 0.25 * $Scale)
53. # $iplImage.Circle($center, $radius, $color, 3, [OpenCvSharp.LineType]::AntiAlias, 0)
54. # }
56. # ウィンドウに表示
57. $window.Image = $iplImage
58. }
60. $srcIplImage.Dispose()
61. $dstIplImage.Dispose()
62. $iplImage.Dispose()
63. $window.Dispose()
64. $camera.Dispose()