(img, landmark=None, borderMode=cv2.BORDER_REPLICATE, flags=cv2.INTER_LINEAR)
| 58 | |
| 59 | |
| 60 | def affine_alignSDK(img, landmark=None, borderMode=cv2.BORDER_REPLICATE, flags=cv2.INTER_LINEAR): |
| 61 | M = None |
| 62 | dst_points = np.array([[70.745156, 111.9996875], [108.23625, 111.9996875], [89.700875, 153.514375]], dtype=np.float32) |
| 63 | default_shape = (178,218) |
| 64 | lmk = landmark.astype(np.float32) |
| 65 | src_points = np.array([ |
| 66 | lmk[0], lmk[1], |
| 67 | (lmk[3] + lmk[4]) / 2 |
| 68 | ], dtype=np.float32) |
| 69 | # src_points = get_trans_points(landmarks) |
| 70 | trans_matrix = kestrel_get_similar_matrix(src_points, dst_points) |
| 71 | trans_matrix = np.concatenate((trans_matrix, [[0, 0, 1]]), axis=0) |
| 72 | # print(rotate_points_106) |
| 73 | return cv2.warpPerspective(img, trans_matrix, default_shape, borderMode, flags=flags), trans_matrix |
nothing calls this directly
no test coverage detected