Как правильно наложить два изображения с opencv

Question

Я пытаюсь наложить прозрачный PNG поверх изображения JPG, я почему-то не могу заставить его работать, что я пытался:

import cv2
import numpy as np

a = cv2.imread("a.jpeg")
b = cv2.imread("b.png", cv2.IMREAD_UNCHANGED)

# add alpha channel to jpeg
(h, w) = a.shape[:2]
a = np.dstack([a, np.ones((h, w), dtype="uint8") * 0])

c = cv2.add(b,a)
cv2.imwrite("out.jpeg", c)

Следующий код объединит два изображения, но прозрачность PNG неверна, она более непрозрачна, чем должна быть. (Я читал, что это может быть проблема с opencv? Неправильно читается альфа из png)

То, что я пытаюсь сделать, это просто сложить два изображения поверх другого, фоновый JPG и поместить поверх PNG, который имеет несколько прозрачных зон, оба изображения имеют одинаковый размер.

Спасибо!

Answer 1

Я нашел пример на github, который немного изменил, работает как положено:

import numpy as np
import cv2

img = cv2.imread('1_calib.jpeg')
overlay_t = cv2.imread('ol.png',-1) # -1 loads with transparency

def overlay_transparent(bg_img, img_to_overlay_t):
    # Extract the alpha mask of the RGBA image, convert to RGB 
    b,g,r,a = cv2.split(img_to_overlay_t)
    overlay_color = cv2.merge((b,g,r))

    mask = cv2.medianBlur(a,5)

    # Black-out the area behind the logo in our original ROI
    img1_bg = cv2.bitwise_and(bg_img.copy(),bg_img.copy(),mask = cv2.bitwise_not(mask))

    # Mask out the logo from the logo image.
    img2_fg = cv2.bitwise_and(overlay_color,overlay_color,mask = mask)

    # Update the original image with our new ROI
    bg_img = cv2.add(img1_bg, img2_fg)

    return bg_img

cv2.imshow('image',overlay_transparent(img, overlay_t))
cv2.waitKey(0)