欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python传统图像处理之皮肤区域检测详解

程序员文章站 2022-03-02 20:52:26
目录1.rgb空间2.ycrcb空间3.ycrcb空间+otsu4.hsv空间5.opencv自带肤色检测类adaptiveskindetector6.基于椭圆模型7.直方图反向投影1.rgb空间肤色...

1.rgb空间

肤色在rgb模型下的范围基本满足以下约束:

在均匀光照下应满足以下判别式:

r>95 and g>40 b>20 and max(r,g,b)-min(r,g,b)>15 and abs(r-g)>15 and r>g and r>b

在侧光拍摄环境下:

r>220 and g>210 and b>170 and abs(r-g)<=15 and r>b and g>b

代码:

def skinmask_rgb(image):
    b, g, r = cv2.split(image)
    mask_uniformity = (r>95)*(g>40)*(b>20)* (np.max(image, axis=2) -np.min(image,axis=2)>15)* (abs(r-g)>15)*(r>g)*(r>b)
    mask_side = (r>220)*(g>210)*(b>170)*(abs(r-g)<=15)*(r>b)*(g>b)
    mask = mask_uniformity|mask_side
    skin = np.array(mask, np.uint8)*255
    ratio = np.sum(skin/255)/(image.shape[0]*image.shape[1])
 
    return skin, ratio

效果:

Python传统图像处理之皮肤区域检测详解Python传统图像处理之皮肤区域检测详解

2.ycrcb空间

在rgb空间里人脸的肤色受亮度影响相当大,所以肤色点很难从非肤色点中分离出来,也就是说在此空间经过处理后,肤色点是离散的点,中间嵌有很多非肤色。如果把rgb转为ycrcb空间的话,可以忽略y(亮度)的影响,因为该空间受亮度影响很小,肤色会产生很好的类聚。这样就把三维的空间降为二维的crcb,肤色点会形成一定得形状。

133≤cr≤173

77≤cb≤127

代码:

def skinmask_ycrcb(image):
    ycrcb = cv2.cvtcolor(image, cv2.color_bgr2ycr_cb) #转换至ycrcb空间
    (y,cr,cb) = cv2.split(ycrcb) #拆分出y,cr,cb值
    skin = np.zeros(cr.shape, dtype = np.uint8)
    (x,y) = cr.shape
    
    cr_mask1 = cr >= 133
    cr_mask2 = cr <= 173
    cr_mask = cr_mask1 * cr_mask2
 
    cb_mask1 = cb >= 77
    cb_mask2 = cb <= 127
    cb_mask = cb_mask1 * cb_mask2
    mask = cr_mask * cb_mask
    skin[mask] = 255
 
    ratio = np.sum(skin/255)/(x*y)
    return skin, ratio

效果:

Python传统图像处理之皮肤区域检测详解Python传统图像处理之皮肤区域检测详解

3.ycrcb空间+otsu

y表示明亮度(luminance或luma),也就是灰阶值。cr反映了rgb输入信号红色部分与rgb信号亮度值之间的差异。而cb反映的是rgb输入信号蓝色部分与rgb信号亮度值之间的差异。

将rgb图像转换到ycrcb颜色空间,提取cr分量图像。对cr做自二值化阈值分割处理(otsu法)。

代码:

def skinmask_ycrcb_otsu(image):
    ycrcb = cv2.cvtcolor(image, cv2.color_bgr2ycr_cb) #转换至ycrcb空间
    (y,cr,cb) = cv2.split(ycrcb) #拆分出y,cr,cb值
    
    ret, skin = cv2.threshold(cr, 0, 255, cv2.thresh_binary | cv2.thresh_otsu)
 
    ratio = np.sum(skin/255)/(image.shape[0]*image.shape[1])
    return skin, ratio

效果:

Python传统图像处理之皮肤区域检测详解Python传统图像处理之皮肤区域检测详解

4.hsv空间

7<h<20

28<s<256

50<v<256

代码:

def skinmask_hsv(image):
    hsv = cv2.cvtcolor(image, cv2.color_bgr2hsv)
    h, s, v = cv2.split(hsv)
 
    mask = (h>7)*(h<20)*(s>28)*(v>50)
    skin = np.array(mask, np.uint8)*255
    ratio = np.sum(skin/255)/(image.shape[0]*image.shape[1])
    return skin, ratio

效果:

Python传统图像处理之皮肤区域检测详解Python传统图像处理之皮肤区域检测详解

5.opencv自带肤色检测类adaptiveskindetector

已经从opencv 3以上版本中移除,opencv 2中函数。

6.基于椭圆模型

将皮肤信息映射到ycrcb空间,则在crcb二维空间中这些皮肤像素点近似成一个椭圆分布。因此如果我们得到了一个crcb的椭圆,下次来一个坐标(cr, cb)我们只需判断它是否在椭圆内(包括边界),如果是,则可以判断其为皮肤,否则就是非皮肤像素点。

代码:

def skinmask_ellipse(image):
    ycrcb = cv2.cvtcolor(image, cv2.color_bgr2ycr_cb) #转换至ycrcb空间
    (y,cr,cb) = cv2.split(ycrcb) #拆分出y,cr,cb值
 
    skincrcbhist = np.zeros((256,256),np.uint8)
    skincrcbhist = cv2.ellipse(skincrcbhist, (113, 155), (23, 15), 43.0, 0.0, 360.0, (255, 255, 255), -1)
    cv2.imwrite("ww.jpg", skincrcbhist)
 
    skin = cv2.remap(skincrcbhist, cb.astype(np.float32), cr.astype(np.float32), interpolation=cv2.inter_linear)
    ratio = np.sum(skin/255)/(image.shape[0]*image.shape[1])
    return skin, ratio

效果:

椭圆模型

Python传统图像处理之皮肤区域检测详解

效果

Python传统图像处理之皮肤区域检测详解Python传统图像处理之皮肤区域检测详解

7.直方图反向投影

反向投影可以用来做图像分割,寻找感兴趣区间。可以寻找任何的roi,这里将其用作寻找人体肤色。它会输出与输入图像大小相同的图像,每一个像素值代表了输入图像上对应点属于目标对象的概率,简言之,输出图像中像素值越高的点越可能代表想要查找的目标。直方图投影经常与camshift(追踪算法)算法一起使用。

操作步骤:

(1)截取若干张不同肤色的图片,最好覆盖常见的所有肤色(如下图)

Python传统图像处理之皮肤区域检测详解

(2)对所有肤色的图片一起做色彩直方图(利用opencv中的calchist函数)

(3)新建一个与待检测图片同尺寸的灰度图片,找到待检测图片中每颗像素点的颜色在色彩直方图中对应栅格的数值(即统计中出现的次数),并将该数值赋值予新建灰度图片中与该检测像素同位置的像素(利用opencv中的calcbackproject函数)。图像中像素亮度越高,待检测图像该位置处是肤色的概率越大。

代码:

def skinmask_hist_backproject(target, roi_image):
    #roi图片,就想要找的的图片
    hsv = cv2.cvtcolor(roi_image,cv2.color_bgr2hsv)
 
    #目标搜索图片
    hsvt = cv2.cvtcolor(target,cv2.color_bgr2hsv)
 
    #计算目标直方图
    roihist = cv2.calchist([hsv],[0,1],none,[180,256],[0,180,0,256])
    #归一化,参数为原图像和输出图像,归一化后值全部在2到255范围
    cv2.normalize(roihist,roihist,0,255,cv2.norm_minmax)
    dst = cv2.calcbackproject([hsvt],[0,1],roihist,[0,180,0,256],1)
 
 
    #卷积连接分散的点
    disc = cv2.getstructuringelement(cv2.morph_ellipse,(5,5))
    dst = cv2.filter2d(dst,-1,disc)
 
    ret,skin = cv2.threshold(dst,50,255,0)
 
    ratio = np.sum(skin/255)/(target.shape[0]*target.shape[1])
return skin, ratio

效果: 

 Python传统图像处理之皮肤区域检测详解

以上就是python传统图像处理之皮肤区域检测详解的详细内容,更多关于python 皮肤区域检测的资料请关注其它相关文章!