使用python实现简单去水印功能
更新时间:2022年05月07日 09:31:07 作者:ZuanChampion
这篇文章主要为大家详细介绍了使用python实现简单去水印功能,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
在做视频或者图片处理的时候,我们经常会遇到存在水印的情况,或者我们需要去除图片的某一个部分,这时候我们就需要想办法去除不需要的这一部分。下面这个工具能够控制鼠标将图片上的任意部分改变颜色,从而达到去除水印的效果,这里我们默认修改为白色。
一、准备工作
1、CV2库
2、PIL库
二、源码
import cv2 # 导入相关库 from PIL import Image global img global point1, point2 def on_mouse(event, x, y, flags, param): global img, point1, point2 img2 = img.copy() if event == cv2.EVENT_LBUTTONDOWN: # 左键点击 point1 = (x, y) cv2.circle(img2, point1, 10, (255, 255, 255), 2) cv2.imshow('image', img2) elif event == cv2.EVENT_MOUSEMOVE and (flags & cv2.EVENT_FLAG_LBUTTON): # 按住左键拖曳 cv2.rectangle(img2, point1, (x, y), (255, 255, 255), 2) cv2.imshow('image', img2) elif event == cv2.EVENT_LBUTTONUP: # 左键释放 point2 = (x, y) cv2.rectangle(img2, point1, point2, (255, 255, 255), 2) cv2.imshow('image', img2) min_x = min(point1[0], point2[0]) min_y = min(point1[1], point2[1]) width = abs(point1[0] - point2[0]) height = abs(point1[1] - point2[1]) cut_img = img[min_y:min_y + height, min_x:min_x + width] cv2.imwrite('first.png',cut_img)
将裁剪出来的图片变为白色
def test2(): i = 1 j = 1 img = Image.open("first.png") # 读取系统的内照片 print(img.size) # 打印图片大小 print(img.getpixel((4, 4))) width = img.size[0] # 长度 height = img.size[1] # 宽度 for i in range(0, width): # 遍历所有长度的点 for j in range(0, height): # 遍历所有宽度的点 data = (img.getpixel((i, j))) # 打印该图片的所有点 print(data) # 打印每个像素点的颜色RGBA的值(r,g,b,alpha) print(data[0]) # 打印RGBA的r值 if (data[0] != 255 and data[1] != 255 and data[2] != 255): # RGBA的r值大于170,并且g值大于170,并且b值大于170 img.putpixel((i, j), (255, 255, 255, 255)) # 则这些像素点的颜色改成大红色 img = img.convert("RGB") # 把图片强制转成RGB img.save("mid.png") def testMain(): img = Image.open('/home/gtwang/1000000.png') img2 = Image.open('mid.png') layer = Image.new('RGBA', img.size, (255, 255, 255, 0)) layer.paste(img2, (point1)) img_after = Image.composite(layer, img, layer) img_after.show() img_after.save('output.png') def main(): global img img = cv2.imread('input.png') cv2.namedWindow('image') cv2.setMouseCallback('image', on_mouse) cv2.imshow('image', img) cv2.waitKey(0) if __name__ == '__main__': main() test2() testMain()
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
最新评论