forked from 170010011/fr
108 lines
4.6 KiB
Python
108 lines
4.6 KiB
Python
|
import cv2
|
|||
|
|
|
|||
|
|
|
|||
|
|
def put_text(img, text, org, font_face, font_scale, color, thickness=1, line_type=8, bottom_left_origin=False):
|
|||
|
|
"""Utility for drawing text with line breaks
|
|||
|
|
|
|||
|
|
:param img: Image.
|
|||
|
|
:param text: Text string to be drawn.
|
|||
|
|
:param org: Bottom-left corner of the first line of the text string in the image.
|
|||
|
|
:param font_face: Font type. One of FONT_HERSHEY_SIMPLEX, FONT_HERSHEY_PLAIN, FONT_HERSHEY_DUPLEX,
|
|||
|
|
FONT_HERSHEY_COMPLEX, FONT_HERSHEY_TRIPLEX, FONT_HERSHEY_COMPLEX_SMALL,
|
|||
|
|
FONT_HERSHEY_SCRIPT_SIMPLEX, or FONT_HERSHEY_SCRIPT_COMPLEX, where each of the font ID’s
|
|||
|
|
can be combined with FONT_ITALIC to get the slanted letters.
|
|||
|
|
:param font_scale: Font scale factor that is multiplied by the font-specific base size.
|
|||
|
|
:param color: Text color.
|
|||
|
|
:param thickness: Thickness of the lines used to draw a text.
|
|||
|
|
:param line_type: Line type. See the line for details.
|
|||
|
|
:param bottom_left_origin: When true, the image data origin is at the bottom-left corner.
|
|||
|
|
Otherwise, it is at the top-left corner.
|
|||
|
|
:return: None; image is modified in place
|
|||
|
|
"""
|
|||
|
|
# Break out drawing coords
|
|||
|
|
x, y = org
|
|||
|
|
|
|||
|
|
# Break text into list of text lines
|
|||
|
|
text_lines = text.split('\n')
|
|||
|
|
|
|||
|
|
# Get height of text lines in pixels (height of all lines is the same)
|
|||
|
|
_, line_height = cv2.getTextSize('', font_face, font_scale, thickness)[0]
|
|||
|
|
# Set distance between lines in pixels
|
|||
|
|
line_gap = line_height // 3
|
|||
|
|
|
|||
|
|
for i, text_line in enumerate(text_lines):
|
|||
|
|
# Find total size of text block before this line
|
|||
|
|
line_y_adjustment = i * (line_gap + line_height)
|
|||
|
|
|
|||
|
|
# Move text down from original line based on line number
|
|||
|
|
if not bottom_left_origin:
|
|||
|
|
line_y = y + line_y_adjustment
|
|||
|
|
else:
|
|||
|
|
line_y = y - line_y_adjustment
|
|||
|
|
|
|||
|
|
# Draw text
|
|||
|
|
cv2.putText(img,
|
|||
|
|
text=text_lines[i],
|
|||
|
|
org=(x, line_y),
|
|||
|
|
fontFace=font_face,
|
|||
|
|
fontScale=font_scale,
|
|||
|
|
color=color,
|
|||
|
|
thickness=thickness,
|
|||
|
|
lineType=line_type,
|
|||
|
|
bottomLeftOrigin=bottom_left_origin)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def put_centered_text(img, text, font_face, font_scale, color, thickness=1, line_type=8):
|
|||
|
|
"""Utility for drawing vertically & horizontally centered text with line breaks
|
|||
|
|
|
|||
|
|
:param img: Image.
|
|||
|
|
:param text: Text string to be drawn.
|
|||
|
|
:param font_face: Font type. One of FONT_HERSHEY_SIMPLEX, FONT_HERSHEY_PLAIN, FONT_HERSHEY_DUPLEX,
|
|||
|
|
FONT_HERSHEY_COMPLEX, FONT_HERSHEY_TRIPLEX, FONT_HERSHEY_COMPLEX_SMALL,
|
|||
|
|
FONT_HERSHEY_SCRIPT_SIMPLEX, or FONT_HERSHEY_SCRIPT_COMPLEX, where each of the font ID’s
|
|||
|
|
can be combined with FONT_ITALIC to get the slanted letters.
|
|||
|
|
:param font_scale: Font scale factor that is multiplied by the font-specific base size.
|
|||
|
|
:param color: Text color.
|
|||
|
|
:param thickness: Thickness of the lines used to draw a text.
|
|||
|
|
:param line_type: Line type. See the line for details.
|
|||
|
|
:return: None; image is modified in place
|
|||
|
|
"""
|
|||
|
|
# Save img dimensions
|
|||
|
|
img_h, img_w = img.shape[:2]
|
|||
|
|
|
|||
|
|
# Break text into list of text lines
|
|||
|
|
text_lines = text.split('\n')
|
|||
|
|
|
|||
|
|
# Get height of text lines in pixels (height of all lines is the same; width differs)
|
|||
|
|
_, line_height = cv2.getTextSize('', font_face, font_scale, thickness)[0]
|
|||
|
|
# Set distance between lines in pixels
|
|||
|
|
line_gap = line_height // 3
|
|||
|
|
|
|||
|
|
# Calculate total text block height for centering
|
|||
|
|
text_block_height = len(text_lines) * (line_height + line_gap)
|
|||
|
|
text_block_height -= line_gap # There's one less gap than lines
|
|||
|
|
|
|||
|
|
for i, text_line in enumerate(text_lines):
|
|||
|
|
# Get width of text line in pixels (height of all lines is the same)
|
|||
|
|
line_width, _ = cv2.getTextSize(text_line, font_face, font_scale, thickness)[0]
|
|||
|
|
|
|||
|
|
# Center line with image dimensions
|
|||
|
|
x = (img_w - line_width) // 2
|
|||
|
|
y = (img_h + line_height) // 2
|
|||
|
|
|
|||
|
|
# Find total size of text block before this line
|
|||
|
|
line_adjustment = i * (line_gap + line_height)
|
|||
|
|
|
|||
|
|
# Adjust line y and re-center relative to total text block height
|
|||
|
|
y += line_adjustment - text_block_height // 2 + line_gap
|
|||
|
|
|
|||
|
|
# Draw text
|
|||
|
|
cv2.putText(img,
|
|||
|
|
text=text_lines[i],
|
|||
|
|
org=(x, y),
|
|||
|
|
fontFace=font_face,
|
|||
|
|
fontScale=font_scale,
|
|||
|
|
color=color,
|
|||
|
|
thickness=thickness,
|
|||
|
|
lineType=line_type)
|