基於python的圖片修復程序-可用於水印去除

圖片修復程序-可用於水印去除

在現實的生活中，咱們可能會遇到一些美好的或是珍貴的圖片被噪聲干擾，好比舊照片的摺痕，好比鏡頭上的灰塵或污漬，更或者是某些咱們想爲我所用但有討厭水印，那麼有沒有一種辦法能夠消除這些噪聲呢？

答案是確定的，依然是被咱們用了無數次的OpenCV這款優秀的框架。

效果預覽

圖片修復原理

那OpenCV到底是怎麼實現的，簡單的來講就是開發者標定噪聲的特徵，在使用噪聲周圍的顏色特徵推理出應該修復的圖片的顏色，從而實現圖片修復的。

程序實現解析

1. 標定噪聲的特徵，使用cv2.inRange二值化標識噪聲對圖片進行二值化處理，具體代碼：cv2.inRange(img, np.array([240, 240, 240]), np.array([255, 255, 255]))，把[240, 240, 240]~[255, 255, 255]之外的顏色處理爲0；
2. 使用OpenCV的dilate方法，擴展特徵的區域，優化圖片處理效果；
3. 使用inpaint方法，把噪聲的mask做爲參數，推理並修復圖片；

完整代碼

#coding=utf-8
#圖片修復

import cv2
import numpy as np

path = "img/inpaint.png"

img = cv2.imread(path)
hight, width, depth = img.shape[0:3]

#圖片二值化處理，把[240, 240, 240]~[255, 255, 255]之外的顏色變成0
thresh = cv2.inRange(img, np.array([240, 240, 240]), np.array([255, 255, 255]))

#建立形狀和尺寸的結構元素
kernel = np.ones((3, 3), np.uint8)

#擴張待修復區域
hi_mask = cv2.dilate(thresh, kernel, iterations=1)
specular = cv2.inpaint(img, hi_mask, 5, flags=cv2.INPAINT_TELEA)

cv2.namedWindow("Image", 0)
cv2.resizeWindow("Image", int(width / 2), int(hight / 2))
cv2.imshow("Image", img)

cv2.namedWindow("newImage", 0)
cv2.resizeWindow("newImage", int(width / 2), int(hight / 2))
cv2.imshow("newImage", specular)
cv2.waitKey(0)
cv2.destroyAllWindows()
複製代碼

圖片擴展與腐蝕更多資料：opencv-python-tutroals.readthedocs.io/en/latest/p…