Python实现人头(脸)抠图的实现方法

Question

有没有python库或者API接口可以实现自动抠图人头(脸)的？不要扣全身的，百度API和rembg.remove库都是扣全身，不能只扣人脸。

Answer 1

参考GPT和自己的思路，您可以使用Python中的OpenCV和dlib库来实现人脸检测和抠图，以下是代码示例：

import cv2
import dlib

# 读取输入图片
img = cv2.imread("input_image.jpg")

# 创建人脸检测器
detector = dlib.get_frontal_face_detector()

# 检测人脸
faces = detector(img, 1)

# 循环遍历每个检测到的人脸
for face in faces:
    # 获取人脸位置坐标
    x1, y1, x2, y2 = face.left(), face.top(), face.right(), face.bottom()
    
    # 创建遮罩层
    mask = np.zeros(img.shape[:2], dtype=np.uint8)
    
    # 在遮罩层上绘制人脸轮廓
    landmarks = predictor(img, face)
    points = cv2.face_utils.shape_to_np(landmarks)
    convexhull = cv2.convexHull(points)
    cv2.drawContours(mask, [convexhull], -1, 255, -1)
    
    # 将遮罩层与输入图像进行按位与操作，实现抠图
    masked_image = cv2.bitwise_and(img, img, mask=mask)
    
    # 显示抠图结果
    cv2.imshow("Masked Image", masked_image)
    cv2.waitKey(0)

其中，dlib库用于人脸检测和关键点定位，可以通过pip安装，安装命令为：

pip install dlib

OpenCV库用于图像处理和显示，可以通过pip安装，安装命令为：

pip install opencv-python

以上代码示例假设已经提前训练好了dlib库中的人脸关键点检测器，可以使用shape_predictor_68_face_landmarks.dat文件进行关键点定位。如果您还没有该文件，可以从dlib官网下载：http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2