使用Python简单实现去除彩色水印

import cv2
import numpy as np
 
def remove_color_watermark(image_path, thres):
 
    """    
        :param image_path: 图片路径
        :param thres: V值的阈值
        :return: 除掉水印后的图片
    """
 
    # 读取图片
    img = cv2.imread(image_path)
    if img is None:
        print("无法读取图片，请检查图片路径。")
        return
 
    # 转换到HSV颜色空间
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    # 某像素点的V值大于阈值则将其转换为白色
    img[hsv[:, :, 2] > thres ] = [255, 255, 255]
    return img
 
# 请替换为你的图片路径
image_path = f"path\\to\\orig\\img.jpg"
result = remove_color_watermark(image_path, 230)
 
if result is not None:
    # 显示原始图片和处理后的图片
    cv2.imshow('Original Image', cv2.imread(image_path))
    cv2.imshow('Processed Image', result)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
 
    # 保存处理后的图片
    cv2.imwrite('processed_image.jpg', result)

import cv2
import os
 
from PIL import Image
 
def remove_color_watermark(image_path, thres):
 
    # 读取图片
    img = cv2.imread(image_path)
    if img is None:
        print("无法读取图片，请检查图片路径。")
        return
 
    # 转换到HSV颜色空间
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    # 某像素点的v值大于阈值则将其转换为白色
    img[hsv[:, :, 2] > thres] = [255, 255, 255]
    return img
 
 
if __name__ == '__main__':
    image_path = 'folder\\path\\to\\orig'
    out_path = 'path\\to\\output\\folder'
    thresh = 230
    images = []
    for filename in os.listdir(image_path):
        if filename.endswith(('.jpg', '.jpeg', '.png', '.bmp', '.tiff')):
            img = remove_color_watermark(os.path.join(image_path, filename), thresh)
            img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
            # OpenCV图片转换为pillow图片
            pillow_image = Image.fromarray(img)
            print(f'{filename}已经处理完毕。')
            # 加入pillow图片数组，为输出PDF文件做准备
            images.append(pillow_image)
    # 将处理后的图片输出为PDF文件
    images[0].save(os.path.join(out_path,'output.pdf'), save_all=True, append_images=images[1:])