python 去除水印_python opencv 去除水印

#!/usr/bin/env python

# -*- coding:utf-8 -*-

# Author : zhibo.wang

# E-mail : d_1206@qq.com

# Date : 18/11/03 20:07:41

# Desc :

import cv2

import numpy as np

def img_clean(img):

height, width = img.shape[0:2]

cv2.namedWindow("Image", 0)

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

cv2.imshow('Image', img)

rects = ((width - 170, height - 38, width, height),(1, 1, 164, 100)) #水印区域

mask = np.zeros((height, width), np.uint8)

for rect in rects:

x1, y1, x2, y2 = rect

cv2.rectangle(mask, (x1, y1), (x2, y2), (255, 255, 255), -1)

img = cv2.inpaint(img, mask, 1.5, cv2.INPAINT_TELEA) #蒙版

cv2.namedWindow("newImage", 0)

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

cv2.imshow('newImage', img)

cv2.waitKey(0)

cv2.destroyAllWindows()

if __name__ == '__main__':

f_img = 'shui1.jpg'

img = cv2.imread(f_img)

img_clean(img)