【Python案例】一键自动抠图生成证件照

0 效果与体验

不想去照相馆？担心肖像隐私被第三方获取？不会抠图？本文实现基于人工智能的一键自动抠图生成证件照。在进入正文之前，先看最终效果：

为了让读者快速体验，我写了个小程序: 证照工具箱，可打开直接体验。

1 人脸检测

在制作证件照时，首选需要检测人脸。根据指定的输出大小将人脸区域等比裁剪。

1.1 人脸检测

这里使用了Retina人脸检测模型，因为篇幅原因，具体使用方法不再描述详细调用过程，如有需要请关注我的公众号：Python学习实战，后台回复：Retina。即可获取作者已封装好的人脸检测调用源码。

1.2 人脸裁剪

具体实现代码逻辑如下：

def crop(rgb, dh, dw):
    oh, ow, _ = rgb.shape
    bgr = cv2.cvtColor(rg