最近在學習OpenCV，看的是微信讀書里的《OpenCV輕松入門：面向Python》，根據(jù)書上的例題，做了一些小程序

參考來源

《OpenCV輕松入門：面向Python》例3.15

加載原始圖片

我從網(wǎng)上找了一張上海的圖片，加載成了黑白照片

原始圖

oriImg = cv2.imread("shanghai.jpg", 0)  # 0表示將圖片轉(zhuǎn)為黑白
r ,c = oriImg.shape
cv2.namedWindow("origin",cv2.WINDOW_NORMAL)  # 設置窗口尺寸，避免圖片太大，顯示不全
cv2.imshow("origin", oriImg)  # 顯示圖像
cv2.imwrite('OriginImg.png',oriImg)  # 保存圖像

加載水印圖片

水印圖

水印照片的處理比原圖復雜一點，主要是分為幾步
1）因為我找個這個圖有點小，所以我把圖片放大了一點

watermark = cv2.imread("watermark.jpg", 0)
r1 ,c1 = watermark.shape
watermark = cv2.resize(watermark,(r1*6,c1*4))

2）圖片處理一下，把圖片轉(zhuǎn)成0和1兩個值

w1 = watermark[:,:]>200
watermark[w1] = 0
w2 = watermark[:,:]>200
watermark[w2] = 1

3）把水印圖和原圖做的一樣大，方便后續(xù)處理

watermarkImg = np.zeros((r,c), dtype=np.uint8)
r1 ,c1 = watermark.shape
watermarkImg[600:600+r1,2000:2000+c1] = watermark
cv2.namedWindow("watermark",cv2.WINDOW_NORMAL)
cv2.imshow("watermark", watermarkImg)
cv2.imwrite('WatermarkImg.png',watermarkImg)

黑白水印圖

水印嵌入

t254 = np.ones((r,c), dtype=np.uint8)*254 
imgH7 = cv2.bitwise_and(oriImg, t254)
resultImg = cv2.bitwise_or(imgH7, watermarkImg)
cv2.namedWindow("result",cv2.WINDOW_NORMAL)
cv2.imshow("result",resultImg)
cv2.imwrite('ResultImg.png',resultImg)

加入水印

水印解碼

t1 = np.ones((r,c), dtype=np.uint8)
delImg = cv2.bitwise_and(resultImg, t1)
w3 = delImg[:,:]>0
delImg[w3] = 255
cv2.namedWindow("delete",cv2.WINDOW_NORMAL)
cv2.imshow("delete",delImg)
cv2.imwrite('DeleteImg.png',delImg)

提取水印

完整代碼

import cv2
import numpy as np

# 加載原始圖片
oriImg = cv2.imread("shanghai.jpg", 0)
r ,c = oriImg.shape
cv2.namedWindow("origin",cv2.WINDOW_NORMAL)  # 設置窗口尺寸
cv2.imshow("origin", oriImg)
cv2.imwrite('OriginImg.png',oriImg)

# 加載水印圖片
watermark = cv2.imread("watermark.jpg", 0)
r1 ,c1 = watermark.shape
watermark = cv2.resize(watermark,(r1*6,c1*4))
w1 = watermark[:,:]>200
watermark[w1] = 0
w2 = watermark[:,:]>200
watermark[w2] = 1
watermarkImg = np.zeros((r,c), dtype=np.uint8)
r1 ,c1 = watermark.shape
watermarkImg[600:600+r1,2000:2000+c1] = watermark
cv2.namedWindow("watermark",cv2.WINDOW_NORMAL)
cv2.imshow("watermark", watermarkImg)
cv2.imwrite('WatermarkImg.png',watermarkImg)

# 水印嵌入
t254 = np.ones((r,c), dtype=np.uint8)*254 
imgH7 = cv2.bitwise_and(oriImg, t254)
resultImg = cv2.bitwise_or(imgH7, watermarkImg)
cv2.namedWindow("result",cv2.WINDOW_NORMAL)
cv2.imshow("result",resultImg)
cv2.imwrite('ResultImg.png',resultImg)

# 水印解碼
t1 = np.ones((r,c), dtype=np.uint8)
delImg = cv2.bitwise_and(resultImg, t1)
w3 = delImg[:,:]>0
delImg[w3] = 255
cv2.namedWindow("delete",cv2.WINDOW_NORMAL)
cv2.imshow("delete",delImg)
cv2.imwrite('DeleteImg.png',delImg)

cv2.waitKey()
cv2.destroyAllWindows()