2009年6月27日土曜日

WMで顔認識その4

恥ずかしいけど、顔認識プログラムのソースを公開しときます。

http://sites.google.com/site/windowsmobilesmile/

顔認識部分に関してはサンプルソースのほぼそのままです

フェイスマーク画像から、合成用のマスク画像を作成に以下の用にしました
  1. bool LoadFaceImage(LPTSTR fpath, LPTSTR mpath)  
  2.   
  3. {  
  4.   
  5.  if(fpath[0]!=0)  
  6.   
  7.  {  
  8.   
  9.   char cpath[MAX_PATH];  
  10.   
  11.   WideCharToMultiByte(CP_ACP,0,fpath,-1,cpath,sizeof(cpath),0,0);  
  12.   
  13.   face_img = cvLoadImage (cpath, CV_LOAD_IMAGE_COLOR);  
  14.   
  15.   if(mpath[0]!=0)  
  16.   
  17.   {  
  18.   
  19.    WideCharToMultiByte(CP_ACP,0,mpath,-1,cpath,sizeof(cpath),0,0);  
  20.   
  21.    mask_img = cvLoadImage (cpath, CV_LOAD_IMAGE_GRAYSCALE);  
  22.   
  23.   }  
  24.   
  25.   
  26.   
  27.   if(face_img && !mask_img)  
  28.   
  29.   {  
  30.   
  31.    IplImage *tmp = 0;  
  32.   
  33.    mask_img = cvCreateImage(cvGetSize(face_img),IPL_DEPTH_8U, 1);  
  34.   
  35. #if 1  
  36.   
  37.    tmp = cvCreateImage(cvGetSize(face_img),IPL_DEPTH_8U, 3);  
  38.   
  39.    cvCvtColor (face_img, tmp, CV_BGR2HSV);  
  40.   
  41.    uchar sh = tmp->imageData[0];  
  42.   
  43.    uchar ss = tmp->imageData[1];  
  44.   
  45.    uchar sv = tmp->imageData[2];  
  46.   
  47.    CvScalar lower = { sh -16, ss -32, sv -32, 0},upper = { sh + 16, ss + 32, sv + 32 , 0};  
  48.   
  49.    cvInRangeS(tmp, lower, upper, mask_img);  
  50.   
  51.    cvReleaseImage(&tmp);  
  52.   
  53. #else  
  54.   
  55.    uchar sb = face_img->imageData[0];  
  56.   
  57.    uchar sg = face_img->imageData[1];  
  58.   
  59.    uchar sr = face_img->imageData[2];  
  60.   
  61.    CvScalar lower = { sb -32, sg -32, sr -32, 0},upper = { sb + 32, sg + 32, sr + 32 , 0};  
  62.   
  63.    mask_img = cvCreateImage(cvGetSize(face_img),IPL_DEPTH_8U, 1);  
  64.   
  65.    cvInRangeS(face_img, lower, upper, mask_img);  
  66.   
  67. #endif  
  68.   
  69.   
  70.   
  71.    tmp = cvCreateImage(cvGetSize(face_img),IPL_DEPTH_8U, 1);  
  72.   
  73.    cvNot(mask_img,tmp);  
  74.   
  75.    cvReleaseImage(&mask_img);  
  76.   
  77.    mask_img = tmp;  
  78.   
  79.   
  80.   
  81.   }  
  82.   
  83.  }  
  84.   
  85.  return true;  
  86.   
  87. }  


(1)フェイスマーク画像をRGBからHSVに変換
cvCvtColor (face_img, tmp, CV_BGR2HSV);

(2)左端の画素のHSV値を取得
uchar sh = tmp->imageData[0];
uchar ss = tmp->imageData[1];
uchar sv = tmp->imageData[2];

(3)誤差範囲を指定して左端の画素と同じ色の画素を0、その他を255とするマスク画像を作成
CvScalar lower = { sh -16, ss -32, sv -32, 0},upper = { sh + 16, ss + 32, sv + 32 , 0};
cvInRangeS(tmp, lower, upper, mask_img);
cvReleaseImage(&tmp);

RGBのまま処理をするばあいは(1)のHSV化を省略する。
誤差範囲を調整して抜け具合を調整する


フェイスマークを画像に合成するコード
  1. bool DrawFaceMark(CvRect *r, int idx)  
  2.   
  3. {  
  4.   
  5.  if(face_img==0)  
  6.   
  7.  {  
  8.   
  9.   CvPoint center;  
  10.   
  11.   int radius;  
  12.   
  13.   center.x = cvRound (r->x + r->width * 0.5);  
  14.   
  15.   center.y = cvRound (r->y + r->height * 0.5);  
  16.   
  17.   radius = cvRound ((r->width + r->height) * 0.25);  
  18.   
  19.   cvCircle (src_img, center, radius, colors[idx % 8],-1, 8, 0);  
  20.   
  21.  }  
  22.   
  23.  else  
  24.   
  25.  {  
  26.   
  27.   CvSize sz;  
  28.   
  29.   sz.width = r->width;  
  30.   
  31.   sz.height = r->height;  
  32.   
  33.   IplImage *rfimg = cvCreateImage(sz,  IPL_DEPTH_8U, 3);  
  34.   
  35.   IplImage *rmimg = cvCreateImage(sz,  IPL_DEPTH_8U, 1);  
  36.   
  37.   IplImage *trimg = cvCreateImage(sz,  IPL_DEPTH_8U, 1);  
  38.   
  39.   cvResize(face_img, rfimg, 1);  
  40.   
  41.   cvResize(mask_img, rmimg, 1);  
  42.   
  43.   cvThreshold (rmimg, trimg, 90, 255, CV_THRESH_BINARY);  
  44.   
  45.   cvReleaseImage (&rmimg);  
  46.   
  47.   rmimg = trimg;  
  48.   
  49.     
  50.   
  51.   
  52.   
  53.      cvSetImageROI (src_img, *r);  
  54.   
  55.   cvCopy(rfimg,src_img,rmimg);  
  56.   
  57.   
  58.   
  59.   cvReleaseImage (&rfimg);  
  60.   
  61.   cvReleaseImage (&rmimg);  
  62.   
  63.  }  
  64.   
  65.  cvResetImageROI(src_img);  
  66.   
  67.    
  68.   
  69.  return true;  
  70.   
  71. }  


(1) r がフェイスマークの位置を示す構造体。rサイズの画像を3枚作る
CvSize sz;
sz.width = r->width;
sz.height = r->height;
IplImage *rfimg = cvCreateImage(sz, IPL_DEPTH_8U, 3);
IplImage *rmimg = cvCreateImage(sz, IPL_DEPTH_8U, 1);
IplImage *trimg = cvCreateImage(sz, IPL_DEPTH_8U, 1);

(2) フェイスマーク画像、マスク画像を(1)の画像にリサイズする
cvResize(face_img, rfimg, 1);
cvResize(mask_img, rmimg, 1);

(3)リサイズしたマスク画像を適当なしきい値で2値化する。(元のマスク画像は2値化されているけど、リサイズで境界部分に中間値ができる)
cvThreshold (rmimg, trimg, 90, 255, CV_THRESH_BINARY);
cvReleaseImage (&rmimg);
rmimg = trimg;

(3)元画像の置き換える部分を指定する
cvSetImageROI (src_img, *r);

(4)マスクを指定して元画像にリサイズしたフェイスマスク画像をコピーする
cvCopy(rfimg,src_img,rmimg);

(5)作った画像を開放して、(3)で指定した範囲を解除する
cvReleaseImage (&rfimg);
cvReleaseImage (&rmimg);
cvResetImageROI(src_img);

ROIを指定すると、画像の一部を書き換えることが出来る

0 件のコメント: