python - 将 Alpha 通道添加到 RGB 图像

Question

如何将 alpha 通道添加到 png 图像？正是我有使用grabcut功能的背景，现在我想将它添加到png图像中。

之后，我想通过使用蒙版将 Alpha 通道设置为透明

img[np.all(img == [0, 0, 0, 255], axis=2)] = [0, 0, 0, 0]

标签： pythonimageopencvimage-processingcomputer-vision

Answer 1

这是填充通道方法。

import cv2 

# path to your own images
rgb = cv2.imread('images/5v63T_RGB.png')
alpha = cv2.imread('images/5v63T_Alpha.png')
 
rgba = cv2.cvtColor(rgb, cv2.COLOR_RGB2RGBA)
one_channel_alpha = cv2.cvtColor(alpha, cv2.COLOR_RGB2GRAY)
 
# fill alpha channel with image (needs to be same size)
rgba[:, :, 3] = one_channel_alpha

cv2.imshow("rgb", rgb)
cv2.imshow("alpha", one_channel_alpha)
cv2.imshow("rgba", rgba)
# you can't visualize alpha channel as transparency.
# the 32 bits PNG image, so let's save and use other view. 
cv2.imwrite("images/result_rgba.png", rgba)

cv2.waitKey(0)
cv2.destroyAllWindows()

你会得到这样的结果：

祝你好运。