OpenCV python去除图片水印

水深无声 2023-05-30 14:59 28阅读 0赞

> 在浏览[阿里巴巴][Link 1]网站的时候发现有些图片有水印,想着怎么能把它们去除.以下的去除只适合单个图片去除,对于大量图片去除水印的功能需要大伙一起研究

HSV范围  
![在这里插入图片描述][20191111081228576.png]  
(图1)  
![在这里插入图片描述][2019111108174327.png]  
(图2)

# -*- coding: utf-8 -*-
    # @Time    : 2019/11/11 16:50
    # @Author  :
    
    
    import cv2 as cv
    import numpy as np
    
    
    def console_location(path):
        """
        控制台输出区域像素的位置
        :param path:
        :return:
        """
        # 定片位置
        img = cv.imread(path)
    
        def on_mouse(event, x, y, flags, param):
            if event == cv.EVENT_LBUTTONDOWN:
                # 宽 高
                print(x, y)
    
        # 构建窗口
        # 回调绑定窗口
        cv.namedWindow("img", cv.WINDOW_NORMAL)
        cv.setMouseCallback("img", on_mouse, 0)
        cv.imshow("img", img)
        # 键盘输入 q退出
        if cv.waitKey() == ord("q"):
            cv.destroyAllWindows()
    
    
    class WaterMark(object):
    
        def mark(self, path):
            # 提取感兴趣区域ROI
            img = cv.imread(path)
            # 通过运行console_location函数后在相应的图片上点击两个点可以获取一下两个参数
            # 高1:高2 宽1:宽2
            roi = img[248:291, 91:596]
            # cv.imwrite('02.jpg', roi)
            roi_hsv = cv.cvtColor(roi, cv.COLOR_BGR2HSV)
            # cv.imwrite('hsv.jpg', roi_hsv)
    
            # 设定白色HSV范围
            lower = np.array([0, 0, 221])
            upper = np.array([180, 30, 247])
    
            # 创建水印蒙层
            kernel = np.ones((3, 3), np.uint8)
            # cv.imwrite('kernel.jpg',kernel)
            mask = cv.inRange(roi_hsv, lower, upper)
            # cv.imwrite(r'mask.jpg',mask)
            # 对水印蒙层进行膨胀操作
            dilate = cv.dilate(mask, kernel, iterations=3)
            res = cv.inpaint(roi, dilate, 7, flags=cv.INPAINT_TELEA)
            # 双边滤波
            res = cv.bilateralFilter(res, 5, 280, 50)
    
            # 高1:高2 宽1:宽2
            img[248:291, 91:596] = res
            cv.imwrite('modified.jpg', img)
    
    
    if __name__ == '__main__':
        path = './img/0fe80411-8d2f-4150-be73-ae43ba2e7cdc.jpg'
        # console_location(path)
        w = WaterMark()
        w.mark(path)

效果  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0NvY2t0YWlsX3B5_size_16_color_FFFFFF_t_70]

[Link 1]: https://www.alibaba.com/
[20191111081228576.png]: https://img-blog.csdnimg.cn/20191111081228576.png
[2019111108174327.png]: https://img-blog.csdnimg.cn/2019111108174327.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0NvY2t0YWlsX3B5_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191111082700325.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0NvY2t0YWlsX3B5,size_16,color_FFFFFF,t_70