python隐藏水印_Python 实现数字图片盲水印（隐水印）

pip install blind-watermark

For the current developer version:

git clone git@github.com:guofei9987/blind_watermark.git

cd blind_watermark

pip install .

嵌入水印

from blind_watermark import WaterMark

bwm1 = WaterMark(password_wm=1, password_img=1)

# 读取原图

bwm1.read_img('pic/原图.jpg')

# 读取水印

bwm1.read_wm('pic/水印.png')

# 打上盲水印

bwm1.embed('output/打上水印的图.png')

提取水印

bwm1 = WaterMark(password_wm=1, password_img=1)

# 注意需要设定水印的长宽wm_shape

bwm1.extract(filename='output/打上水印的图.png', wm_shape=(128, 128), out_wm_name='output/解出的水印.png', )

原图：

水印：

打上隐水印的图：

从“打上隐水印的图”中提取的水印

旋转攻击45度

提取水印：

多遮挡

提取水印：

横向裁剪50%：

提取水印：

纵向裁剪50%

提取水印：

缩放攻击(1200X1920->600X800)

提取水印：

椒盐攻击

提取水印：

亮度提高10%：

提取水印：

亮度调低10%

提取水印：

嵌入字符

嵌入：

from blind_watermark import WaterMark

bwm1 = WaterMark(password_img=1, password_wm=1)

bwm1.read_img('pic/ori_img.jpg')

wm = '@guofei9987 开源万岁！'

bwm1.read_wm(wm, mode='str')

bwm1.embed('output/embedded.png')

len_wm = len(bwm1.wm_bit)

print('Put down the length of wm_bit{len_wm}'.format(len_wm=len_wm))

提取水印：

bwm1 = WaterMark(password_img=1, password_wm=1)

wm_extract = bwm1.extract('output/embedded.png', wm_shape=len_wm, mode='str')

print(wm_extract)@guofei987 开源万岁！

作为 demo， 如果要嵌入是如下长度为6的二进制数据

wm = [True, False, True, True, True, False]

嵌入水印

# 除了嵌入图片，也可以嵌入比特类数据

from blind_watermark import WaterMark

bwm1 = WaterMark(password_img=1, password_wm=1)

bwm1.read_ori_img('pic/原图.jpg')

bwm1.read_wm([True, False, True, True, True, False], mode='bit')

bwm1.embed('output/打上水印的图.png')

解水印：(注意设定水印形状 wm_shape)

bwm1 = WaterMark(password_img=1, password_wm=1, wm_shape=6)

wm_extract = bwm1.extract('output/打上水印的图.png', mode='bit')

print(wm_extract)

解出的水印是一个0～1之间的实数，方便用户自行卡阈值。如果水印信息量远小于图片可容纳量，实测偏差极小。