utils.py

import numpy as np
import cv2

def resize_image(img, shape):
    return cv2.resize(img, (shape[1], shape[0]), interpolation=cv2.INTER_LINEAR)

def text_to_image(text, img_shape, font=cv2.FONT_HERSHEY_SIMPLEX, font_scale=3, thickness=5):
    text_size = cv2.getTextSize(text, font, font_scale, thickness)[0]
    text_x = (img_shape[1] - text_size[0]) // 2
    text_y = (img_shape[0] + text_size[1]) // 2

    img_wm = np.zeros(img_shape, dtype=np.uint8)
    cv2.putText(img_wm, text, (text_x, text_y), font, font_scale, (255, 255, 255), thickness)

    return img_wm