johnberg commited on
Commit
96ae396
·
1 Parent(s): 5568eef

Update align_faces_parallel.py

Browse files
Files changed (1) hide show
  1. align_faces_parallel.py +2 -2
align_faces_parallel.py CHANGED
@@ -103,7 +103,7 @@ def align_face(img, predictor, output_size=256, transform_size=256):
103
  shrink = int(np.floor(qsize / output_size * 0.5))
104
  if shrink > 1:
105
  rsize = (int(np.rint(float(img.size[0]) / shrink)), int(np.rint(float(img.size[1]) / shrink)))
106
- img = img.resize(rsize, PIL.Image.ANTIALIAS)
107
  quad /= shrink
108
  qsize /= shrink
109
 
@@ -138,7 +138,7 @@ def align_face(img, predictor, output_size=256, transform_size=256):
138
  # Transform.
139
  img = img.transform((transform_size, transform_size), PIL.Image.QUAD, (quad + 0.5).flatten(), PIL.Image.BILINEAR)
140
  if output_size < transform_size:
141
- img = img.resize((output_size, output_size), PIL.Image.ANTIALIAS)
142
 
143
  # Save aligned image.
144
  return img
 
103
  shrink = int(np.floor(qsize / output_size * 0.5))
104
  if shrink > 1:
105
  rsize = (int(np.rint(float(img.size[0]) / shrink)), int(np.rint(float(img.size[1]) / shrink)))
106
+ img = img.resize(rsize, PIL.Image.LANCZOS)
107
  quad /= shrink
108
  qsize /= shrink
109
 
 
138
  # Transform.
139
  img = img.transform((transform_size, transform_size), PIL.Image.QUAD, (quad + 0.5).flatten(), PIL.Image.BILINEAR)
140
  if output_size < transform_size:
141
+ img = img.resize((output_size, output_size), PIL.Image.LANCZOS)
142
 
143
  # Save aligned image.
144
  return img