效果图
未加水印前:
水印掩码:
透明度为 0.3 的水印图片:
透明度为 1.0 的水印图片:
我自己编写的这段代码,其效果似乎比opencv官网上的示例更好,因为水印中没有夹带黑色噪点。
代码语言:JavaScript 代码运行次数:0
# coding=utf-8 import cv2 import numpy as np <p>bottom = cv2.imread('elegent.jpg') mark = cv2.imread('mask.jpg')</p><h1>加水印</h1><p>def waterMark(src, mask, transparency):</p><h1>对掩码:转为float型 --> 乘上透明度 --> 地板整数 --> 回到uint8型</h1><pre class="brush:php;toolbar:false">mask = np.floor(mask.astype(np.float32) * transparency).astype(np.uint8) dst = cv2.add(src, mask) # 防止溢出像素值域 dst = dst * (dst <= 255) + 255 * (dst > 255) # 回到uint8型 dst = dst.astype(np.uint8) return dst
b_rows, b_cols, b_channels = bottom.shape m_rows, m_cols, m_channels = mark.shape
将图片右下角矩形块 替换为 加过水印的矩形块
bottom[b_rows-m_rows:b_rows, b_cols-m_cols:b_cols, ] = waterMark(bottom[b_rows-m_rows:b_rows, b_cols-m_cols:b_cols, ], mark, 0.3)
保存水印图片
cv2.imwrite(‘waterMark.jpg’, bottom) cv2.imshow(‘waterMark’, bottom) cv2.waitKey(3000) cv2.destroyAllWindows()
对比OpenCV官网上的类似例程:
具体代码:
代码语言:JavaScript 代码运行次数:0
# coding=utf-8 import cv2</p><p>img1 = cv2.imread('elegent.jpg') img2 = cv2.imread('mask.jpg')</p><p>b_rows, b_cols, b_channels = img1.shape m_rows, m_cols, m_channels = img2.shape</p><h1>I want to put logo on top-left corner, So I create a ROI</h1><p>rows,cols,channels = img2.shape roi = img1[b_rows-m_rows:b_rows, b_cols-m_cols:b_cols, ]</p><h1>Now create a mask of logo and create its inverse mask also</h1><p>img2gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY) ret, mask = cv2.threshold(img2gray, 10, 255, cv2.THRESH_BINARY) mask_inv = cv2.bitwise_not(mask)</p><h1>Now black-out the area of logo in ROI</h1><p>img1_bg = cv2.bitwise_and(roi,roi,mask = mask_inv)</p><h1>Take only region of logo from logo image.</h1><p>img2_fg = cv2.bitwise_and(img2,img2,mask = mask)</p><h1>Put logo in ROI and modify the main image</h1><p>dst = cv2.add(img1_bg,img2_fg)</p><h1>img1[0:rows, 0:cols ] = dst</h1><h1>将图片右下角矩形块 替换为 加过水印的矩形块</h1><p>img1[b_rows-m_rows:b_rows, b_cols-m_cols:b_cols, ] = dst</p><p>cv2.imwrite('res.jpg', img1) cv2.imshow('res',img1) cv2.waitKey(0) cv2.destroyAllWindows()
该例程的实现效果:
劣势:通过细节放大图可以看出,OpenCV官网例程的效果较差,不仅无法自定义水印透明度,而且水印中还会夹带黑色噪点,影响美观: