肤色检测与分割方法

几种肤色分割算法

        在手势识别和人脸识别中,肤色分割是非常重要的,特将几种肤色分割方法总结了一下,将代码贴出。

        ps:有部分代码非原创,若有侵权,修改。

        包括在rgb、rg空间上进行分割,以及大津分割法在多个颜色空间上的实现。

[cpp]  view plain  copy
  1. #include "highgui.h"  
  2. #include "cv.h"  
  3. #include<iostream>  
  4. using namespace std;  
  5. // skin region location using rgb limitation  
  6. void SkinRGB(IplImage* rgb,IplImage* _dst)  
  7. {  
  8.     assert(rgb->nChannels==3&& _dst->nChannels==3);  
  9.   
  10.     static const int R=2;  
  11.     static const int G=1;  
  12.     static const int B=0;  
  13.   
  14.     IplImage* dst=cvCreateImage(cvGetSize(_dst),8,3);  
  15.     cvZero(dst);  
  16.   
  17.     for (int h=0; h<rgb->height; h++)  
  18.     {  
  19.         unsigned char* prgb=(unsigned char*)rgb->imageData+h*rgb->widthStep;  
  20.         unsigned char* pdst=(unsigned char*)dst->imageData+h*dst->widthStep;  
  21.         for (int w=0; w<rgb->width; w++)  
  22.         {  
  23.             if ((prgb[R]>95 && prgb[G]>40 && prgb[B]>20 &&  
  24.                     prgb[R]-prgb[B]>15 && prgb[R]-prgb[G]>15)||//uniform illumination  
  25.                     (prgb[R]>200 && prgb[G]>210 && prgb[B]>170 &&  
  26.                      abs(prgb[R]-prgb[B])<=15 && prgb[R]>prgb[B]&& prgb[G]>prgb[B])//lateral illumination  
  27.                )  
  28.             {  
  29.                 memcpy(pdst,prgb,3);  
  30.             }  
  31.             prgb+=3;  
  32.             pdst+=3;  
  33.         }  
  34.     }  
  35.     cvCopyImage(dst,_dst);  
  36.     cvReleaseImage(&dst);  
  37. }  
  38. // skin detection in rg space  
  39. void cvSkinRG(IplImage* rgb,IplImage* gray)  
  40. {  
  41.     assert(rgb->nChannels==3&&gray->nChannels==1);  
  42.   
  43.     const int R=2;  
  44.     const int G=1;  
  45.     const int B=0;  
  46.   
  47.     double Aup=-1.8423;  
  48.     double Bup=1.5294;  
  49.     double Cup=0.0422;  
  50.     double Adown=-0.7279;  
  51.     double Bdown=0.6066;  
  52.     double Cdown=0.1766;  
  53.     for (int h=0; h<rgb->height; h++)  
  54.     {  
  55.         unsigned char* pGray=(unsigned char*)gray->imageData+h*gray->widthStep;  
  56.         unsigned char* pRGB=(unsigned char* )rgb->imageData+h*rgb->widthStep;  
  57.         for (int w=0; w<rgb->width; w++)  
  58.         {  
  59.             int s=pRGB[R]+pRGB[G]+pRGB[B];  
  60.             double r=(double)pRGB[R]/s;  
  61.             double g=(double)pRGB[G]/s;  
  62.             double Gup=Aup*r*r+Bup*r+Cup;  
  63.             double Gdown=Adown*r*r+Bdown*r+Cdown;  
  64.             double Wr=(r-0.33)*(r-0.33)+(g-0.33)*(g-0.33);  
  65.             if (g<Gup && g>Gdown && Wr>0.004)  
  66.             {  
  67.                 *pGray=255;  
  68.             }  
  69.             else  
  70.             {  
  71.                 *pGray=0;  
  72.             }  
  73.             pGray++;  
  74.             pRGB+=3;  
  75.         }  
  76.     }  
  77.   
  78. }  
  79. // implementation of otsu algorithm  
  80. // author: onezeros#yahoo.cn  
  81. // reference: Rafael C. Gonzalez. Digital Image Processing Using MATLAB  
  82. void cvThresholdOtsu(IplImage* src, IplImage* dst)  
  83. {  
  84.     int height=src->height;  
  85.     int width=src->width;  
  86.   
  87.     //histogram  
  88.     float histogram[256]= {0};  
  89.     for(int i=0; i<height; i++)  
  90.     {  
  91.         unsigned char* p=(unsigned char*)src->imageData+src->widthStep*i;  
  92.         for(int j=0; j<width; j++)  
  93.         {  
  94.             histogram[*p++]++;  
  95.         }  
  96.     }  
  97.     //normalize histogram  
  98.     int size=height*width;  
  99.     for(int i=0; i<256; i++)  
  100.     {  
  101.         histogram[i]=histogram[i]/size;  
  102.     }  
  103.   
  104.     //average pixel value  
  105.     float avgValue=0;  
  106.     for(int i=0; i<256; i++)  
  107.     {  
  108.         avgValue+=i*histogram[i];  
  109.     }  
  110.   
  111.     int threshold;  
  112.     float maxVariance=0;  
  113.     float w=0,u=0;  
  114.     for(int i=0; i<256; i++)  
  115.     {  
  116.         w+=histogram[i];  
  117.         u+=i*histogram[i];  
  118.   
  119.         float t=avgValue*w-u;  
  120.         float variance=t*t/(w*(1-w));  
  121.         if(variance>maxVariance)  
  122.         {  
  123.             maxVariance=variance;  
  124.             threshold=i;  
  125.         }  
  126.     }  
  127.   
  128.     cvThreshold(src,dst,threshold,255,CV_THRESH_BINARY);  
  129. }  
  130.   
  131. void cvSkinOtsu(IplImage* src, IplImage* dst)//yCbCr  
  132. {  
  133.     assert(dst->nChannels==1&& src->nChannels==3);  
  134.   
  135.     IplImage* ycrcb=cvCreateImage(cvGetSize(src),8,3);  
  136.     IplImage* cr=cvCreateImage(cvGetSize(src),8,1);  
  137.     cvCvtColor(src,ycrcb,CV_BGR2YCrCb);  
  138.     cvSplit(ycrcb,0,cr,0,0);  
  139.   
  140.     cvThresholdOtsu(cr,cr);  
  141.     cvCopyImage(cr,dst);  
  142.     cvReleaseImage(&cr);  
  143.     cvReleaseImage(&ycrcb);  
  144. }  
  145.   
  146. void cvSkinOtsu_H(IplImage* src, IplImage* dst)//H通道上做分割,找到区域后再映射到S通道上继续做分割  
  147. {  
  148.     assert(dst->nChannels==1&& src->nChannels==3);  
  149.   
  150.     IplImage* HSV=cvCreateImage(cvGetSize(src),8,3);  
  151.     IplImage* H=cvCreateImage(cvGetSize(src),8,1);  
  152.     IplImage* S=cvCreateImage(cvGetSize(src),8,1);  
  153.     cvCvtColor(src,HSV,CV_BGR2HSV);  
  154.     cvSplit(HSV,H,S,0,0);  
  155.   
  156.     cvThresholdOtsu(H,S);  
  157.     cvCopyImage(S,dst);  
  158.     cvReleaseImage(&H);  
  159.     cvReleaseImage(&HSV);  
  160. }  
  161.   
  162. void cvSkinYUV(IplImage* src,IplImage* dst)  
  163. {  
  164.     IplImage* ycrcb=cvCreateImage(cvGetSize(src),8,3);  
  165.     //IplImage* cr=cvCreateImage(cvGetSize(src),8,1);  
  166.     //IplImage* cb=cvCreateImage(cvGetSize(src),8,1);  
  167.     cvCvtColor(src,ycrcb,CV_BGR2YCrCb);  
  168.     //cvSplit(ycrcb,0,cr,cb,0);  
  169.   
  170.     static const int Cb=2;  
  171.     static const int Cr=1;  
  172.     static const int Y=0;  
  173.   
  174.     //IplImage* dst=cvCreateImage(cvGetSize(_dst),8,3);  
  175.     cvZero(dst);  
  176.   
  177.     for (int h=0; h<src->height; h++)  
  178.     {  
  179.         unsigned char* pycrcb=(unsigned char*)ycrcb->imageData+h*ycrcb->widthStep;  
  180.         unsigned char* psrc=(unsigned char*)src->imageData+h*src->widthStep;  
  181.         unsigned char* pdst=(unsigned char*)dst->imageData+h*dst->widthStep;  
  182.         for (int w=0; w<src->width; w++)  
  183.         {  
  184.             if (pycrcb[Cr]>=133&&pycrcb[Cr]<=173&&pycrcb[Cb]>=77&&pycrcb[Cb]<=127)  
  185.             {  
  186.                 memcpy(pdst,psrc,3);  
  187.             }  
  188.             pycrcb+=3;  
  189.             psrc+=3;  
  190.             pdst+=3;  
  191.         }  
  192.     }  
  193.     //cvCopyImage(dst,_dst);  
  194.     //cvReleaseImage(&dst);  
  195. }  
  196.   
  197. void cvSkinHSV(IplImage* src,IplImage* dst)  
  198. {  
  199.     IplImage* hsv=cvCreateImage(cvGetSize(src),8,3);  
  200.     //IplImage* cr=cvCreateImage(cvGetSize(src),8,1);  
  201.     //IplImage* cb=cvCreateImage(cvGetSize(src),8,1);  
  202.     cvCvtColor(src,hsv,CV_BGR2HSV);  
  203.     //cvSplit(ycrcb,0,cr,cb,0);  
  204.   
  205.     static const int V=2;  
  206.     static const int S=1;  
  207.     static const int H=0;  
  208.   
  209.     //IplImage* dst=cvCreateImage(cvGetSize(_dst),8,3);  
  210.     cvZero(dst);  
  211.   
  212.     for (int h=0; h<src->height; h++)  
  213.     {  
  214.         unsigned char* phsv=(unsigned char*)hsv->imageData+h*hsv->widthStep;  
  215.         unsigned char* psrc=(unsigned char*)src->imageData+h*src->widthStep;  
  216.         unsigned char* pdst=(unsigned char*)dst->imageData+h*dst->widthStep;  
  217.         for (int w=0; w<src->width; w++)  
  218.         {  
  219.             if (phsv[H]>=7&&phsv[H]<=29)  
  220.             {  
  221.                 memcpy(pdst,psrc,3);  
  222.             }  
  223.             phsv+=3;  
  224.             psrc+=3;  
  225.             pdst+=3;  
  226.         }  
  227.     }  
  228.     //cvCopyImage(dst,_dst);  
  229.     //cvReleaseImage(&dst);  
  230. }  
  231.   
  232. void Skin_HSV_2(IplImage *initial,IplImage *distinction)  
  233. {  
  234.     CvScalar Avg,Sdv;  
  235.     IplImage *temp = cvCreateImage(cvGetSize(initial),8,3);  
  236.     cvCvtColor(initial,temp,CV_BGR2HSV);  
  237.     IplImage *h_img = cvCreateImage(cvGetSize(initial),8,1);  
  238.     IplImage *s_img = cvCreateImage(cvGetSize(initial),8,1);  
  239.     IplImage *v_img = cvCreateImage(cvGetSize(initial),8,1);  
  240.   
  241.     double h_avg,s_avg,v_avg;  
  242.     double h_sdv,s_sdv,v_sdv;  
  243.   
  244.     cvAvgSdv(temp,&Avg,&Sdv,0);  
  245.         
  246.     h_avg = Avg.val[0];  
  247.     s_avg = Avg.val[1];  
  248.     v_avg = Avg.val[2];  
  249.     //cout<<h_avg<<'\n'<<s_img<<'\n'<<v_img;  
  250.     h_sdv = Sdv.val[0];  
  251.     s_sdv = Sdv.val[1];  
  252.     v_sdv = Sdv.val[2];  
  253.   
  254.   
  255.     cvSplit(temp, h_img, s_img, v_img, 0);  
  256.     /*cvShowImage("h_img1",h_img); 
  257.     cvShowImage("s_img1",s_img); 
  258.     cvShowImage("v_img1",v_img);*/  
  259.     int value = 0;  
  260.     for(int i= 0;i< h_img->height;i++)  
  261.     {  
  262.         for(int j = 0; j< h_img->width; j++)  
  263.         {  
  264.             value = cvGetReal2D(h_img, i, j);  
  265.             if((value<(h_avg+h_sdv)) && (value>(h_avg-h_sdv)) )  
  266.             {  
  267.                 *(h_img->imageData+i*h_img->widthStep+j) = 0;  
  268.             }  
  269.             else  
  270.             {  
  271.                 *(h_img->imageData+i*h_img->widthStep+j) = 255;  
  272.             }  
  273.         }  
  274.     }  
  275.   
  276.   
  277.     for(int i= 0;i< s_img->height;i++)  
  278.     {  
  279.         for(int j = 0; j< s_img->width; j++)  
  280.         {  
  281.             value = cvGetReal2D(s_img, i, j);  
  282.             if((value<(s_avg+0.5*s_sdv))&&(value>(s_avg-0.5*s_sdv)) )  
  283.             {  
  284.                 *(s_img->imageData+i*s_img->widthStep+j) = 0;  
  285.             }  
  286.             else  
  287.             {  
  288.                 *(s_img->imageData+i*s_img->widthStep+j) = 255;  
  289.             }  
  290.         }  
  291.     }  
  292.   
  293.     for(int i= 0;i< v_img->height;i++)  
  294.     {  
  295.         for(int j = 0; j< v_img->width; j++)  
  296.         {  
  297.             value = cvGetReal2D(v_img, i, j);  
  298.             if((value<(v_avg+v_sdv)) && (value>(v_avg-v_sdv)))  
  299.             {  
  300.                 *(v_img->imageData+i*v_img->widthStep+j) = 0;  
  301.             }  
  302.             else  
  303.                   
  304.             {  
  305.                     *(v_img->imageData+i*v_img->widthStep+j) = 255;  
  306.             }  
  307.   
  308.         }  
  309.     }  
  310.     cvShowImage("h_img",h_img);  
  311.     /*cvShowImage("s_img",s_img); 
  312.     cvShowImage("v_img",v_img);*/  
  313.     cvAnd(h_img,s_img,distinction);  
  314.     //cvAnd(v_img,distinction,distinction);  
  315.   
  316.     /*cvShowImage("distinction",distinction);*/  
  317.       
  318.   
  319.     /*cvReleaseImage(&temp);*/  
  320.     cvReleaseImage(&h_img);  
  321.     cvReleaseImage(&s_img);  
  322.     cvReleaseImage(&v_img);  
  323. }  
  324.   
  325. void main()  
  326. {  
  327.   
  328.     IplImage* img= cvLoadImage("original.jpg"); //随便放一张jpg图片在D盘或另行设置目录  
  329.     IplImage* dstRGB=cvCreateImage(cvGetSize(img),8,3);  
  330.     IplImage* dstRG=cvCreateImage(cvGetSize(img),8,1);  
  331.     IplImage* dst_crotsu=cvCreateImage(cvGetSize(img),8,1);  
  332.     IplImage* dst_YUV=cvCreateImage(cvGetSize(img),8,3);  
  333.     IplImage* dst_HSV=cvCreateImage(cvGetSize(img),8,3);  
  334.     IplImage* dst_crotsu_H=cvCreateImage(cvGetSize(img),8,1);  
  335.   
  336.     IplImage* dst_HSV_2=cvCreateImage(cvGetSize(img),8,1);  
  337.   
  338.     cvNamedWindow("inputimage", CV_WINDOW_AUTOSIZE);  
  339.     cvShowImage("inputimage", img);  
  340.     cvWaitKey(0);  
  341.   
  342.     SkinRGB(img,dstRGB);  
  343.     cvNamedWindow("SkinRGB", CV_WINDOW_AUTOSIZE);  
  344.     cvShowImage("SkinRGB", dstRGB);  
  345.     ///cvWaitKey(0);  
  346.     cvSkinRG(img,dstRG);  
  347.     cvNamedWindow("cvSkinRG", CV_WINDOW_AUTOSIZE);  
  348.     cvShowImage("cvSkinRG", dstRG);  
  349.     //cvWaitKey(0);  
  350.     cvSkinOtsu(img,dst_crotsu);  
  351.     cvNamedWindow("cvSkinOtsu", CV_WINDOW_AUTOSIZE);  
  352.     cvShowImage("cvSkinOtsu", dst_crotsu);  
  353.   
  354.     cvSkinOtsu_H(img,dst_crotsu_H);  
  355.     cvNamedWindow("cvSkinOtsu_H", CV_WINDOW_AUTOSIZE);  
  356.     cvShowImage("cvSkinOtsu_H", dst_crotsu_H);  
  357.     //cvWaitKey(0);  
  358.     cvSkinYUV(img,dst_YUV);  
  359.     cvNamedWindow("cvSkinYUV", CV_WINDOW_AUTOSIZE);  
  360.     cvShowImage("cvSkinYUV", dst_YUV);  
  361.     //cvWaitKey(0);  
  362.     cvSkinHSV(img,dst_HSV);  
  363.     cvNamedWindow("cvSkinHSV", CV_WINDOW_AUTOSIZE);  
  364.     cvShowImage("cvSkinHSV", dst_HSV);  
  365.   
  366.   
  367.     Skin_HSV_2(img, dst_HSV_2);  
  368.     cvShowImage("cvSkinHSV2", dst_HSV_2);  
  369.     cvWaitKey(0);  
  370.       
  371. }  

       经过多次实验,发现在HSV空间上的大津分割较好。

[cpp]  view plain  copy
  1. void cvThresholdOtsu(IplImage* src, IplImage* dst)  
  2. {  
  3.     int height=src->height;  
  4.     int width=src->width;  
  5.   
  6.     //histogram  
  7.     float histogram[256]= {0};  
  8.     for(int i=0; i<height; i++)  
  9.     {  
  10.         unsigned char* p=(unsigned char*)src->imageData+src->widthStep*i;  
  11.         for(int j=0; j<width; j++)  
  12.         {  
  13.             histogram[*p++]++;  
  14.         }  
  15.     }  
  16.     //normalize histogram  
  17.     int size=height*width;  
  18.     for(int i=0; i<256; i++)  
  19.     {  
  20.         histogram[i]=histogram[i]/size;  
  21.     }  
  22.   
  23.     //average pixel value  
  24.     float avgValue=0;  
  25.     for(int i=0; i<256; i++)  
  26.     {  
  27.         avgValue+=i*histogram[i];  
  28.     }  
  29.   
  30.     int threshold;  
  31.     float maxVariance=0;  
  32.     float w=0,u=0;  
  33.     for(int i=0; i<256; i++)  
  34.     {  
  35.         w+=histogram[i];  
  36.         u+=i*histogram[i];  
  37.   
  38.         float t=avgValue*w-u;  
  39.         float variance=t*t/(w*(1-w));  
  40.         if(variance>maxVariance)  
  41.         {  
  42.             maxVariance=variance;  
  43.             threshold=i;  
  44.         }  
  45.     }  
  46.   
  47.     cvThreshold(src,dst,threshold,255,CV_THRESH_BINARY);  
  48. }  
  49.   
  50. void cvSkinOtsu(IplImage* src, IplImage* dst)//yCbCr  
  51. {  
  52.     assert(dst->nChannels==1&& src->nChannels==3);  
  53.   
  54.     IplImage* ycrcb=cvCreateImage(cvGetSize(src),8,3);  
  55.     IplImage* cr=cvCreateImage(cvGetSize(src),8,1);  
  56.     cvCvtColor(src,ycrcb,CV_BGR2YCrCb);  
  57.     cvSplit(ycrcb,0,cr,0,0);  
  58.   
  59.     cvThresholdOtsu(cr,cr);  
  60.     cvCopyImage(cr,dst);  
  61.     cvReleaseImage(&cr);  
  62.     cvReleaseImage(&ycrcb);  
  63. }  
  64.   
  65. void cvSkinOtsu_H(IplImage* src, IplImage* dst)//H通道上做分割,找到区域后再映射到S通道上继续做分割  
  66. {  
  67.     assert(dst->nChannels==1&& src->nChannels==3);  
  68.   
  69.     IplImage* HSV=cvCreateImage(cvGetSize(src),8,3);  
  70.     IplImage* H=cvCreateImage(cvGetSize(src),8,1);  
  71.     IplImage* S=cvCreateImage(cvGetSize(src),8,1);  
  72.     IplImage* V=cvCreateImage(cvGetSize(src),8,1);  
  73.     cvCvtColor(src,HSV,CV_BGR2HSV);  
  74.     cvSplit(HSV,H,S,V,0);  
  75.     cvThresholdOtsu(S,S);  
  76.     cvThresholdOtsu(V,V);  //H通道,效果不好,S通道还行  
  77.     cvAnd(S,V,dst);   //S,V通道与  
  78.     //cvCopyImage(V,dst);  
  79.     cvReleaseImage(&H);  
  80.     cvReleaseImage(&HSV);  
  81. }  

此外,还有一种在HSV空间上判决的方法,通过(S+V)/H 的值进行判决的,是在一篇论文上看到的,然而,后来没有找到出处。我根据需要对阈值进行了变化,效果在特定情况下比较好,不具有普遍性。

[cpp]  view plain  copy
  1. void Skin_HSV_new(IplImage *img,IplImage *dst)  
  2. {  
  3.     //cvEqualizeHist(img,img); //直方图均衡  
  4.     cvCvtColor(img,img, CV_BGR2HSV);  
  5.     for(int i=0;i<img->width;i++)  
  6.     {  
  7.         for(int j=0;j<img->height;j++)  
  8.         {  
  9.             CvScalar temp=cvGet2D(img,j,i);   
  10.             int value = (((temp.val[1]+temp.val[2])*1.0)/temp.val[0]);  
  11.             if(value<8)  
  12.             {  
  13.                 *(dst->imageData+j*dst->widthStep+i)=0;  
  14.             }  
  15.             else  
  16.             {  
  17.                 *(dst->imageData+j*dst->widthStep+i)=255;  
  18.             }  
  19.             //cout<<value<<'\n';  
  20.         }  
  21.     }  
  22.   
  23.     //cvSmooth(dst,dst,CV_MEDIAN);  
  24.     /*cvDilate(dst,dst); 
  25.     cvErode(dst,dst); 
  26.     cvErode(dst,dst); 
  27.     cvDilate(dst,dst); 
  28.  
  29.     cvDilate(dst,dst); 
  30.     cvErode(dst,dst); 
  31.     cvErode(dst,dst); 
  32.     cvSmooth(dst,dst,CV_MEDIAN); 
  33.     cvDilate(dst,dst); 
  34.     cvErode(dst,dst);*/  
  35.     cvDilate(dst,dst);  
  36.     cvErode(dst,dst);  
  37.     //cvSmooth(dst,dst,CV_BILATERAL);  
  38.     //cvSmooth(dst,dst);  
  39. }  
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值