OpenCV-图像的混合

参考链接

将一幅logo图片添加到一张图片中去

import numpy as np
import cv2

img = cv2.imread('logo.png') # 读取logo图片
img2 = cv2.imread('a.JPEG')
# cv2.imshow('img', img)
rows, cols, channels = img.shape    # logo的大小
roi = img2[0: rows, 0:cols] # 在图片a中获取同样的大小

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)    # 转为灰度图
ret, mask = cv2.threshold(gray, 230, 255, cv2.THRESH_BINARY) # mask为白底，有颜色部分为黑色
# cv2.imshow('gray', gray)
cv2.imshow('mask', mask)
mask_inv = cv2.bitwise_not(mask) # 非运算，颜色反转
cv2.imshow('mask_inv', mask_inv)

# 与运算：
# 黑色是0，白色是1
# 任何乘0都是0，任何乘1都是本身
# 所以与黑色相与是黑色，与白色相与是本身

# img_1在图片位置中把logo位置变成黑色，防止add时颜色的相加
# img_2把logo白底变成黑底
# 然后二者相加
img1_bg = cv2.bitwise_and(roi, roi, mask = mask) # roi与mask做与运算，颜色相加
cv2.imshow('img1_bg', img1_bg)
img2_fg = cv2.bitwise_and(img, img, mask = mask_inv) # img与mask_inv做与运算
cv2.imshow('im2_fg', img2_fg)
det = cv2.add(img1_bg, img2_fg)

img2[0:rows, 0:cols] = det
cv2.imshow('img2', img2)
cv2.waitKey(0)