下面是使用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给图片换背景色的攻略。
本站部分内容来源互联网,如果有图片或者内容侵犯了您的权益,请联系我们,我们会在确认后第一时间进行删除!