Python批量圖片去水印,提高工作效率

chingho發表於2021-05-26

​平常工作中,有時為了採用網路的一些素材,但這些素材往往被打了水印,如果我們不懂PS就無法去掉水印,或者無法批量去掉水印。這些就很影響我們的工作效率。

今天我們就一起來,用Python + OpenCV三步去除水印,去水印需要使用的庫:cv2、numpy。cv2是基於OpenCV的影像處理庫,可以對影像進行腐蝕,膨脹等操作;numpy這是一個強大的處理矩陣和維度運算的庫。

 

圖片去水印原理

1、標定噪聲的特徵,使用cv2.inRange二值化標識噪聲對圖片進行二值化處理,具體程式碼:cv2.inRange(img, np.array([200, 200, 240]), np.array([255, 255, 255])),把[200, 200, 200]~[255, 255, 255]以外的顏色處理為0;

2、使用OpenCV的dilate方法,擴充套件特徵的區域,優化圖片處理效果;

3、使用inpaint方法,把噪聲的mask作為引數,推理並修復圖片。

 

去掉右下角的水印步驟

1、從源圖片,擷取右下角部分,另存為新圖片;

2、識別水印,顏色值為:[200, 200, 200]~[255, 255, 255]

3、去掉水印,還原圖片;

4、把源圖片、去掉水印的新圖片,進行重疊合並;

 

實現程式碼

import cv2
import numpy as np
from PIL import Image
import os
​
dir = os.getcwd()
path = "1.jpg"
newPath = "new.jpg"
img=cv2.imread(path,1)
hight,width,depth=img.shape[0:3]
​
#擷取
cropped = img[int(hight*0.8):hight, int(width*0.7):width]  # 裁剪座標為[y0:y1, x0:x1]
cv2.imwrite(newPath, cropped)
imgSY = cv2.imread(newPath,1)
​
#圖片二值化處理,把[200,200,200]-[250,250,250]以外的顏色變成0
thresh = cv2.inRange(imgSY,np.array([200,200,200]),np.array([250,250,250]))
#建立形狀和尺寸的結構元素
kernel = np.ones((3,3),np.uint8)
#擴充套件待修復區域
hi_mask = cv2.dilate(thresh,kernel,iterations=10)
specular = cv2.inpaint(imgSY,hi_mask,5,flags=cv2.INPAINT_TELEA)
cv2.imwrite(newPath, specular)
​
#覆蓋圖片
imgSY = Image.open(newPath)
img = Image.open(path)
img.paste(imgSY, (int(width*0.7),int(hight*0.8),width,hight))
img.save(newPath)

  

效果對比

實現效果如下圖,這個預設是去掉白色右下角的水印,大家可以根據自己的需求進行更改。

Python批量圖片去水印,提高工作效率

 

好了,今天就分享到這邊,大家有什麼想法,歡迎給我留言討論。​如果需要原始碼與軟體的,請在公眾號回覆【去水印】獲取。

Python批量圖片去水印,提高工作效率

文章首發公眾號【程式設計樂趣】,歡迎關注

 

相關文章