https://github.com/guofei9987/blind_watermark
Blind&Invisible Watermark ，图片盲水印，提取水印无须原图！
https://github.com/guofei9987/blind_watermark
blind-watermark image-processing watermark watermark-image
Last synced: 5 months ago
JSON representation
Blind&Invisible Watermark ，图片盲水印，提取水印无须原图！
- Host: GitHub
- URL: https://github.com/guofei9987/blind_watermark
- Owner: guofei9987
- License: mit
- Created: 2019-07-16T14:08:51.000Z (over 6 years ago)
- Default Branch: master
- Last Pushed: 2024-06-23T12:29:59.000Z (over 1 year ago)
- Last Synced: 2025-05-06T20:42:46.074Z (6 months ago)
- Topics: blind-watermark, image-processing, watermark, watermark-image
- Language: Python
- Homepage: https://blindwatermark.github.io/blind_watermark/#/en/
- Size: 4.35 MB
- Stars: 6,385
- Watchers: 38
- Forks: 739
- Open Issues: 30
-
Metadata Files:
- Readme: README.md
- Funding: .github/FUNDING.yml
- License: LICENSE
Awesome Lists containing this project
README
# blind-watermark
Blind watermark based on DWT-DCT-SVD.
[](https://pypi.org/project/blind_watermark/)
[](https://travis-ci.com/guofei9987/blind_watermark)
[](https://codecov.io/gh/guofei9987/blind_watermark)
[](https://github.com/guofei9987/blind_watermark/blob/master/LICENSE)
[](https://github.com/guofei9987/blind_watermark/)
[](https://github.com/guofei9987/blind_watermark/fork)
[](https://pepy.tech/project/blind-watermark)
[](https://github.com/guofei9987/blind_watermark/discussions)
- **Documentation:** [https://BlindWatermark.github.io/blind_watermark/#/en/](https://BlindWatermark.github.io/blind_watermark/#/en/)
- **文档：** [https://BlindWatermark.github.io/blind_watermark/#/zh/](https://BlindWatermark.github.io/blind_watermark/#/zh/)
- **中文 readme** [README_cn.md](README_cn.md)
- **Source code:** [https://github.com/guofei9987/blind_watermark](https://github.com/guofei9987/blind_watermark)
# install
```bash
pip install blind-watermark
```
For the current developer version:
```bach
git clone git@github.com:guofei9987/blind_watermark.git
cd blind_watermark
pip install .
```
# How to use
## Use in bash
```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
 + '@guofei9987 开源万岁！' = 
See the [codes](/examples/example_str.py)
Embed watermark:
```python
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:
```python
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:
```python
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:
```python
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](/examples/example_bit.py)
As demo, we embed 6 bytes data:
```python
wm = [True, False, True, True, True, False]
```
Embed:
```python
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:
```python
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
```python
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](https://github.com/guofei9987/text_blind_watermark)
- HideInfo（hide as image, hide as sounds, hide as text）：[https://github.com/guofei9987/HideInfo](https://github.com/guofei9987/HideInfo)