利用Python自帶PIL庫擴展圖片大小給圖片加文字描述的方法示例

最近的一個項目中需要在圖片上添加文字，使用了OpenCV，結果發現利用opencv給圖像添加文字有局限?？衫玫淖煮w類型比較少，需要安裝Freetype擴展，比較復雜。而且不能用putText函數輸出中文，否則就會出現亂碼的情況。只好選擇使用Python PIL函數庫對照片進行處理，利用Python自帶的PIL庫擴展圖片大小給圖片加上文字描述，大多都是庫函數調用，只是給定圖片寬度后計算文字所需行數的代碼需要寫。 代碼比較丑，but it works.

代碼示例
#!/usr/bin/env python3
 
from PIL import Image, ImageDraw, ImageFont
import math
 
text="盡管曾作為皇家獵場而存在，意大利大帕拉迪索國家公園一直保留著其野性的一面。畫面里的赤狐靜靜地匍匐在秋草叢中等待時機，它的身軀與自然融為一體。所有狐貍都是機會主義者，生活在大帕拉迪索的赤狐也不例外；如果有可能，無論是魚類還是野兔，即便是人類野餐留下的殘羹冷炙，它們也不介意吃個一干二凈。"
 
 
def make_text_image(width, white, text, save_path, mode = "rgb"):
 """
 生成一個文字圖形, white=1，表示白底黑字，否則為黑底白字
 """
 
 # 字體可能要改
 # linux查看支持的漢字字體 # fc-list :lang=zh
 ft = ImageFont.truetype("DroidSansFallbackFull.ttf", 15)
 w, h = ft.getsize(text)
 
 # 計算要幾行
 lines = math.ceil(w / width) + 1
 height = h * lines
 
 # 一個漢字的寬度
 one_zh_width, h = ft.getsize("中")
 
 if len(mode) == 1: # L, 1
  background = (255)
  color = (0)
 if len(mode) == 3: # RGB
  background = (255, 255, 255)
  color = (0,0,0)
 if len(mode) == 4: # RGBA, CMYK
  background = (255, 255, 255, 255)
  color = (0,0,0,0)
 
 newImage = Image.new(mode, (width, height), background if white else color)
 draw = ImageDraw.Draw(newImage)
 
 # 分割行
 text = text + " " #處理最后少一個字問題
 text_list = []
 start = 0
 end = len(text) - 1
 while start < end:
  for n in range(end):
   try_text = text[start:start+n]
   w,h = ft.getsize(try_text)
   if w + 2*one_zh_width > width:
    break
  text_list.append(try_text[0:-1])
  start = start + n - 1;
 
 # print(text_list)
 
 i = 0
 for t in text_list:
  draw.text((one_zh_width, i * h), t, color if white else background, font=ft)
  i = i + 1
 
 newImage.save(save_path);
 
 
def resize_canvas(org_image="aa.jpg", add_image="222.jpg", new_image_path="save2.jpg"):
 
 org_im = Image.open(org_image)
 org_width, org_height = org_im.size
 
 mode = org_im.mode
 
 make_text_image(org_width, 0, text, "222.jpg", mode)
 
 add_im = Image.open(add_image)
 add_width, add_height = add_im.size
 
 mode = org_im.mode
 
 newImage = Image.new(mode, (org_width, org_height + add_height))
 
 newImage.paste(org_im, (0, 0, org_width, org_height))
 newImage.paste(add_im, (0, org_height, add_width, add_height + org_height))
 newImage.save(new_image_path)
 
resize_canvas()

原圖

改之后的圖