Spaces:
Runtime error
Runtime error
edit face3d preprocess.py
Browse files
Demo_TFR_Pirenderer/src/face3d/util/preprocess.py
CHANGED
@@ -98,6 +98,7 @@ def align_img(img, lm, lm3D, mask=None, target_size=224., rescale_factor=102.):
|
|
98 |
|
99 |
# processing the image
|
100 |
img_new, lm_new, mask_new = resize_n_crop_img(img, lm, t, s, target_size=target_size, mask=mask)
|
|
|
101 |
trans_params = np.array([w0, h0, s, t[0], t[1]])
|
102 |
|
103 |
return trans_params, img_new, lm_new, mask_new
|
|
|
98 |
|
99 |
# processing the image
|
100 |
img_new, lm_new, mask_new = resize_n_crop_img(img, lm, t, s, target_size=target_size, mask=mask)
|
101 |
+
print([w0, h0, s, t[0], t[1]])
|
102 |
trans_params = np.array([w0, h0, s, t[0], t[1]])
|
103 |
|
104 |
return trans_params, img_new, lm_new, mask_new
|