13 Star 132 Fork 24

guofei9987 / blind_watermark

Create your Gitee Account
Explore and code with more than 8 million developers,Free private repositories !:)
Sign up
Clone or Download
Cancel
Notice: Creating folder will generate an empty file .keep, because not support in Git
Loading...
README.md

blind-watermark

Blind watermark based on dct and svd.

PyPI Build Status codecov License Python Platform stars fork Downloads Discussions

install

pip install blind-watermark

For the current developer version:

git clone git@github.com: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.png

Use in Python

Original Image + Watermark = Watermarked Image

origin_image + '@guofei9987 开源万岁!' = 打上水印的图

See the codes

Embed watermark:

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

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 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.

Concurrency

WaterMark(..., processes=None)
  • processes: number of processes, can be integer. Default None, meaning use all processes.

Related Project

text_blind_watermark: https://github.com/guofei9987/text_blind_watermark
Embed message into text.

Repository Comments ( 11 )

Sign in to post a comment

About

Cancel

Releases

No release

blind_watermark

Contributors

All

Activities

Load More
can not load any more
Python
1
https://gitee.com/guofei9987/blind_watermark.git
git@gitee.com:guofei9987/blind_watermark.git
guofei9987
blind_watermark
blind_watermark
master

Search

101014 b92fc32e 1850385 101014 af024cb7 1850385