SHREYSH commited on
Commit
092997a
·
verified ·
1 Parent(s): 2c62202

Update align/align_trans.py

Browse files
Files changed (1) hide show
  1. 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 is '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 is 'affine':
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: