Spaces:
Running
Running
File size: 2,207 Bytes
6f6b4ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import numbers
import numpy as np
import cv2
def center_crop(image: np.ndarray, output_size):
if isinstance(output_size, numbers.Number):
output_size = (int(output_size), int(output_size))
elif isinstance(output_size, (tuple, list)) and len(output_size) == 1:
output_size = (output_size[0], output_size[0])
image_height, image_width, c = image.shape
crop_height, crop_width = output_size
if crop_width > image_width or crop_height > image_height:
padding_ltrb = [
(crop_width - image_width) // 2 if crop_width > image_width else 0,
(crop_height - image_height) //
2 if crop_height > image_height else 0,
(crop_width - image_width + 1) //
2 if crop_width > image_width else 0,
(crop_height - image_height + 1) //
2 if crop_height > image_height else 0,
]
image = cv2.copyMakeBorder(image,
padding_ltrb[1],
padding_ltrb[3],
padding_ltrb[0],
padding_ltrb[2],
cv2.BORDER_CONSTANT,
value=(0, 0, 0))
image_height, image_width, c = image.shape
if crop_width == image_width and crop_height == image_height:
return image
crop_top = int(round((image_height - crop_height) / 2.0))
crop_left = int(round((image_width - crop_width) / 2.0))
return image[crop_top:crop_top + crop_height,
crop_left:crop_left + crop_width]
def pad(image,
left,
top,
right,
bottom,
fill: int = 0,
padding_mode: str = "constant"):
if padding_mode == 'constant':
return cv2.copyMakeBorder(image,
top,
bottom,
left,
right,
cv2.BORDER_CONSTANT,
value=(fill, fill, fill))
else:
raise UserWarning('padding mode {} not supported.'.format(padding_mode))
|