| import cv2 |
| import numpy as np |
| from PIL import Image |
|
|
|
|
| def resize_and_center(image, target_width, target_height): |
| img = np.array(image) |
|
|
| if img.shape[-1] == 4: |
| img = cv2.cvtColor(img, cv2.COLOR_RGBA2RGB) |
| elif len(img.shape) == 2 or img.shape[-1] == 1: |
| img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB) |
|
|
| original_height, original_width = img.shape[:2] |
|
|
| scale = min(target_height / original_height, target_width / original_width) |
| new_height = int(original_height * scale) |
| new_width = int(original_width * scale) |
|
|
| resized_img = cv2.resize(img, (new_width, new_height), |
| interpolation=cv2.INTER_CUBIC) |
|
|
| padded_img = np.ones((target_height, target_width, 3), |
| dtype=np.uint8) * 255 |
|
|
| top = (target_height - new_height) // 2 |
| left = (target_width - new_width) // 2 |
|
|
| padded_img[top:top + new_height, left:left + new_width] = resized_img |
|
|
| return Image.fromarray(padded_img) |
|
|
