Spaces:
Runtime error
Runtime error
Update align/align_trans.py
Browse files- align/align_trans.py +2 -2
align/align_trans.py
CHANGED
@@ -276,10 +276,10 @@ def warp_and_crop_face(src_img,
|
|
276 |
raise FaceWarpException(
|
277 |
'facial_pts and reference_pts must have the same shape')
|
278 |
|
279 |
-
if align_type
|
280 |
tfm = cv2.getAffineTransform(src_pts[0:3], ref_pts[0:3])
|
281 |
# #print(('cv2.getAffineTransform() returns tfm=\n' + str(tfm))
|
282 |
-
elif align_type
|
283 |
tfm = get_affine_transform_matrix(src_pts, ref_pts)
|
284 |
# #print(('get_affine_transform_matrix() returns tfm=\n' + str(tfm))
|
285 |
else:
|
|
|
276 |
raise FaceWarpException(
|
277 |
'facial_pts and reference_pts must have the same shape')
|
278 |
|
279 |
+
if align_type == 'cv2_affine':
|
280 |
tfm = cv2.getAffineTransform(src_pts[0:3], ref_pts[0:3])
|
281 |
# #print(('cv2.getAffineTransform() returns tfm=\n' + str(tfm))
|
282 |
+
elif align_type == 'affine':
|
283 |
tfm = get_affine_transform_matrix(src_pts, ref_pts)
|
284 |
# #print(('get_affine_transform_matrix() returns tfm=\n' + str(tfm))
|
285 |
else:
|