OpenCV——肤色检测

void SkinRGB(IplImage* src,IplImage* dst)
{
//RGB颜色空间
//均匀照明：R>95,G>40,B>20,R-B>15,R-G>15,R>B%R
//侧向照明:R>200,G>210,B>170,R-B<=15,R>B,G>B int height = src->height, width = src->width, channel = src->nChannels, step = src->widthStep;
int b = , g = , r = ; cvZero(dst);
unsigned char* p_src = (unsigned char*)src->imageData;
unsigned char* p_dst = (unsigned char*)dst->imageData; for(int j = ; j < height; j++){
for(int i = ; i < width; i++){
if((p_src[j*step+i*channel+r] > 95 && p_src[j*step+i*channel+g] > 40 && p_src[j*step+i*channel+b] > 20 &&
(p_src[j*step+i*channel+r] - p_src[j*step+i*channel+b]) > 15 && (p_src[j*step+i*channel+r] - p_src[j*step+i*channel+g]) > ) ||
(p_src[j*step+i*channel+r] > 200 && p_src[j*step+i*channel+g] > 210 && p_src[j*step+i*channel+b] > 170 &&
(p_src[j*step+i*channel+r] - p_src[j*step+i*channel+b]) <= 15 && p_src[j*step+i*channel+r] > p_src[j*step+i*channel+b] &&
p_src[j*step+i*channel+g] > p_src[j*step+i*channel+b]))
p_dst[j*width+i]=;
}
}
}

void cvSkinRG(IplImage* rgb,IplImage* gray)
{
assert(rgb->nChannels==&&gray->nChannels==); const int R=;
const int G=;
const int B=; double Aup=-1.8423;
double Bup=1.5294;
double Cup=0.0422;
double Adown=-0.7279;
double Bdown=0.6066;
double Cdown=0.1766;

for (int h=;h<rgb->height;h++) {
unsigned char* pGray=(unsigned char*)gray->imageData+h*gray->widthStep;
unsigned char* pRGB=(unsigned char* )rgb->imageData+h*rgb->widthStep;
for (int w=;w<rgb->width;w++){
int s=pRGB[R]+pRGB[G]+pRGB[B];
double r=(double)pRGB[R]/s;
double g=(double)pRGB[G]/s;
double Gup=Aup*r*r+Bup*r+Cup;
double Gdown=Adown*r*r+Bdown*r+Cdown;
double Wr=(r-0.33)*(r-0.33)+(g-0.33)*(g-0.33);
if (g<Gup && g>Gdown && Wr>0.004){
*pGray=;
}
else{
*pGray=;
}
pGray++;
pRGB+=;
}
}
}

void cvSkinOtsu(IplImage* src, IplImage* dst)
{
//Cr自适应阈值法
// IplImage* img_ycrcb=cvCreateImage(cvGetSize(src),IPL_DEPTH_8U,);
IplImage* img_cr=cvCreateImage(cvGetSize(src),IPL_DEPTH_8U,); cvCvtColor(src,img_ycrcb,CV_BGR2YCrCb);
cvSplit(img_ycrcb,,img_cr,,);
cvThresholdOtsu(img_cr,img_cr);
cvCopy(img_cr,dst); cvReleaseImage(&img_ycrcb);
cvReleaseImage(&img_cr);
} void cvThresholdOtsu(IplImage* src, IplImage* dst)
{
int height=src->height,width=src->width,threshold=;
double histogram[]={};
double average=0.0,max_variance=0.0,w=0.0,u=0.0;
IplImage* temp=cvCreateImage(cvGetSize(src),src->depth,);
if(src->nChannels!=)cvCvtColor(src,temp,CV_BGR2GRAY);
else cvCopy(src,temp); unsigned char* p_temp=(unsigned char*)temp->imageData; //计算灰度直方图
//
for(int j=;j<height;j++) {
for(int i=;i<width;i++) {
histogram[p_temp[j*width+i]]++;
}
}
for(int i=;i<;i++)histogram[i]=histogram[i]/(double)(height*width); //计算平局值
for(int i=;i<;i++)average+=i*histogram[i]; for(int i=;i<;i++) {
w+=histogram[i];
u+=i*histogram[i]; double t=average*w-u;
double variance=t*t/(w*(-w));
if(variance>max_variance) {
max_variance=variance;
threshold=i;
}
}
cvThreshold(temp,dst,threshold,,CV_THRESH_BINARY); cvReleaseImage(&temp);
}

//构造函数
CvAdaptiveSkinDetector(int samplingDivider = , int morphingMethod = MORPHING_METHOD_NONE);/*
参数1：样本采样的间隔，默认情况下为1，即表示不进行降采样
参数2：图形学操作方式，即对用皮肤检测后的图像进行图形学操作。其取值有3种可能——MORPHING_METHOD_ERODE（只进行一次腐蚀操作）；MORPHING_METHOD_ERODE_ERODE（连续进行2次腐蚀操作）；MORPHING_METHOD_ERODE_DILATE（先进行一次腐蚀操作，后进行一次膨胀操作）
*/

virtual void process(IplImage *inputBGRImage, IplImage *outputHueMask);/*
参数1：需要进行皮肤检测的输入图像
参数2：输出皮肤的掩膜图像——值为1代表该像素为皮肤，值为0代表非皮肤。
*/

#include <opencv\cv.h>
#include <opencv\highgui.h>
#include <contrib\contrib.hpp>
#include <core\core.hpp>
#include <imgproc\imgproc.hpp>
#include<time.h> int main()
{
CvCapture* capture=cvCreateCameraCapture();
cvNamedWindow("Input Video",);
cvNamedWindow("Output Video",); IplImage* img_src=NULL;
IplImage* input_img=NULL;
IplImage* output_mask=NULL;
IplImage* output_img=NULL; clock_t start,finish;
double duration; CvAdaptiveSkinDetector skin_detector(,CvAdaptiveSkinDetector::MORPHING_METHOD_ERODE_DILATE); //定义肤色检测算子 while()
{
img_src=cvQueryFrame(capture);
if(!img_src) break;
cvShowImage("Input Video",img_src);
if(input_img==NULL){
input_img=cvCreateImage(cvGetSize(img_src),img_src->depth,img_src->nChannels);
}
cvCopy(img_src,input_img); output_img=cvCreateImage(cvGetSize(img_src),img_src->depth,img_src->nChannels);
cvZero(output_img); if(output_mask==NULL){
output_mask=cvCreateImage(cvGetSize(img_src),img_src->depth,);
} //肤色检测
//
start=clock();
skin_detector.process(input_img,output_mask);
finish=clock();
duration=(double)(finish-start)/CLOCKS_PER_SEC;
printf("elapsed time :%.0f 毫秒\n",duration*); cvCopy(img_src,output_img,output_mask);
cvShowImage("Output Video",output_img); char c=cvWaitKey();
if(c==)break;
} cvReleaseCapture(&capture);
cvDestroyWindow("Video");
}

void cvSkinHSV(IplImage* src,IplImage* dst)
{
IplImage* hsv=cvCreateImage(cvGetSize(src),,);
//IplImage* cr=cvCreateImage(cvGetSize(src),8,1);
//IplImage* cb=cvCreateImage(cvGetSize(src),8,1);
cvCvtColor(src,hsv,CV_BGR2HSV);
//cvSplit(ycrcb,0,cr,cb,0); static const int V=;
static const int S=;
static const int H=; //IplImage* dst=cvCreateImage(cvGetSize(_dst),8,3);
cvZero(dst); for (int h=;h<src->height;h++) {
unsigned char* phsv=(unsigned char*)hsv->imageData+h*hsv->widthStep;
unsigned char* psrc=(unsigned char*)src->imageData+h*src->widthStep;
unsigned char* pdst=(unsigned char*)dst->imageData+h*dst->widthStep;
for (int w=;w<src->width;w++) {
if (phsv[H]>=&&phsv[H]<=)
{
memcpy(pdst,psrc,);
}
phsv+=;
psrc+=;
pdst+=;
}
}
//cvCopyImage(dst,_dst);
//cvReleaseImage(&dst);
}