基于python的图像去水印

1 代码

python 复制代码
import cv2
import numpy as np

def remove_watermark(image_path, output_path):
    # 读取图片
    image = cv2.imread(image_path)

    # 转换为灰度图
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

    # 使用中值滤波去除噪声
    median_filtered = cv2.medianBlur(gray, 5)

    # 计算图像的梯度
    laplacian = cv2.Laplacian(median_filtered, cv2.CV_64F)

    # 将梯度图像转换为8位无符号整数
    laplacian_8u = np.uint8(np.absolute(laplacian))

    # 使用阈值操作找到潜在的水印区域
    _, thresholded = cv2.threshold(laplacian_8u, 30, 255, cv2.THRESH_BINARY)

    # 对阈值图像进行形态学操作,填充孔洞并平滑边缘
    kernel = np.ones((3, 3), np.uint8)
    closing = cv2.morphologyEx(thresholded, cv2.MORPH_CLOSE, kernel, iterations=2)

    # 创建一个掩码,将潜在的水印区域设置为白色
    mask = np.zeros_like(image)
    mask[closing == 255] = [255, 255, 255]

    # 将掩码转换为单通道图像
    mask_gray = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)

    # 使用掩码去除水印
    result = cv2.inpaint(image, mask_gray, 3, cv2.INPAINT_TELEA)

    # 保存去水印后的图片
    cv2.imwrite(output_path, result)

remove_watermark('input.jpg', 'output.jpg')

2 去水印效果

图2-1 去水印前后对比

从图2-1可以看出并没有完美去除水印,这是由水印与背景颜色接近造成的,后面我会用深度学习的方法探索一番。

相关推荐
动能小子ohhh9 分钟前
AI智能体(Agent)大模型入门【2】--基于llamaindx部署本地的聊天模型。
人工智能·python·aigc·ai编程
MediaTea19 分钟前
Python 第三方库:SymPy(符号计算工具)
开发语言·python·数学建模
Goona_19 分钟前
PyQt数字转大写金额GUI工具开发及财务规范实现
python·小程序·交互·pyqt
Pocker_Spades_A38 分钟前
Python快速入门专业版(二十二):if语句进阶:嵌套if与条件表达式(简洁写法技巧)
开发语言·python
看海的四叔1 小时前
【Python】Python解决阿里云DataWorks导出数据1万条限制的问题
开发语言·python·阿里云·dataworks·maxcomputer
吾日三省吾码1 小时前
用 Python UTCP 直调 HTTP、CLI、MCP……
开发语言·python·http
深耕AI1 小时前
【参数详解与使用指南】PyTorch MNIST数据集加载
人工智能·pytorch·python
站大爷IP1 小时前
PID控制算法原理与Python实现:从理论到实践的通俗解析
python
aopstudio1 小时前
如何优雅地清理Hugging Face缓存到本地的模型文件(2025最新版)
人工智能·python·缓存·语言模型
Wiktok1 小时前
[Wit]CnOCR模型训练全流程简化记录(包括排除BUG)
python·深度学习·bug