文章目錄
項目說明
在日常生活中,我們經常會存取一些朋友們的醜照,在這個項目中,我們以萌萌噠的熊貓頭作爲背景,然後試着在背景圖上加入朋友們的照片。效果如下圖所示:
實現步驟
- 導入朋友的照片(前景照片);
- 處理前景照片(縮放、旋轉,填充);
- 導入熊貓頭照片(背景照片);
- 將前景和背景拼接起來形成表情包;
- 在表情包下面添加文字。
Python 實現
1、導入需要的庫
import cv2
import numpy as mp
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont
這個項目主要是通過 opencv 完成,但如果要在表情包下面寫中文的話,PIL(pillow)庫是必不可少的。
2、繪圖函數
這裏寫一個繪圖函數,方便繪圖操作。
def plt_show(img):
imageRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.imshow(imageRGB)
plt.show()
3、導入前景照片
image = cv2.imread('fore.jpg', 0) # 導入灰度圖即可
plt_show(image)
4、等比例縮放前景照片
因爲我們發現前景照片的尺寸比背景尺寸還要大,這顯然是不合適的,所以要先對其進行等比例(0.3)縮放。
image_resize = cv2.resize(image, None, fx=0.3, fy=0.3, interpolation = cv2.INTER_CUBIC)
plt_show(image_resize)
5、對前景照片進行二值化處理
在這裏,我們將像素值大於 70 的區域設置爲 255;小於 70 的區域設置成 0。
ret, image_binary = cv2.threshold(image_resize, 70, 255, cv2.THRESH_BINARY)
plt_show(image_binary)
6、提取出感興趣區域
image_roi = image_binary[250: 550, 90: 380]
plt_show(image_roi)
7、旋轉圖片
因爲我們的背景圖片(熊貓頭)是正的,而前景圖片有些向右傾斜,所以要先對其進行旋轉操作(大概逆時針旋轉 15 度即可)。
rows, cols = image_roi.shape
M = cv2.getRotationMatrix2D(((cols-1)/2.0, (rows-1)/2.0), 15, 1) # (旋轉中心,逆時針旋轉角度,各個方向同等擴大比例)
image_rotate = cv2.warpAffine(image_roi, M, (280, 275)) # (280, 275) 是指旋轉後的畫布大小
plt_show(image_rotate)
8、將一些不需要的黑色區域刪除掉
在這裏我們使用 cv2.fillPoly 函數對不需要的區域用白色進行填充。
image_rotate_copy = image_rotate.copy()
pts1 = np.array([[0, 33], [122, 0], [0, 0]], np.int32)
pts2 = np.array([[0, 145], [0, h], [30, h]], np.int32)
pts3 = np.array([[180, h], [w, h], [w, 75]], np.int32)
pts4 = np.array([[250, 0], [w, 0], [w, 100]], np.int32)
foreground = cv2.fillPoly(image_rotate_copy, [pts1, pts2, pts3, pts4], (255, 255, 255)) # (圖片,填充區域,填充顏色)
plt_show(foreground)
9、導入背景圖片
background = cv2.imread('back.jpg', 0)
plt_show(background)
10、組合兩張圖片成表情包
h_f, w_f = foreground.shape
h_b, w_b = background.shape
left = (w_b - w_f)//2 # 前景圖片在背景圖片中的左邊的橫座標
right = left + w_f # 前景圖片在背景圖片中的右邊的橫座標
top = 80 # 前景圖片在背景圖片中的上邊的縱座標
bottom = top + h_f # 前景圖片在背景圖片中的下邊的縱座標
emoji = background
emoji[top: bottom, left: right] = foreground
plt_show(emoji)
11、在表情包下面添加文本
11.1 添加英文文本
如果只是要添加英文文本,用 opencv 就可以解決:
emoji_copy = emoji.copy()
# (圖片,文本,位置,字體,文本大小,文本顏色,文本粗細)
cv2.putText(emoji_copy, "hhh", (230, 500), cv2.FONT_HERSHEY_SIMPLEX, 1.2, (0, 0, 0), 5)
plt_show(emoji_copy)
11.2 添加中文文本
如果要添加中文文本,我們需要藉助 PIL 庫來實現。
PilImg = Image.fromarray(emoji) # cv2 轉 PIL
draw = ImageDraw.Draw(PilImg) # 創建畫筆
ttfront = ImageFont.truetype('simhei.ttf', 34) # 設置字體
draw.text((210, 450),"嘿嘿嘿",fill=0, font=ttfront) # (位置,文本,文本顏色,字體)
emoji_text = cv2.cvtColor(np.array(PilImg),cv2.COLOR_RGB2BGR) # PIL 轉回 cv2
plt_show(emoji_text)
12、保存表情包
cv2.imwrite('./emoji.png', np.array(emoji_text))
完整代碼
import cv2
import numpy as mp
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont
def plt_show(img):
imageRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.imshow(imageRGB)
plt.show()
image = cv2.imread('fore.jpg', 0) # 導入前景圖片
image_resize = cv2.resize(image, None, fx=0.3, fy=0.3, interpolation = cv2.INTER_CUBIC) # 縮放
ret, image_binary = cv2.threshold(image_resize, 70, 255, cv2.THRESH_BINARY) # 圖片二值化
image_roi = image_binary[250: 550, 90: 380] # 感興趣區域
rows, cols = image_roi.shape
# 旋轉
M = cv2.getRotationMatrix2D(((cols-1)/2.0, (rows-1)/2.0), 15, 1)
image_rotate = cv2.warpAffine(image_roi, M, (280, 275))
# 填充不需要的區域
h, w = image_rotate.shape
image_rotate_copy = image_rotate.copy()
pts1 = np.array([[0, 33], [122, 0], [0, 0]], np.int32)
pts2 = np.array([[0, 145], [0, h], [30, h]], np.int32)
pts3 = np.array([[180, h], [w, h], [w, 75]], np.int32)
pts4 = np.array([[250, 0], [w, 0], [w, 100]], np.int32)
foreground = cv2.fillPoly(image_rotate_copy, [pts1, pts2, pts3, pts4], (255, 255, 255))
background = cv2.imread('back.jpg', 0) # 導入背景圖片
# 拼接兩張圖片
h_f, w_f = foreground.shape
h_b, w_b = background.shape
left = (w_b - w_f)//2
right = left + w_f
top = 80
bottom = top + h_f
emoji = background
emoji[top: bottom, left: right] = foreground
PilImg = Image.fromarray(emoji) # cv2 轉 PIL
draw = ImageDraw.Draw(PilImg) # 創建畫筆
ttfront = ImageFont.truetype('simhei.ttf', 34) # 設置字體
draw.text((210, 450),"嘿嘿嘿",fill=0, font=ttfront) # (位置,文本,文本顏色,字體)
emoji_text = cv2.cvtColor(np.array(PilImg),cv2.COLOR_RGB2BGR) # PIL 轉回 cv2
cv2.imwrite('./emoji.png', np.array(emoji_text)) # 保存表情包