Python隐形水印invisible-watermark教程

隐形水印，顾名思义，在图片上打上肉眼不可见的水印，但是通过特定的解码方式可以查看水印标识。
invisible-watermark，一个Python库，实现对图片加上隐形水印，从而形成对图片的保护作用。

安装

pip install invisible-watermark -i https://pypi.douban.com/cimple/

加水印

准备待加水印图片cat.png¹

加水印脚本embed.py内容如下，水印标识为’test’

#!/usr/bin/python3

import cv2
from imwatermark import WatermarkEncoder

bgr = cv2.imread('cat.png')
wm = 'test'

encoder = WatermarkEncoder()
encoder.set_watermark('bytes',wm.encode('utf-8'))
bgr_encoded = encoder.encode(bgr, 'dwtDct')

cv2.imwrite('cat_wm.png',bgr_encoded)

执行embed脚本python embed.py得到加水印后的图片cat_wm.png²

获取水印标识脚本decode.py内容如下

#!/usr/bin/env python3

import cv2
from imwatermark import WatermarkDecoder

bgr = cv2.imread('cat_wm.png')

decoder = WatermarkDecoder('bytes', 32)
watermark = decoder.decode(bgr, 'dwtDct')
print(watermark.decode('utf-8'))

执行脚本python decode.py看见水印输出’test’³

当图片过小时会报错：

Traceback (most recent call last):
  File "decode.py", line 10, in <module>
    print(watermark.decode('utf-8'))
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xf7 in position 2: invalid start byte

官方对加水印后的图片进行多种测试，有jpg压缩、图片噪声(Noise)、图片色彩变换(Brightness)、图片涂鸦(Overlay)、图片马赛克(Mask)、图片裁剪(crop 7*5)、图片大小重置(Resize)、图片角度转换(Rotate)，并且说明了结果。详情见：https://github.com/ShieldMnt/invisible-watermark

图片的隐形水印保护，是一种特别的保护方式，如果ps过度，很难鉴定是否还是原图，这种情况另当别论。