19,468
社区成员
发帖
与我相关
我的任务
分享
//皮肤检测,并针对皮肤进行增强,模糊
void MySkinEnhance(Mat &frame)
{
Mat input_image =frame;
Mat output_mask;
Mat output_image;
Mat mask;
//肤色椭圆
/*椭圆皮肤模型*/
Mat skinCrCbHist = Mat::zeros(Size(256, 256), CV_8UC1);
ellipse(skinCrCbHist, Point(113, 155.6), Size(23.4, 15.2), 43.0, 0.0, 360.0, Scalar(255, 255, 255), -1);
Mat element = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1) );
if(input_image.empty())
return ;
Mat ycrcb_image;
output_mask = Mat::zeros(input_image.size(), CV_8UC1);
cvtColor(input_image, ycrcb_image, CV_BGR2YCrCb); //首先转换成到YCrCb空间
for(int i = 0; i < input_image.rows; i++) //利用椭圆皮肤模型进行皮肤检测
{
uchar* p = (uchar*)output_mask.ptr<uchar>(i);
Vec3b* ycrcb = (Vec3b*)ycrcb_image.ptr<Vec3b>(i);
for(int j = 0; j < input_image.cols; j++)
{
if(skinCrCbHist.at<uchar>(ycrcb[j][1], ycrcb[j][2]) > 0)
p[j] = 255;
}
}
//morphologyEx(output_mask,output_mask,MORPH_CLOSE,element);
//output_mask.setTo(0);
dilate(output_mask,output_mask,Mat(16,16,CV_8U),Point(-1,-1),2);
//imwrite("dilate.jpg",dst);
// output_image.setTo(0);
input_image.copyTo(output_image, output_mask);
//for(int i = 0; i < input_image.rows; i++) //将底色设置为白色
//{
// uchar* p = (uchar*)output_mask.ptr<uchar>(i);
//
// for(int j = 0; j < input_image.cols; j++)
// {
// if((p[j]) == 0)
// {
// output_image.at<Vec3b>(i,j)[0] = 255;
// output_image.at<Vec3b>(i,j)[1] = 255;
// output_image.at<Vec3b>(i,j)[2] = 255;
// }
// }
//}
Mat enhance = output_image;
medianBlur(output_image,enhance,11);
//blur(enhance,enhance,Size(4,4),Point(-1,-1),4);
imshow("blur face",enhance);
for(int i = 0; i < output_image.rows; i++) //
{
uchar* p = (uchar*)output_mask.ptr<uchar>(i);
for(int j = 0; j < output_image.cols; j++)
{
if((enhance.at<Vec3b>(i,j)[0] < 50) && (enhance.at<Vec3b>(i,j)[1] < 50)&& (enhance.at<Vec3b>(i,j)[2] < 50) )
{
//不是纯黑的背景,加了这个判断慢了很多
}
else
{
frame.at<Vec3b>(i,j)[0] = enhance.at<Vec3b>(i,j)[0];
frame.at<Vec3b>(i,j)[1] = enhance.at<Vec3b>(i,j)[1];
frame.at<Vec3b>(i,j)[2] = enhance.at<Vec3b>(i,j)[2];
}
}
}
// 图像融合
//addWeighted(input_image, 0.95, enhance, 0.05, 0.0, input_image);
imshow("ouput image",frame);
}