python ビット演算による画像の合成

Posted 2021-05-09

import cv2
import numpy as np


img1 = cv2.imread('bird.jpg')
img2 = cv2.imread('opencv_logo.jpg')

# ロゴをimg1の左上に置きたいためROIを定義
rows, cols, channels = img2.shape
roi = img1[0:rows, 0:cols]

# マスクを作成
img2gray = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)

# しきい値10より大きいピクセルは255でそれ以外は0はになる
ret, mask = cv2.threshold(img2gray, 10, 255, cv2.THRESH_BINARY)
# 白と黒をひっくり返す
# ロゴの色がついているところが黒（0）になる
mask_inv = cv2.bitwise_not(mask)

# ROIのロゴ部分を黒で塗りつぶす
img1_bg = cv2.bitwise_and(roi, roi, mask=mask_inv)

# ロゴの模様以外を塗りつぶす
img2_fg = cv2.bitwise_and(img2, img2, mask=mask)

# ROIの部分を合成
dst = cv2.add(img1_bg, img2_fg)

# 画像全体でROIを置き換え
img1[0:rows, 0:cols] = dst

cv2.imshow('result', img1)
cv2.waitKey(0)
cv2.destroyAllwindows()