OpenCV实现LSB算法（数字水印）

本文参考自《计算机视觉40例从入门到深度学习（OpenCV-Python）》
LSB算法的原理就不在过多的介绍了，直接上代码。
lenacolor.png

watermark.bmp

# LSB算法
import numpy as np 
import cv2
lena = cv2.imread("D:\C\copy\lenacolor.png", 0)
#-------------嵌入
#s1 生成内部都是254的数组
r,c=lena.shape
t1=np.ones((r,c),np.uint8)*254
#s2 获取原始载体图像的高7位 最低有效位清0
lsb0=cv2.bitwise_and(lena,t1)
#s3 水印信息处理
w=cv2.imread("D:\C\copy\watermark.bmp",0)

#将水印图像内的255转换成1
wt=w.copy()
wt[w>0]=1
#s4 将水印图像wt嵌入lsb0内
wo=cv2.bitwise_or(lsb0,wt)
#--------------提取
#s1 生成内部都是1的数组
t2=np.ones((r,c),np.uint8)
#s2 从载体图像内提取水印图像
ewb=cv2.bitwise_and(wo,t2)
#s3 将水印图像内的数值1转换成255
ew=ewb.copy()
ew[ewb>0]=255
#--------------显示结果
cv2.imshow("lena",lena)
cv2.imshow("watermark",w)
cv2.imshow("wo",wo)
cv2.imshow("ew",ew)
cv2.waitKey()
cv2.destroyAllWindows()