OpenCV的简单抠图尝试（白色背景的图片）

使用场景

��       一键抠图，但是，仅限于大面积背景的物品图片，本脚本主要是针对近白色背景的图片。

（）

需要用到的库

import cv2import numpy as npfrom PIL import Imageimport DwkCvShow

先定义背景的颜色范围（注意是HSV色彩空间的范围）

# 定义白色在HSV中的范围# H（Hue）表示色调，范围是0-179；# S（Saturation）表示饱和度，范围是0-255；# V（Value）表示亮度，范围是0-255。# lower_white = np.array([0, 0, 168])lower_white = np.array([0, 0, 150])upper_white = np.array([172, 111, 255])

读取图片并且转换成HSV颜色空间

# 读取图片img = cv2.imread('./input/002.jpg')# 将BGR图像转换为HSV图像hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)print("转换成hsv色彩空间")DwkCvShow.resize_and_show_image(hsv)

创建白色掩码并且降噪处理

# 创建一个白色的掩码mask = cv2.inRange(hsv, lower_white, upper_white)# 定义结构元素kernel = np.ones((5, 5), np.uint8)print("创建一个白色掩码")DwkCvShow.resize_and_show_image(mask)# 先腐蚀eroded = cv2.erode(mask, kernel, iterations=2)print("创建一个白色掩码先腐蚀")DwkCvShow.resize_and_show_image(eroded)# 再膨胀dilated = cv2.dilate(eroded, kernel, iterations=2)print("创建一个白色掩码先腐蚀再膨胀")DwkCvShow.resize_and_show_image(dilated)

反转掩码并且降噪处理

# 反转掩码：将白色变为黑色，将黑色变为白色mask_inv = cv2.bitwise_not(dilated)print("反转掩码")DwkCvShow.resize_and_show_image(mask_inv)# 定义结构元素kernel = np.ones((5, 5), np.uint8)# 先腐蚀eroded = cv2.erode(mask_inv, kernel, iterations=2)print("反转掩码后先腐蚀")DwkCvShow.resize_and_show_image(eroded)# 再膨胀dilated = cv2.dilate(eroded, kernel, iterations=2)print("反转掩码后先腐蚀再膨胀")DwkCvShow.resize_and_show_image(dilated)

绘制产品物品的最大轮廓

# 最终二值图# 找到所有轮廓contours, hierarchy = cv2.findContours(dilated, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)# 找到最大的轮廓max_contour = max(contours, key=cv2.contourArea)# # 使用Douglas-Peucker算法来平滑轮廓# epsilon = 0.005*cv2.arcLength(max_contour, True)# approx = cv2.approxPolyDP(max_contour, epsilon, True)# 创建一个全黑的背景，大小和原图像一样background = np.zeros_like(dilated)# 在黑色背景上绘制白色的最大轮廓# cv2.drawContours(background, [approx], -1, (255), thickness=cv2.FILLED)cv2.drawContours(background, [max_contour], -1, (255), thickness=cv2.FILLED)DwkCvShow.resize_and_show_image(background)

抠图并且保存图片

# 使用反转的掩码移除背景res = cv2.bitwise_and(img, img, mask=background)print("白色的掩码反转后移除背景")print("准备写入新的图像")DwkCvShow.resize_and_show_image(res)# 将OpenCV图像转换为PIL图像res = cv2.cvtColor(res, cv2.COLOR_BGR2RGB)pil_img = Image.fromarray(res)# 创建一个新的透明背景图像transparent_img = Image.new("RGBA", pil_img.size)for x in range(pil_img.width):    for y in range(pil_img.height):        r, g, b = pil_img.getpixel((x, y))        if (r, g, b) != (0, 0, 0):            transparent_img.putpixel((x, y), (r, g, b, 255))# 保存透明背景图像transparent_img.save('颜色掩码方法+轮廓绘制.png', 'PNG')

总结

        简单的算法是很难实现平滑抠图的，需要用到高等数学。大概的方向是计算曲线前面部分导数（当然不仅仅是导数，总之跟曲线的末端切线方向有关）。

（）