Spaces:
Build error
Build error
Update align_faces_parallel.py
Browse files- 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.
|
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.
|
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
|