python OpenCV - thêm kênh alpha vào hình ảnh RGB

Question

Cách tốt nhất để chuyển đổi hình ảnh RGB thành RGBA trong python bằng cách sử dụng opencv là gì?

Hãy nói rằng tôi có một mảng với hình dạng

(185, 198, 3) - it is RGB 

và khác là alpha mặt nạ với hình dạng (185, 198)

Làm thế nào để hợp nhất chúng lại và lưu vào tập tin?

Answer 1

Bạn có thể sử dụng cv2.merge() để thêm kênh alpha để hình ảnh RGB nhất định, nhưng trước tiên bạn cần để phân chia các hình ảnh RGB để R, G and B kênh, theo documentation:

Python: cv2.merge (mv [, dst])

• mv - mảng đầu vào hoặc vectơ của ma trận được hợp nhất; tất cả các ma trận trong mv phải có cùng kích thước và cùng độ sâu.

Và điều này có thể được thực hiện như:

b_channel, g_channel, r_channel = cv2.split(img) 

alpha_channel = np.ones(b_channel.shape, dtype=b_channel.dtype) * 50 #creating a dummy alpha channel image. 

img_BGRA = cv2.merge((b_channel, g_channel, r_channel, alpha_channel)) 

Answer 2

Dưới đây là một ví dụ đơn giản khác sử dụng Grabcut, nó giúp để có được thứ tự đúng của các kênh khi lưu hình ảnh trên đĩa vs pyplot.

from matplotlib import pyplot as plt 
import numpy as np 
import cv2 

img = cv2.imread('image.jpg') 

mask = np.zeros(img.shape[:2], np.uint8) 
bgdModel = np.zeros((1,65), np.float64) 
fgdModel = np.zeros((1,65), np.float64) 
rect = (50, 50, 450, 290) 

# Grabcut 
cv2.grabCut(img, mask, rect, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_RECT) 

r_channel, g_channel, b_channel = cv2.split(img) 
a_channel = np.where((mask==2)|(mask==0), 0, 255).astype('uint8') 

img_RGBA = cv2.merge((r_channel, g_channel, b_channel, a_channel)) 
cv2.imwrite("test.png", img_RGBA) 

# Now for plot correct colors : 
img_BGRA = cv2.merge((b_channel, g_channel, r_channel, a_channel)) 

plt.imshow(img_BGRA), plt.colorbar(),plt.show()