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

更新時(shí)間：2017年08月08日 10:49:21 作者：普朗克常量

最近的一個(gè)工程項(xiàng)目是講文字添加到圖像上，所以下面這篇文章主要給大家介紹了關(guān)于利用Python自帶PIL庫擴(kuò)展圖片大小給圖片加文字描述的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，需要的朋友可以參考借鑒，下面來一起看看吧。

前言

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

代碼示例

#!/usr/bin/env python3

from PIL import Image, ImageDraw, ImageFont
import math

text="盡管曾作為皇家獵場而存在，意大利大帕拉迪索國家公園一直保留著其野性的一面。畫面里的赤狐靜靜地匍匐在秋草叢中等待時(shí)機(jī)，它的身軀與自然融為一體。所有狐貍都是機(jī)會(huì)主義者，生活在大帕拉迪索的赤狐也不例外；如果有可能，無論是魚類還是野兔，即便是人類野餐留下的殘羹冷炙，它們也不介意吃個(gè)一干二凈。"


def make_text_image(width, white, text, save_path, mode = "rgb"):
 """
 生成一個(gè)文字圖形, white=1，表示白底黑字，否則為黑底白字
 """

 # 字體可能要改
 # linux查看支持的漢字字體 # fc-list :lang=zh
 ft = ImageFont.truetype("DroidSansFallbackFull.ttf", 15)
 w, h = ft.getsize(text)

 # 計(jì)算要幾行
 lines = math.ceil(w / width) + 1
 height = h * lines

 # 一個(gè)漢字的寬度
 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 + " " #處理最后少一個(gè)字問題
 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()

原圖