天天看點

用Python頭像秒變成表情包

在日常生活中,我們經常會存取一些朋友們的醜照,在這個項目中,我們以萌萌哒的熊貓頭作為背景,然後試着在背景圖上加入朋友們的照片。效果如下圖所示:

用Python頭像秒變成表情包

用Python頭像秒變成表情包

用Python頭像秒變成表情包

 很多人學習python,不知道從何學起。

很多人學習python,掌握了基本文法過後,不知道在哪裡尋找案例上手。

很多已經做案例的人,卻不知道如何去學習更加高深的知識。

那麼針對這三類人,我給大家提供一個好的學習平台,免費領取視訊教程,電子書籍,以及課程的源代碼!

QQ群:609616831

二、實作步驟

  1. 導入朋友的照片(前景照片);
  2. 處理前景照片(縮放、旋轉,填充);
  3. 導入熊貓頭照片(背景照片);
  4. 将前景和背景拼接起來形成表情包;
  5. 在表情包下面添加文字。

三、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(\'SXC.jpg\', 0)  # 導入灰階圖即可
plt_show(image)
           
用Python頭像秒變成表情包

4、等比例縮放前景照片

因為我們發現前景照片的尺寸比背景尺寸還要大,這顯然是不合适的,是以要先對其進行等比例(0.3)縮放。

image_resize = cv2.resize(image, None, fx=0.3, fy=0.3, interpolation = cv2.INTER_CUBIC)
plt_show(image_resize)
           
用Python頭像秒變成表情包

5、對前景照片進行二值化處理

在這裡,我們将像素值大于 80 的區域設定為 255;小于 80 的區域設定成 0。

ret, image_binary = cv2.threshold(image_resize, 80, 255, cv2.THRESH_BINARY)
plt_show(image_binary)
           
用Python頭像秒變成表情包

6、提取出感興趣區域

image_roi = image_binary[74: 185, 0: 150]
plt_show(image_roi)
           
用Python頭像秒變成表情包

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, (140, 130))  # (140, 130) 是指旋轉後的畫布大小
plt_show(image_rotate)
           
用Python頭像秒變成表情包

8、将一些不需要的黑色區域删除掉

在這裡我們使用 cv2.fillPoly 函數對不需要的區域用白色進行填充。

h, w = image_rotate.shape

image_rotate_copy = image_rotate.copy()
pts1 = np.array([[0, 20],  [64, 0], [0, 0]], np.int32)
pts2 = np.array([[0, 18],  [0, h], [80, h]], np.int32)
pts3 = np.array([[0, 100],  [0, h], [w, h], [w, 100]], np.int32)
pts4 = np.array([[111, 0],  [w, 0], [w, 30]], np.int32)
pts5 = np.array([[124, 0],  [115, h], [w, h]], np.int32)
pts6 = np.array([[120, 40],  [95, 100], [120, 100]], np.int32)
foreground = cv2.fillPoly(image_rotate_copy, [pts1], (255, 255, 255))  # (圖檔,填充區域,填充顔色)
foreground = cv2.fillPoly(image_rotate_copy, [pts2], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts3], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts4], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts5], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts6], (255, 255, 255))

plt_show(foreground)
           
用Python頭像秒變成表情包

9、再次提取感興趣區域并縮放

foreground_roi = foreground[0: 93, 0: 125]
plt_show(foreground_roi)

foreground_roi_resize = cv2.resize(foreground_roi, None, fx=2.5, fy=2.5, interpolation = cv2.INTER_CUBIC)
plt_show(foreground_roi_resize)
           
用Python頭像秒變成表情包

用Python頭像秒變成表情包

10、導入背景圖檔

background = cv2.imread(\'back.jpg\', 0)
plt_show(background)
           
用Python頭像秒變成表情包

11、組合兩張圖檔成表情包

h_f, w_f = foreground.shape
h_b, w_b = background.shape

left = (w_b - w_f)//2  # 前景圖檔在背景圖檔中的左邊的橫坐标
right = left + w_f  # 前景圖檔在背景圖檔中的右邊的橫坐标
top = 100  # 前景圖檔在背景圖檔中的上邊的縱坐标
bottom = top + h_f  # 前景圖檔在背景圖檔中的下邊的縱坐标

emoji = background
emoji[top: bottom, left: right] = foreground
plt_show(emoji)
           
用Python頭像秒變成表情包

12、在表情包下面添加文本

12.1 添加英文文本

如果隻是要添加英文文本,用 opencv 就可以解決:

emoji_copy = emoji.copy()
# (圖檔,文本,位置,字型,文本大小,文本顔色,文本粗細)
cv2.putText(emoji_copy, "FXXK!!", (210, 500), cv2.FONT_HERSHEY_SIMPLEX, 1.2, (0, 0, 0), 5)
plt_show(emoji_copy)
           
用Python頭像秒變成表情包

12.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)
           
用Python頭像秒變成表情包

13、儲存表情包

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(\'SXC.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, 80, 255, cv2.THRESH_BINARY)  # 圖檔二值化

image_roi = image_binary[74: 185, 0: 150]  # 感興趣區域

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, (140, 130))
# 填充不需要的區域
h, w = image_rotate.shape
image_rotate_copy = image_rotate.copy()
pts1 = np.array([[0, 20],  [64, 0], [0, 0]], np.int32)
pts2 = np.array([[0, 18],  [0, h], [80, h]], np.int32)
pts3 = np.array([[0, 100],  [0, h], [w, h], [w, 100]], np.int32)
pts4 = np.array([[111, 0],  [w, 0], [w, 30]], np.int32)
pts5 = np.array([[124, 0],  [115, h], [w, h]], np.int32)
pts6 = np.array([[120, 40],  [95, 100], [120, 100]], np.int32)
foreground = cv2.fillPoly(image_rotate_copy, [pts1], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts2], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts3], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts4], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts5], (255, 255, 255))
foreground = cv2.fillPoly(image_rotate_copy, [pts6], (255, 255, 255))

foreground_roi = foreground[0: 93, 0: 125]
foreground_roi_resize = cv2.resize(foreground_roi, None, fx=2.5, fy=2.5, interpolation = cv2.INTER_CUBIC)

background = cv2.imread(\'back.jpg\', 0)  # 導入背景圖檔
# 拼接兩張圖檔
h_f, w_f = foreground_roi_resize.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_roi_resize

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))  # 儲存表情包           

 在這裡還是要推薦下我自己建的Python學習群:609616831,群裡都是學Python的,如果你想學或者正在學習Python ,歡迎你加入,大家都是軟體開發黨,不定期分享幹貨(隻有Python軟體開發相關的),包括我自己整理的一份2020最新的Python進階資料和零基礎教學,歡迎進階中和對Python感興趣的小夥伴加入!

用Python頭像秒變成表情包