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