(摘) Python 实现数字图片盲水印(隐水印)

声明:内容源自网络,版权归原作者所有。若有侵权请联系我们

开源代码: https://github.com/guofei9987/blind_watermark

安装: pip install blind-watermark

作者的文档: https://github.com/guofei9987/blind_watermark/blob/master/README_cn.md

原图:

加过水印:

看起来效果不错,提交图片的时候可以用这个。

使用

它需要安装pywt: pip install PyWavelets

嵌入水印

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', )

#### 嵌入字符

嵌入:

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)

隐水印还可以是二进制数据

嵌入水印:

# 除了嵌入图片,也可以嵌入比特类数据
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')

解水印:

bwm1 = WaterMark(password_img=1, password_wm=1, wm_shape=6)
wm_extract = bwm1.extract('output/打上水印的图.png', mode='bit')
print(wm_extract)

解出的水印是一个0~1之间的实数,方便用户自行卡阈值。


使用感觉速度有点慢,画质有损伤。