| | 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 |
| |
|
