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
|