使用python實現(xiàn)簡單去水印功能

更新時間：2022年05月07日 09:31:07 作者：ZuanChampion

這篇文章主要為大家詳細介紹了使用python實現(xiàn)簡單去水印功能，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

在做視頻或者圖片處理的時候，我們經常會遇到存在水印的情況，或者我們需要去除圖片的某一個部分，這時候我們就需要想辦法去除不需要的這一部分。下面這個工具能夠控制鼠標將圖片上的任意部分改變顏色，從而達到去除水印的效果，這里我們默認修改為白色。

一、準備工作

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)

將裁剪出來的圖片變?yōu)榘咨?/p>

def test2():
i = 1
j = 1
img = Image.open("first.png") # 讀取系統(tǒng)的內照片
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()

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: