网上好多都是直接用写txt的方式实现的,其实实际中美工会提供水印图片,让添加进去。用cv2读取水印图片的时候,后面的参数设置为-1即可读出透明度通道,然后在原图的指定位置将水印图片和原图像的像素进行融合即可。
import cv2
import numpy as npdef watermark(src_path, mask_path, alpha = 0.3):img = cv2.imread(src_path)h,w = img.shape[0], img.shape[1]mask = cv2.imread(mask_path, -1)if w > h:rate = int(w * 0.1) / mask.shape[1]else:rate = int(h * 0.1) / mask.shape[0]mask = cv2.resize(mask, None, fx=rate, fy=rate)mask_h, mask_w = mask.shape[0], mask.shape[1]mask_channels = cv2.split(mask)dst_channels = cv2.split(img)b, g, r, a = cv2.split(mask)ul_points = (int(h * 0.9), int(int(w/2) - mask_w / 2))dr_points = (int(h * 0.9) + mask_h, int(int(w/2) + mask_w / 2))for i in range(3):dst_channels[i][ul_points[0] : dr_points[0], ul_points[1] : dr_points[1]] = dst_channels[i][ul_points[0] : dr_points[0], ul_points[1] : dr_points[1]] * (255.0 - a * alpha) / 255dst_channels[i][ul_points[0]: dr_points[0], ul_points[1]: dr_points[1]] += np.array(mask_channels[i] * (a * alpha / 255), dtype=np.uint8)dst_img = cv2.merge(dst_channels)return dst_img
原图
水印图:
合成之后的图,分别选择alpha=1 和 alpha=0.3