用opencv给图片换背景色的示例代码

下面是使用OpenCV给图片换背景色的完整攻略：

步骤一：导入OpenCV库

首先需要通过pip安装opencv-python库，然后再通过以下代码导入：

import cv2
import numpy as np

步骤二：读取图片

使用OpenCV可以通过以下代码读取图片：

img = cv2.imread("img.jpg")

其中，“img.jpg”是要读取的图片文件路径。

步骤三：将背景色设为白色

为了将图片的背景色换成其他颜色，首先需要将背景色设为一种基础颜色，例如白色。可以使用以下代码将背景色设为白色：

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY)
mask = cv2.bitwise_not(thresh)
white_bg = np.full(img.shape, 255, dtype=np.uint8)
white_bg = cv2.bitwise_and(white_bg, white_bg, mask=mask)

上述代码中，首先将图片转为灰度图像（gray），然后将灰度图像进行二值化处理（thresh），将背景色转为黑色（0值）；接着通过对二值化图像进行反转（mask = cv2.bitwise_not(thresh)）将背景色转为白色（255值），最后用白色的背景图与原图进行“与”的位运算，得到新的白色背景图（white_bg）。

步骤四：将背景色设为其他颜色

将图片的背景色设为其他颜色，可以使用以下代码：

new_bg = np.full(img.shape, (0, 255, 0), dtype=np.uint8)
result = cv2.addWeighted(white_bg, 0.5, new_bg, 0.5, 0)

上述代码中，首先创建一个与原图大小相同的背景图（new_bg），将背景色设为绿色，然后通过cv2.addWeighted()函数进行加权叠加，得到新的结果图（result）。

示例说明一：将背景色设为红色

假设要将图片的背景色设为红色，可以更改上述代码中的颜色值，例如将new_bg的颜色值改为(0, 0, 255)，即：

new_bg = np.full(img.shape, (0, 0, 255), dtype=np.uint8)
result = cv2.addWeighted(white_bg, 0.5, new_bg, 0.5, 0)

示例说明二：将背景色设为蓝色

假设要将图片的背景色设为蓝色，可以更改上述代码中的颜色值，例如将new_bg的颜色值改为(255, 0, 0)，即：

new_bg = np.full(img.shape, (255, 0, 0), dtype=np.uint8)
result = cv2.addWeighted(white_bg, 0.5, new_bg, 0.5, 0)

至此，便完成了使用OpenCV给图片换背景色的攻略。