本期推荐的是一个添加、提取图片盲水印的Python工具——Blind Watermark。

blind_watermark是一个可以添加、提取图片盲水印的Python工具，支持添加数字、嵌入图片、嵌入文本、嵌入二进制四种方式。可以防止旋转角度、随机截图、多遮挡、纵向裁剪、横向裁剪、缩放攻击等效果。

安装

pip install blind-watermark

或者安装最新开发版本

git clone git@github.com:guofei9987/blind_watermark.git
cd blind_watermark
pip install .

使用

命令行中使用

# 嵌入水印：
blind_watermark --embed --pwd 1234 examples/pic/ori_img.jpeg "watermark text" examples/output/embedded.png

# 提取水印：
blind_watermark --extract --pwd 1234 --wm_shape 111 examples/output/embedded.png

Python中使用

• 嵌入文字水印

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_wm=1, password_img=1)
# read original image
bwm1.read_img('pic/ori_img.jpg')
# read watermark
bwm1.read_wm('pic/watermark.png')
# embed
bwm1.embed('output/embedded.png')

• 提取图片水印

bwm1 = WaterMark(password_wm=1, password_img=1)
# notice that wm_shape is necessary
bwm1.extract(filename='output/embedded.png', wm_shape=(128, 128), out_wm_name='output/extracted.png', )

各种攻击后的效果

此项目使用MIT开源协议，更多内容大家可自行前往阅读。