python - 如何将 "textbox"中的文本添加到图像中？

Question

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

CV2:

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()

PIL:

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 内部执行此操作？

感谢您的帮助

Answer 1

枕头有ImageFont选择可以在 text(..., font=...) 中使用的字体、字体大小。

在较新版本中，您可以使用 text((center_x, center_y),anchor='mm') 将文本在区域中居中。查看更多:Text Anchor

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

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

---

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())