教你用 30 行代码实现自动戴口罩

最新推荐文章于 2024-04-26 18:32:23 发布

编程IT圈

最新推荐文章于 2024-04-26 18:32:23 发布

阅读量334

点赞数

文章标签：人脸识别计算机视觉人工智能编程语言 opencv

自1月20日钟南山院士肯定此次新型肺炎存在着人传人的现象起，到今日30个省份启动一级响应，多省市发布“重大突发公共卫生事件一级响应”意味着什么？，我们人民群众需要提高对此次疫情的重视程度，做好自身的防范工作，平平安安，过个好年。

尽量少去人群密集的场所，如果非要出门一定要戴上口罩，勤用肥皂和清水或含有酒精的洗手液洗手也是非常有必要的。

在网络世界中，我们可以通过给头像戴口罩，来呼吁广大群众积极保护自身安全。在这里，我们运用Python简单的几十行代码来实现在社交网络中也给自己的头像自动戴上口罩。

大致的效果如下：

用到的技术主要有人脸识别（当然，调用了接口）、openCV图像处理这两项。

人脸识别

旷视提供了人脸识别的API，输入图片便可以得到人脸的各个稠密关键点的所在位置，通俗来讲，就是可以自动勾画出五官。

我们通过使用其嘴巴的位置数据，可以定位口罩的佩戴位置。并通过计算人脸嘴巴的大小，自动调整口罩的大小进行适配。

def get_mouth(dst_pic):
    with open(dst_pic, 'rb') as f:
        base64_data = base64.b64encode(f.read())
    url='https://api-cn.faceplusplus.com/facepp/v1/face/thousandlandmark'
    headers={'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/76.0.3809.132 Safari/537.36'}
    data={
             # api_key,api_secret需自己申请
          'api_key':'',
          'api_secret':'',
          'return_landmark': 'mouth',
          'image_base64': base64_data
                         }
    r=requests.post(url,headers=headers,data=data)
    mouth=r.json()['face']['landmark']['mouth']
    x,y=[],[]
    for i in mouth.values():
        y.append(i['y'])
        x.append(i['x'])
    y_max=max(y)
    y_min=min(y)
    x_max=max(x)
    x_min=min(x)
    middle_x=int((x_max+x_min)/2)
    middle_y=int((y_max+y_min)/2)
    size=(int(3*(x_max-x_min)),int(5*(y_max-y_min)))
    return (middle_x,middle_y),size

图像处理

有了口罩的估计大小，可以自动更改口罩的图像大小，根据计算得到的口罩的中心位置来摆放口罩，并通过简单的掩膜处理，利用seamlessClone函数将口罩照片添加到头像图片上。

def add_mask(img_path,img_outPath):
    src_pic="/root/Documents/abc.jpg"
    center,size=get_mouth(img_path)
    src=cv2.imread(src_pic)
    src=cv2.resize(src,size)
    dst=cv2.imread(img_path)
    # 掩膜mask
    mask=255*np.ones(src.shape, src.dtype)
    output=cv2.seamlessClone(src, dst, mask, center, cv2.NORMAL_CLONE)
    cv2.imwrite(img_outPath, output)

当然，这只是一个小demo, 有许多地方可以改进，例如我们可以根据鼻梁的倾斜程度判断人脸的倾斜程度，从而对口罩做相应的旋转操作，更加适配人脸。

-END-

新关注我们公众号的朋友，可以在公众号后台聊天框回复【1024】，可以免费获取2T的编程视频，算法、人工智能、Python、Java、Linux、Go、C语言、软考、英语等等资源。