温馨提示×

python去水印的代码怎么写

小亿
124
2023-09-18 14:46:45
栏目: 编程语言

去除水印的代码取决于水印是如何添加到图像中的。以下是一种常见的方法,使用OpenCV库和图像处理技术来去除简单的文本水印:

import cv2
import numpy as np
def remove_watermark(image_path, watermark_path):
# 读取原始图像和水印图像
image = cv2.imread(image_path)
watermark = cv2.imread(watermark_path)
# 确保图像和水印存在
if image is None or watermark is None:
print("无法读取图像或水印图像")
return
# 获取图像和水印的尺寸
image_height, image_width = image.shape[:2]
watermark_height, watermark_width = watermark.shape[:2]
# 确保水印图像小于原始图像
if watermark_height > image_height or watermark_width > image_width:
print("水印图像大于原始图像")
return
# 提取ROI(感兴趣区域):选择图像的右下角作为ROI
roi = image[-watermark_height:, -watermark_width:]
# 创建一个蒙版,将ROI中水印的位置设置为255(白色),其他位置设置为0(黑色)
mask = cv2.cvtColor(watermark, cv2.COLOR_BGR2GRAY)
ret, mask = cv2.threshold(mask, 10, 255, cv2.THRESH_BINARY)
# 将蒙版应用于ROI以去除水印
roi = cv2.bitwise_and(roi, roi, mask=cv2.bitwise_not(mask))
# 将处理后的ROI放回原始图像中
image[-watermark_height:, -watermark_width:] = roi
# 保存去除水印后的图像
output_path = "output.jpg"
cv2.imwrite(output_path, image)
print("去除水印成功,保存为:", output_path)
# 使用示例
remove_watermark("image.jpg", "watermark.jpg")

请注意,这只是一种简单的方法,适用于简单的文本水印。对于复杂的水印,可能需要更复杂的图像处理算法。此外,去除水印可能涉及到侵权行为,请确保遵守相关法律法规。

0