OpenCV圖像處理 - 為圖像添加Logo

1藻三，目的

將logo圖標(biāo)疊加到一張圖片的右上角梁沧，要求有顏色的區(qū)域?yàn)椴煌该髌鹛瑁鐚⒗镉皹I(yè)的Logo疊加到戰(zhàn)狼2的海報上柜与，原始圖和效果圖如下。

wolf.jpg

ali-movie.png

img_target.png

2悄蕾，思路

不同的logo有不同的處理方法票顾，有的是黑色需要透明，有的是白色需要透明帆调，但是奠骄，整理的思路是一致的。

如下圖所示番刊，從上向下含鳞，對要形成的效果進(jìn)行分解，一共分為3層芹务。

第一層需要第二層的兩張圖片蝉绷，做cv2.add運(yùn)算即可
第二層第一張圖片需要第三層前兩張圖片，做cv2.bitwise_and運(yùn)算即可枣抱，后一張做mask
第二層第二張圖片需要第三層后兩張圖片熔吗，做cv2.bitwise_and運(yùn)算即可，后一張做mask

method of img deal.png

在分解過程中佳晶，需要思考和明確要調(diào)用的OpenCV函數(shù)桅狠。如果一步解決不了，就拆分成多步解決轿秧。

3中跌，步驟分析

基于上述思路，制定詳細(xì)的圖像處理步驟淤刃，并進(jìn)行編碼調(diào)試晒他。處理步驟如下圖所示。

ali-logo deal process.png

3.1逸贾，讀取logo圖片陨仅，并對logo按照20%進(jìn)行縮放

img_logo = cv2.imread("d:/1/ali-movie.png", cv2.IMREAD_COLOR)
img_logo = cv2.resize(img_logo, (0, 0), fx=0.2, fy=0.2, interpolation=cv2.INTER_NEAREST)

3.2津滞，對logo進(jìn)行灰度化處理，得到兩個mask

img_logo_gray = cv2.cvtColor(img_logo, cv2.COLOR_BGR2GRAY)
ret, img_logo_mask = cv2.threshold(img_logo_gray, 200, 255, cv2.THRESH_BINARY)  # 二值化函數(shù)
img_logo_mask1 = cv2.bitwise_not(img_logo_mask)

3.3灼伤，提取目標(biāo)圖片的ROI

img_target = cv2.imread("d:/1/wolf.jpg", cv2.IMREAD_COLOR)
rows, cols, channel = img_logo.shape
rows1, cols1, channel1 = img_target.shape
img_roi = img_target[:rows, cols1 - cols:cols1].copy()

3.4触徐，ROI和Logo圖像融合

img_res0 = cv2.bitwise_and(img_roi, img_roi, mask=img_logo_mask)
img_res1 = cv2.bitwise_and(img_logo, img_logo, mask=img_logo_mask1)
img_res2 = cv2.add(img_res0, img_res1)
# img_res2 = img_res0 + img_res1
img_target[:rows, cols1 - cols:cols1] = img_res2[:, :]
cv2.imwrite("img_target.png", img_target)

4，疑難點(diǎn)介紹

4.1狐赡，cv2.add與矩陣相加的區(qū)別

測試程序

import numpy as np
import cv2
x = np.uint8([250])
y = np.uint8([10])
print cv2.add(x,y)   #輸出# [[255]]
print x+y            #輸出# [4]  (250+10)%255

測試結(jié)論

cv2.add方法和numpy矩陣相加撞鹉，都可以完成兩張圖片的相加運(yùn)算
numpy矩陣相加是模運(yùn)算
cv2.add是滲透運(yùn)算，如果元素之和大于255颖侄，則按照255返回鸟雏，可以直接相加兩張圖片試試效果，代碼和效果圖如下所示

img_logo = cv2.imread("d:/1/ali-movie.png", cv2.IMREAD_COLOR)
img_target = cv2.imread("d:/1/wolf.jpg", cv2.IMREAD_COLOR)
print img_logo.shape, img_target.shape
img_new_add = cv2.add(img_target[:516, :798], img_logo[:516, :798])
cv2.imshow("img_new_add", img_new_add)
cv2.waitKey(0)
cv2.destroyAllWindows()

cv2.add.png

4.2览祖，matplotlib的subplot簡單介紹

代碼如下孝鹊，效果圖如method of img deal.png所示。

代碼

# 顯示圖片展蒂，調(diào)用matplotlib展示
plt.figure()
plt.subplot(332), plt.imshow(img_convert(img_res2), cmap='gray'), plt.title("img_res2")
plt.subplot(323), plt.imshow(img_convert(img_res0), cmap='gray'), plt.title("img_res0")
plt.subplot(324), plt.imshow(img_convert(img_res1), cmap='gray'), plt.title("img_res1")
plt.subplot(3, 4, 9), plt.imshow(img_convert(img_roi), cmap='gray'), plt.title("img_roi")
plt.subplot(3, 4, 10), plt.imshow(img_convert(img_logo_mask), cmap='gray'), plt.title("img_logo_mask")
plt.subplot(3, 4, 11), plt.imshow(img_convert(img_logo), cmap='gray'), plt.title("img_logo")
plt.subplot(3, 4, 12), plt.imshow(img_convert(img_logo_mask1), cmap='gray'), plt.title("img_logo_mask1")
plt.show()

效果圖