python - 如何将“文本框”中的文本添加到图像中？

Question

我正在开发一种工具来将文本添加到图像的特定区域，例如在对话气泡内。我正在生成一个边界框，并希望将文本限制在此边界框内，但到目前为止，查看 cv2 和 PIL 库它们似乎只是为文本而不是边界框取了一个起点。

简历2：

import cv2

myimg = "471.jpg"

img = cv2.imread(myimg)

f_face = cv2.FONT_HERSHEY_PLAIN
f_size = 1
f_color = (255,255,255)
f_thickness = 2
text = 'Hello World'
x,y = 400,800
img = cv2.putText(img, text, (x,y), f_face, f_size, f_color, f_thickness)

cv2.imshow("Test", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

皮尔：

from PIL import Image
from PIL import ImageDraw

myimg = "471.jpg"

img = Image.open(myimg)

id = ImageDraw.Draw(img)

id.text((400, 800), "Hello World", fill=(255, 255, 255))

img.show()

理想情况下，我会寻找一个库，它可以让我对文本进行与使用 Photoshop 时相同的控制，这意味着：字体大小、字体、中心（到边界框/文本框）垂直和水平居中（到边界框/文本框）以及自动调整字体大小或边界框以使其适合文本不会被切断（或至少给出某种警告，即文本已溢出文本框，以便我可以制作一个减小字体大小并重试的过程）。

有没有其他图书馆有这种控制？或者有没有办法在 cv2 或 PIL 中做到这一点？

谢谢您的帮助

标签： pythonpython-3.ximageimage-processingpython-imaging-library

Answer 1

Pillow有ImageFont来选择字体，字体大小，你可以在text(..., font=...).

在较新的版本中，您可以使用text((center_x, center_y), anchor='mm')在区域中居中文本。查看更多：文本锚点

问题仅导致自动调整字体大小。

您必须使用ImageFont.FreeTypeFont.getbbox()来计算具有选定字体大小的文本使用的区域大小。如果您使用for-loop 来计算不同字体大小的区域大小，那么您可以自动调整它的大小。

---

import PIL
print('PIL version:', PIL.__version__) 

from PIL import Image, ImageDraw, ImageFont

# create empty image
img = Image.new(size=(400, 300), mode='RGB')
draw = ImageDraw.Draw(img)

# draw white rectangle 200x100 with center in 200,150
draw.rectangle((200-100, 150-50, 200+100, 150+50), fill='white')
draw.line(((0, 150), (400, 150)), 'gray')
draw.line(((200, 0), (200, 300)), 'gray')

# find font size for text `"Hello World"` to fit in rectangle 200x100
selected_size = 1
for size in range(1, 150):
    arial = ImageFont.FreeTypeFont('/home/furas/.wine/drive_c/windows/Fonts/arial.ttf', size=size)
    w, h = arial.getsize("Hello World")  # older versions
    #left, top, right, bottom = arial.getbbox("Hello World")  # needs PIL 8.0.0
    #w = right - left
    #h = bottom - top
    print(w, h)
    
    if w > 200 or h > 100:
        break
        
    selected_size = size

    print(arial.size)
    
# draw text in center of rectangle 200x100        
arial = ImageFont.FreeTypeFont('/home/furas/.wine/drive_c/windows/Fonts/arial.ttf', size=selected_size)

#draw.text((200-w//2, 150-h//2), "Hello World", fill='black', font=arial)  # older versions
#img.save('center-older.png')

draw.text((200, 150), "Hello World", fill='black', anchor='mm', font=arial)
img.save('center-newer.png')

img.show()

---

结果(200-w//2, 150-h//2)与getbbox()

结果(200-w//2, 150-h//2)与getsize()

结果anchor='mm'（带getbbox()或getsize()）