使用OpenCV给图片添加Logo

import cv2
import numpy as np
imgcat = cv2.imread("d:/img/cat.jpg")

imglogo = cv2.imread("d:/img/logo.jpg")

imglogo = cv2.resize(imglogo,(0,0),fx=0.4,fy=0.4)

width,height = imglogo.shape[0:2]

# for w in range(0,width):
#     for h in range(0,height):
#         is255 = True
#         for d in range(0,3):
#             if(imglogo[w,h,d]!=255):
#                 is255=False
#         if is255 ==True:
#             for d in range(0, 3):
#                 imglogo[w,h,d]=0

# imgcat[0:width,0:height]=imgcat[0:width,0:height] + imglogo
#
# print(imgcat[0,0])

imglogotr = cv2.cvtColor(imglogo,cv2.COLOR_BGR2GRAY)

ret,imglogoms = cv2.threshold(imglogotr,170,255,cv2.THRESH_BINARY_INV)

imglogotr = cv2.add(imglogo,imglogo,mask=imglogoms)

imglogoms = cv2.bitwise_not(imglogoms)

imgcattr = imgcat[0:width,0:height]

imgcatzero = np.zeros((width,height,3),dtype=np.uint8)

imgcattr = cv2.add(imgcattr,imgcatzero,mask=imglogoms)

imgadd = cv2.add(imgcattr,imglogotr)

imgcat[0:width,0:height] = imgadd

cv2.imshow('cat',imgcat)

cv2.waitKey(0)