Python圖像處理：Pillow庫(kù)實(shí)現(xiàn)自動(dòng)換行文本標(biāo)注

Python憑借其豐富的開源庫(kù)，已成為圖像處理領(lǐng)域的領(lǐng)先編程語言。Pillow便是其中一個(gè)常用的圖像處理庫(kù)，它簡(jiǎn)潔易用且文檔完善，常用于圖像縮放、裁剪、亮度調(diào)整和標(biāo)注等操作。

然而，Pillow在文本標(biāo)注方面存在一個(gè)問題：當(dāng)文本超過文本框?qū)挾葧r(shí)，不會(huì)自動(dòng)換行。Pillow庫(kù)本身并沒有提供此功能，需要我們自行編寫邏輯實(shí)現(xiàn)。

本教程將演示如何使用Pillow庫(kù)在Python中添加一個(gè)自動(dòng)換行的文本框，從而實(shí)現(xiàn)正確的圖像文本標(biāo)注。最終效果如下：

上圖是我的Dev.to個(gè)人資料截圖，我們將以此為例進(jìn)行講解。綠色文本框即為我們添加的文本標(biāo)注。

準(zhǔn)備工作

本教程要求您具備基本的Python編程知識(shí)，例如條件語句(if, else)、for循環(huán)等。您還需要以下工具和軟件：

1. Python3 : 用于運(yùn)行Python腳本的解釋器。
2. Pillow: Python圖像處理庫(kù)。
3. 代碼編輯器: 例如Pycharm、VScode等。

創(chuàng)建新項(xiàng)目

按照以下步驟創(chuàng)建一個(gè)新項(xiàng)目：

A. 使用終端/命令行創(chuàng)建新文件夾：

mkdir image_annotation

B. 使用pip安裝virtualenv (如果您已安裝，請(qǐng)?zhí)^此步驟)：

pip install virtualenv

C. 將工作目錄切換到image_annotation文件夾：

cd image_annotation

D. 創(chuàng)建新的虛擬環(huán)境：

virtualenv env

E. 激活虛擬環(huán)境 (Windows使用命令提示符)：

Windows:

.\env\Scripts\activate

Linux/macOS:

source env/bin/activate

F. 使用pip安裝Pillow庫(kù)：

pip install pillow

在代碼編輯器中打開項(xiàng)目，并在項(xiàng)目文件夾中創(chuàng)建一個(gè)名為script.py的新Python文件。

準(zhǔn)備基礎(chǔ)圖像

您要標(biāo)注的圖像即為基礎(chǔ)圖像。使用Pillow的ImageDraw模塊打開并準(zhǔn)備圖像。在script.py文件中編寫以下代碼：

from PIL import Image, ImageDraw, ImageFont
image_file = "path_to_image"  # 請(qǐng)?zhí)鎿Q為您的圖片路徑

# 打開圖像
image = Image.open(image_file)

# 初始化ImageDraw
draw = ImageDraw.Draw(image)

添加圖像標(biāo)注

Pillow可以添加純文本和帶背景填充的文本框，文本可以是單行或多行。本教程重點(diǎn)講解如何添加多行文本框。

ImageDraw.multiline_text()方法可以添加多行純文本，但沒有背景填充。ImageDraw.rectangle()方法可以添加帶背景填充的文本框。

在script.py文件中添加以下代碼：

mkdir image_annotation

這段代碼設(shè)置了文本、字體和文本框?qū)挾取?code>x和y變量表示繪圖的起始點(diǎn)，end_x和end_y表示文本框的右下角坐標(biāo)。文本框的寬度和高度分別為200和50。

ImageDraw.rectangle()和ImageDraw.multiline_text()方法分別用于繪制文本框和多行文本。image.show()方法用于顯示處理后的圖像。您可以使用image.save("new_image.png")保存圖像。結(jié)果如下：

上圖中標(biāo)注仍然存在問題，多行文本沒有自動(dòng)換行。下一節(jié)將講解如何解決這個(gè)問題。

實(shí)現(xiàn)自動(dòng)換行

換行符n用于指定換行位置。在之前的例子中，換行符n之后的內(nèi)容會(huì)換行。但在實(shí)際應(yīng)用中，文本長(zhǎng)度通常是動(dòng)態(tài)的，很難確定換行符的位置。

Pillow的ImageDraw模塊的.textlength()屬性可以計(jì)算文本長(zhǎng)度，將其與文本框?qū)挾冗M(jìn)行比較，從而確定換行位置。

在script.py文件頂部（import語句之后）創(chuàng)建一個(gè)名為wrap_text()的新函數(shù)，包含自動(dòng)換行邏輯：

pip install virtualenv

在text, font, max_width變量之后添加以下代碼：

cd image_annotation

將draw.multiline_text()方法替換為以下代碼：

virtualenv env

從文本中刪除換行符n并運(yùn)行代碼：

.\env\Scripts\activate

運(yùn)行結(jié)果顯示文本仍然超出文本框高度。雖然文本可以根據(jù)文本框?qū)挾茸詣?dòng)調(diào)整，但文本框高度固定，導(dǎo)致文本溢出。

設(shè)置動(dòng)態(tài)文本框高度

動(dòng)態(tài)文本框高度根據(jù)文本行數(shù)確定。第一步是將文本框的end_y變量更改為動(dòng)態(tài)值：

source env/bin/activate

此公式是在多次試驗(yàn)后得到的，它似乎是此用例中獲得動(dòng)態(tài)文本框高度的最佳解決方案。wrapped_lines列表包含所有要添加到文本框的行，因此該列表的長(zhǎng)度等于文本框的總行數(shù)。

結(jié)果如下：

您可能需要將總行數(shù)乘以不同的值才能獲得適合您用例的完美解決方案。

添加文本填充

文本距離文本框邊緣太近，影響可讀性和樣式。可以通過在文本框內(nèi)添加填充來解決這個(gè)問題。在script.py文件中添加一個(gè)新的padding變量并更改文本框尺寸：

pip install pillow

這段代碼允許文本和文本框邊緣之間留出間距。

添加指針

指針可以方便地指示標(biāo)注/標(biāo)簽所指的圖像部分。指針應(yīng)該位于標(biāo)注之前。這意味著指針將繪制在文本框的當(dāng)前位置，而文本框?qū)⑾蛴乙苿?dòng)。

因此，文本框的x軸將與新的box_x變量關(guān)聯(lián)。此更改也必須反映在使用文本框x軸的其他變量中。以下是更新后的代碼：

mkdir image_annotation

在上面的代碼中，ImageDraw.circle()方法（其中10是半徑）用于在指定點(diǎn)繪制指針。box_x變量是文本框x軸的新值。

完整代碼

以下是script.py文件的完整代碼：

pip install virtualenv

結(jié)論

圖像處理并不總是像看起來那么困難。雖然有些圖像處理庫(kù)無法直接用其模塊解決您的問題，但您可以使用現(xiàn)有模塊為您的用例實(shí)現(xiàn)特定解決方案。這就是編碼的魅力——能夠用自定義和特定解決方案來解決問題。

在本教程中，您學(xué)習(xí)了如何使用Python的Pillow庫(kù)來標(biāo)注圖像，添加自動(dòng)換行的多行文本框等。您還學(xué)習(xí)了如何編寫可以幫助您進(jìn)行圖像處理的數(shù)學(xué)公式。

請(qǐng)參考Pillow文檔以詳細(xì)了解所用模塊。

以上是使用 Python 的 Pillow 庫(kù)在圖像上包裹和渲染多行文本的詳細(xì)內(nèi)容。更多信息請(qǐng)關(guān)注PHP中文網(wǎng)其他相關(guān)文章！