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.

    项目简介

    Blind/Invisible Watermark (图片盲水印,提取水印无须原图!)

    🚀 Github 镜像仓库 🚀

    源项目地址

    https://github.com/guofei9987/blind_watermark

    发行版本 4

    version 0.2.1

    全部发行版

    贡献者 3

    guofei9987 @guofei99871
    G guofei9987 @guofei9987
    C chy31694 @chy31694

    开发语言

    • Python 100.0 %