Blind watermark based on wavelet transform.
- Documentation: https://BlindWatermark.github.io/blind_watermark/#/en/
- 文档： https://BlindWatermark.github.io/blind_watermark/#/zh/
- 中文 readme README_cn.md
- Source code: https://github.com/guofei9987/blind_watermark
pip install blind-watermark
For the current developer version:
git clone [email protected]:guofei9987/blind_watermark.git
cd blind_watermark
pip install .
How to use
How to embed watermark:
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')
How to extract watermark
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', )
demos:
| origin image | watermark |
|---|---|
 |
 |
| image embedded with watermark | extracted watermark |
|---|---|
 |
 |
Attack on the embedded image
| attack method | image after attack | extracted watermark |
|---|---|---|
| Rotate 45 Degrees 旋转攻击.py |
 |
 |
| Many Coverage 多遮挡攻击.py |
 |
 |
| 50% Horizontal Crop 横向裁剪攻击.py |
 |
 |
| 50% Vertical Crop 纵向裁剪攻击.py |
 |
 |
| Resize（1200X1920->600X800） 缩放攻击.py |
 |
 |
| Pepper Noise 椒盐击.py |
 |
 |
| Brightness 10% Up 亮度调高攻击.py |
 |
 |
| Brightness 10% Down 亮度调暗攻击.py |
 |
 |
embed string
Embed:
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))
Extract:
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 开源万岁！
embed array of bits
As demo, we embed 6 bytes data:
wm = [True, False, True, True, True, False]
Embed:
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/embedded.png')
Extract:
bwm1 = WaterMark(password_img=1, password_wm=1, wm_shape=6) wm_extract = bwm1.extract('output/打上水印的图.png', mode='bit') print(wm_extract)
Notice that wm_shape (shape of watermark) is necessary
The output wm_extract is an array of float. set a threshold such as 0.5.
WaterMark(..., processes=None)
processes: number of processes, can be integer. DefaultNone, meaning use all processes.