/VisualC++数字图像处理技术详解源代码/第12章 VC++结合OpenCV编程/FaceDetection.cpp

https://github.com/wyrover/book-code · C++ · 78 lines · 66 code · 9 blank · 3 comment · 3 complexity · 16115ea82c9810ce9f0881874cd9059b MD5 · raw file 

    

  1. // FaceDetection.cpp : ̨Ӧóڵ㡣
    2. 

  2. //
    3. 

  3. 

  4. #include "stdafx.h"
    5. 

  5. #include "cv.h"
    6. 

  6. #include "highgui.h"
    7. 

  7. #include <stdio.h>
    8. 

  8. 

  9. static CvHaarClassifierCascade* cascade = 0;
    10. 

  10. static CvMemStorage* storage = 0;
    11. 

  11. 

  12. void detect_and_draw( IplImage* image );
    13. 

  13. const char* cascade_name ="haarcascade_frontalface_alt.xml";                //Ҫõķ
    14. 

  14. 

  15. int _tmain(int argc, _TCHAR* argv[])
    16. 

  16. {
    17. 

  17.      cascade = (CvHaarClassifierCascade*)cvLoad( cascade_name, 0, 0, 0 );   //õķ
    18. 

  18.      if( !cascade )
    19. 

  19.      {
    20. 

  20.         fprintf( stderr, "ERROR: Could not load classifier cascade\n" );
    21. 

  21.         return -1;
    22. 

  22.      }
    23. 

  23.      storage = cvCreateMemStorage(0);                                       //̬洢ṹ洢ͼеλ
    24. 

  24.      cvNamedWindow( "result", 1 );
    25. 

  25.      const char* filename = "lena.jpg";                                     //ͼ
    26. 

  26.      IplImage* image = cvLoadImage( filename, 1 );                          //ͼ
    27. 

  27.      detect_and_draw( image );                                              //Լصͼм
    28. 

  28.      cvWaitKey(0);
    29. 

  29.      cvReleaseImage( &image );
    30. 

  30.      cvDestroyWindow("result");
    31. 

  31.      return 0;
    32. 

  32. }
    33. 

  33. 

  34. void detect_and_draw( IplImage* img )
    35. 

  35. {
    36. 

  36.     static CvScalar colors[] = 
    37. 

  37.     {
    38. 

  38.         {{0,0,255}},
    39. 

  39.         {{0,128,255}},
    40. 

  40.         {{0,255,255}},
    41. 

  41.         {{0,255,0}},
    42. 

  42.         {{255,128,0}},
    43. 

  43.         {{255,255,0}},
    44. 

  44.         {{255,0,0}},
    45. 

  45.         {{255,0,255}}
    46. 

  46.     };
    47. 

  47.  
    48. 

  48.     double scale = 1.3;
    49. 

  49.     IplImage* gray = cvCreateImage( cvSize(img->width,img->height), 8, 1 );
    50. 

  50.     IplImage* small_img = cvCreateImage( cvSize( cvRound (img->width/scale),
    51. 

  51.                          cvRound (img->height/scale)), 8, 1 );
    52. 

  52.  
    53. 

  53.     cvCvtColor( img, gray, CV_BGR2GRAY );
    54. 

  54.     cvResize( gray, small_img, CV_INTER_LINEAR );
    55. 

  55.     cvEqualizeHist( small_img, small_img );
    56. 

  56.     cvClearMemStorage( storage );
    57. 

  57.  
    58. 

  58.     if( cascade )
    59. 

  59.     {
    60. 

  60.         /*cvHaarDetectObjectsͼеĿ꣬OpenCVṩ*/
    61. 

  61.         CvSeq* faces = cvHaarDetectObjects( small_img, cascade, storage, 1.1, 2, 0 ,
    62. 

  62.                                             cvSize(30, 30) );
    63. 

  63.         for( int i = 0; i < (faces ? faces->total : 0); i++ )
    64. 

  64.         {
    65. 

  65.             CvRect* r = (CvRect*)cvGetSeqElem( faces, i );
    66. 

  66.             CvPoint center;
    67. 

  67.             int radius;
    68. 

  68.             center.x = cvRound((r->x + r->width*0.5)*scale);
    69. 

  69.             center.y = cvRound((r->y + r->height*0.5)*scale);
    70. 

  70.             radius = cvRound((r->width + r->height)*0.25*scale);
    71. 

  71.             cvCircle( img, center, radius, colors[i%8], 3, 8, 0 );
    72. 

  72.         }
    73. 

  73.     }
    74. 

  74.  
    75. 

  75.     cvShowImage( "result", img );
    76. 

  76.     cvReleaseImage( &gray );
    77. 

  77.     cvReleaseImage( &small_img );
    78. 

  78. }
    79.