|
import numpy as np |
|
import cv2 |
|
|
|
|
|
def crop_face(img, lmk_extractor, expand=1.5): |
|
result = lmk_extractor(img) |
|
|
|
if result is None: |
|
return None |
|
|
|
H, W, _ = img.shape |
|
lmks = result['lmks'] |
|
lmks[:, 0] *= W |
|
lmks[:, 1] *= H |
|
|
|
x_min = np.min(lmks[:, 0]) |
|
x_max = np.max(lmks[:, 0]) |
|
y_min = np.min(lmks[:, 1]) |
|
y_max = np.max(lmks[:, 1]) |
|
|
|
width = x_max - x_min |
|
height = y_max - y_min |
|
|
|
if width*height >= W*H*0.15: |
|
if W == H: |
|
return img |
|
size = min(H, W) |
|
offset = int((max(H, W) - size)/2) |
|
if size == H: |
|
return img[:, offset:-offset] |
|
else: |
|
return img[offset:-offset, :] |
|
else: |
|
center_x = x_min + width / 2 |
|
center_y = y_min + height / 2 |
|
|
|
width *= expand |
|
height *= expand |
|
|
|
size = max(width, height) |
|
|
|
x_min = int(center_x - size / 2) |
|
x_max = int(center_x + size / 2) |
|
y_min = int(center_y - size / 2) |
|
y_max = int(center_y + size / 2) |
|
|
|
top = max(0, -y_min) |
|
bottom = max(0, y_max - img.shape[0]) |
|
left = max(0, -x_min) |
|
right = max(0, x_max - img.shape[1]) |
|
img = cv2.copyMakeBorder(img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=0) |
|
|
|
cropped_img = img[y_min + top:y_max + top, x_min + left:x_max + left] |
|
|
|
return cropped_img |
|
|
