daddyjin commited on
Commit
869f7cd
1 Parent(s): 0bc6ace

edit face3d preprocess.py

Browse files
Demo_TFR_Pirenderer/src/face3d/util/preprocess.py CHANGED
@@ -98,7 +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
- 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
 
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
+ trans_params = np.array([w0, h0, s, t[0][0], t[1][0] ])
103
 
104
  return trans_params, img_new, lm_new, mask_new