基于小波变换的数字盲水印
- Documentation: https://BlindWatermark.github.io/blind_watermark/#/en/
- 文档: https://BlindWatermark.github.io/blind_watermark/#/zh/
- English readme README.md
- Source code: https://github.com/guofei9987/blind_watermark
pip install blind-watermark
或者安装最新开发版本
git clone [email protected]: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/ori_img.jpg')
# 读取水印
bwm1.read_wm('pic/watermark.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度 旋转攻击.py |
||
多遮挡 多遮挡攻击.py |
||
横向裁剪50% 横向裁剪攻击.py |
||
纵向裁剪50% 纵向裁剪攻击.py |
||
缩放攻击(1200X1920->600X800) 缩放攻击.py |
||
椒盐攻击 椒盐击.py |
||
亮度提高10% 亮度调高攻击.py |
||
亮度调低10% 亮度调暗攻击.py |
嵌入:
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)
Output:
@guofei9987 开源万岁!
作为 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/ori_img.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之间的实数,方便用户自行卡阈值。如果水印信息量远小于图片可容纳量,偏差极小。
WaterMark(..., processes=None)
processes
: 整数,指定线程数。默认为None
, 表示使用全部线程。