OpenCV Error:(mtype == CV_8U || mtype == CV_8S) && _mask.sameSize(*psrc1) in function ‘binary_op‘

程序如下：

import cv2
import numpy as np

# 按位运算

# Load two images
img1 = cv2.imread('../data/messi5.jpg')
img2 = cv2.imread('../data/opencv_logo.png')

# I want to put logo on top-left corner, So I create a ROI
rows, cols, channels = img2.shape
roi = img1[0:rows, 0:cols]

# Now create a mask of logo and create its inverse mask also
img2gray = cv2.cvtColor(img2, cv2.COLOR_BGR2GRAY)
ret, mask = cv2.threshold(img2gray, 10, 255, cv2.THRESH_BINARY)
mask_inv = cv2.bitwise_not(mask)

# Now black-out the area of logo in ROI
img1_bg = cv2.bitwise_and(roi, roi, mask=mask_inv)  # opencv 3.0
#TODO error

# Take only region of logo from logo image.
img2_fg = cv2.bitwise_and(img2, img2, mask=mask)

# Put logo in ROI and modify the main image
dst = cv2.add(img1_bg, img2_fg)
img1[0:rows, 0:cols] = dst

cv2.imshow('res', img1)
cv2.waitKey(0)
cv2.destroyAllWindows()

运行之后然后报错;

cv2.error: OpenCV(4.4.0) /tmp/pip-req-build-vu_aq9yd/opencv/modules/core/src/arithm.cpp:250: error: (-215:Assertion failed) (mtype == CV_8U || mtype == CV_8S) && _mask.sameSize(*psrc1) in function ‘binary_op’

找到的原因是两张图片的大小不一样，导致出现错误。两张图片的shape如下：
(342, 548, 3)
(555, 599, 3)

解决的方法是将两张图片的大小resize为一样大小。

img1 = cv2.resize(img1, (400, 400), interpolation=cv2.INTER_CUBIC)
img2 = cv2.resize(img2, (400, 400), interpolation=cv2.INTER_CUBIC)

参考：
https://www.cnblogs.com/qq1261275789/p/13052554.html