python +opencv去除水印

1.工具

python 3.7   opencv4.1

2.思路

1）获取图片水印的三原色，利用opencv 的inRange 二值化处理 （获取水印三原色范围）

2）opencv  的dilate  进行膨胀处理 （将水印三原色覆盖掉）

3）opencv   的inpaint 进行修复

3.代码

# 图片修复

import cv2

import numpyas np

import os

import time

def cv_imread(filePath):##  处理cv2 读取中文文件问题  ##

    cv_img=cv2.imdecode(np.fromfile(filePath,dtype=np.uint8),-1)

cv_img=cv2.cvtColor(cv_img,cv2.IMREAD_COLOR)

return cv_img

def test(path,file):

img = cv_imread(path)

hight, width, depth = img.shape[0:3]

# 图片二值化处理，把[240, 240, 240]~[255, 255, 255]以外的颜色变成0

    thresh = cv2.inRange(img, np.array([240, 240, 240]), np.array([255, 255, 255]))

# 创建形状和尺寸的结构元素

    kernel = np.ones((3, 3), np.uint8)

# 扩张待修复区域

    hi_mask = cv2.dilate(thresh, kernel, iterations=1)

specular = cv2.inpaint(img, hi_mask, 5, flags=cv2.INPAINT_TELEA)

cv2.namedWindow("Image", 0)

cv2.resizeWindow("Image", int(width /2), int(hight /2))

cv2.imshow("Image", img)

cv2.namedWindow("newImage", 0)

cv2.resizeWindow("newImage", int(width /2), int(hight /2))

cv2.imshow("newImage", specular)

#cv2.imwrite("C:/Users/xxx/Desktop/aa/"+file,specular)

    cv2.imencode('.jpg', specular)[1].tofile("C:/Users/xxx/Desktop/aa/"+file)####将去水印图片放入指定文件夹#####

#cv2.waitKey(0)

#cv2.destroyAllWindows()

    return

path ="C:/Users/xxxx/Desktop/dd/"

for root,dirs,filein os.walk(path):###循环获取文件夹下 文件绝对路径 ##

    for sin file:

_path = path + s

print(_path)

test(_path,s)

4.效果

5 缺点

这个方法是用的颜色进行判断 如果图片颜色和 水印颜色相近 会照成图片污染

如果水印颜色和取值范围颜色不一致会有残留  

水印过小效果不明显