OpenCV/Pillow – 画像にテキストを描画する方法

import cv2
from IPython.display import Image, display


def imshow(img):
    """ndarray 配列をインラインで Notebook 上に表示する。"""
    ret, encoded = cv2.imencode(".jpg", img)
    display(Image(encoded))

import cv2
import numpy as np

# 描画用の画像を生成する。
img = np.zeros((100, 300, 3), dtype=np.uint8)

cv2.putText(
    img,
    "Hello World",  # テキスト
    (0, 30),  # テキストの位置 (ベースラインの左端)
    fontFace=cv2.FONT_HERSHEY_SIMPLEX,  # フォントの種類
    fontScale=1.0,  # フォントのスケール
    color=(255, 255, 255),  # フォントの色
    thickness=2,  # フォントの太さ
    lineType=cv2.LINE_AA,  # 描画方法
)
imshow(img)

img = np.zeros((100, 300, 3), dtype=np.uint8)

text = "Hello World"  # 描画する文字
fontface = cv2.FONT_HERSHEY_SIMPLEX  # フォントの種類
fontscale = 1.0  # 文字のスケール
thickness = 2  # 文字の太さ
x, y = 50, 50  # ベースラインの始点

# 文字列を描画した際の大きさを取得する。

(w, h), baseline = cv2.getTextSize(text, fontface, fontscale, thickness)
print(f"size: ({w}, {h}), baseline: {baseline}")

# 文字を囲む矩形を描画する。
cv2.rectangle(img, (x, y - h), (x + w, y + baseline), (0, 0, 255), thickness)

# ベースラインを描画する。
cv2.line(img, (x, y), (x + w, y), (0, 255, 255), thickness)

# 文字列を描画する。
cv2.putText(img, text, (x, y), fontface, fontscale, (255, 255, 255), thickness)
imshow(img)

import cv2
import numpy as np

img = np.zeros((100, 300, 3), dtype=np.uint8)

text_h = 20
text = "Hello World"  # 描画する文字
fontface = cv2.FONT_HERSHEY_SIMPLEX  # フォントの種類
fontscale = cv2.getFontScaleFromHeight(fontface, text_h)  # 文字のスケール
thickness = 2  # 文字の太さ
x, y = 50, 50  # ベースラインの始点

# 文字列を描画した際の大きさを取得する。

(text_w, text_h), baseline = cv2.getTextSize(text, fontface, fontscale, thickness)
print(f"size: ({text_w}, {text_h}), baseline: {baseline}")

# 文字を囲む矩形を描画する。
cv2.rectangle(img, (x, y - text_h), (x + text_w, y + baseline), (0, 0, 255), thickness)

# ベースラインを描画する。
cv2.line(img, (x, y), (x + text_w, y), (0, 255, 255), thickness)

# 文字列を描画する。
cv2.putText(img, text, (x, y), fontface, fontscale, (255, 255, 255), thickness)
imshow(img)

import cv2
import numpy as np

# 描画用の画像を生成する。
img = np.zeros((100, 300, 3), dtype=np.uint8)

cv2.putText(
    img,
    "日本語",  # テキスト
    (0, 30),  # テキストの左下の位置
    fontFace=cv2.FONT_HERSHEY_SIMPLEX,  # フォントの種類
    fontScale=1.0,  # フォントのスケール
    color=(255, 255, 255),  # フォントの色
    thickness=2,  # フォントの太さ
    lineType=cv2.LINE_AA,  # 描画方法
)
imshow(img)

import cv2
import numpy as np
from PIL import Image as PILImage, ImageDraw, ImageFont


def put_text(
    img,
    text,
    org,
    fontname,
    fontsize=10,
    color=(0, 0, 0),
):
    # PIL の Image に変換する。
    img = PILImage.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))

    font = ImageFont.truetype(fontname, size=fontsize)
    draw = ImageDraw.Draw(img, mode="RGBA")
    draw.text(org, text, fill=color, font=font)

    return cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)


def get_text_size(img, text, org, fontname, fontsize=10):
    # PIL の Image に変換する。
    img = PILImage.fromarray(cv2.cvtColor(img, cv2.COLOR_RGB2BGR))

    font = ImageFont.truetype(fontname, size=fontsize)
    draw = ImageDraw.Draw(img, mode="RGBA")
    text_box = draw.textbbox(org, text, font=font)

    return tuple(text_box[2:])


# 描画用の画像を生成する。
img = np.zeros((100, 300, 3), dtype=np.uint8)

# Ubuntu の場合 (IPA フォント)
# インストールされていない場合は、sudo apt install -y fonts-ipafont でインストールする。
font_name = "ipagp.ttf"
# Windows の場合
# font_name = "meiryo.ttc"
text = "日本語"
position = (0, 0)
fontsize = 30

# テキストのサイズを取得する。
text_w, text_h = get_text_size(img, text, position, font_name, fontsize)
cv2.rectangle(img, position, (text_w, text_h), (255, 255, 255), -1)
print(f"text size: ({text_w}, {text_h})")


# テキストを描画する。
img = put_text(
    img,
    text,
    position,
    font_name,
    fontsize,
    color=(255, 0, 0),  # RGB で指定
)

imshow(img)