python利用opencv去除水印方法

OpenCV（Open Source Computer Vision Library）是一个跨平台计算机视觉库，实现了图像处理和计算机视觉方面的很多通用算法

在python中可以利用opencv来去除水印

opencv安装

window10

OpenCV 3：http://www.lfd.uci.edu/~gohlke/pythonlibs/#opencv

Numpy：http://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy

将下载好的whl文件放置到一个方便的目录，打开cmd，安装

pip install xxx

测试

import cv2

没有报错则说明成功

参考自https://www.ywlib.com/archives/39.html

下面开始去除水印， 使用58图片做测试

基于OpenCV的两种去水印方案

1. 基于 inpaint 方法（网上的方法，处理质量较低）

• 算法理论：基于Telea在2004年提出的基于快速行进的修复算法（FMM算法），先处理待修复区域边缘上的像素点，然后层层向内推进，直到修复完所有的像素点
• 处理方式：由ui人员制作出黑底白色水印且相同位置的水印蒙版图(必须单通道灰度图)，然后使用inpaint方法处理原始图像，具体使用时可把水印区放粗，这样处理效果会好点

需要带水印的图，和该图一样大小的的黑底白色水印且位置相同的水印图

    test_dir = 'xxx'
    mask_dir = 'xxx'
    save_dir ='xxx'
    src = cv2.imread(rental_dir)
    mask = cv2.imread(wm_dir, cv2.IMREAD_GRAYSCALE)
    dst = cv2.inpaint(src, mask, 3, cv2.INPAINT_TELEA)
    cv2.imwrite(save_dir, dst)

2. 基于像素的反色中和（处理质量较高）

参考自ps去水印原理，通过一张白底的反色水印图来中和原图水印

 需要到水印的图，和该图一样大小的白底棕色(128,128,128)的水印且位置相同的水印图，128的效果比较好

    test_dir = 'xxx'
    mask_dir = 'xxx'
    save_dir ='xxx'
    src = cv2.imread(test_dir)
    mask = cv2.imread(mask_dir)
    save = numpy.zeros(src.shape, numpy.uint8)
    for row in range(src.shape[0]):
        for col in range(src.shape[1]):
            for channel in range(src.shape[2]):
                if mask[row, col, channel] == 0:
                    val = 0
                else:
                    reverse_val = 255 - src[row, col, channel]
                    val = 255 - reverse_val * 256 / mask[row, col, channel]
                    if val < 0:
                        val = 0
                save[row, col, channel] = val
    cv2.imwrite(save_dir, save)

第二种方面明显比第一种要好，但是水印的边角处理不好，可以先用二在用一，效果会更好

参考自https://my.oschina.net/u/2400083/blog/732321