Repository: guofei9987/blind_watermark
Stars: 12315
README.md
blind-watermark
Blind watermark based on DWT-DCT-SVD.
!Python
!Platform
<a href="https://hellogithub.com/repository/guofei9987/blind_watermark" target="_blank"><img src="https://abroad.hellogithub.com/v1/widgets/recommend.svg?rid=3834302ff46a40f188a651ef8bd26ff5&claim_uid=se0WHo8cbiLv2w1&theme=small" alt="Featured｜HelloGitHub" /></a>
- 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
install
pip install blind-watermarkFor the current developer version:
git clone [email protected]:guofei9987/blind_watermark.git
cd blind_watermark
pip install .How to use
Use in bash
embed watermark into image:
blind_watermark --embed --pwd 1234 examples/pic/ori_img.jpeg "watermark text" examples/output/embedded.png
extract watermark from image:
blind_watermark --extract --pwd 1234 --wm_shape 111 examples/output/embedded.pngUse in Python
Original Image + Watermark = Watermarked Image
!origin_image + '@guofei9987 开源万岁！' = !打上水印的图
See the codes
Embed watermark:
from blind_watermark import WaterMarkbwm1 = 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 watermark:
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 开源万岁！
attacks on Watermarked Image
|attack method|image after attack|extracted watermark|
|--|--|--|
|Rotate 45 Degrees|!旋转攻击|'@guofei9987 开源万岁！'|
|Random crop|!截屏攻击|'@guofei9987 开源万岁！'|
|Masks| !多遮挡攻击 |'@guofei9987 开源万岁！'|
|Vertical cut|!横向裁剪攻击|'@guofei9987 开源万岁！'|
|Horizontal cut|!纵向裁剪攻击|'@guofei9987 开源万岁！'|
|Resize|!缩放攻击|'@guofei9987 开源万岁！'|
|Pepper Noise|!椒盐攻击|'@guofei9987 开源万岁！'|
|Brightness 10% Down|!亮度攻击|'@guofei9987 开源万岁！'|
embed images
embed watermark:
from blind_watermark import WaterMarkbwm1 = 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')
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', )
|attack method|image after attack|extracted watermark|
|--|--|--|
|Rotate 45 Degrees|!旋转攻击||
|Random crop|!截屏攻击|!多遮挡_提取水印|
|Mask| !多遮挡攻击 |!多遮挡_提取水印|
embed array of bits
See it here
As demo, we embed 6 bytes data:
wm = [True, False, True, True, True, False]Embed:
from blind_watermark import WaterMarkbwm1 = 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 necessaryThe output wm_extract is an array of float. set a threshold such as 0.5.
Concurrency
WaterMark(..., processes=None)-
processes number of processes, can be integer. Default None, which means using all processes. Related Project
- text_blind_watermark (Embed message into text): https://github.com/guofei9987/text_blind_watermark
- HideInfo（hide as image, hide as sounds, hide as text）：https://github.com/guofei9987/HideInfo