fix numpy bug

修复numpy版本导致的bug
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/14816762
* fix numpy bug
This commit is contained in:
biwen.lbw
2023-11-28 17:17:29 +08:00
committed by wenmeng.zwm
parent 5ba9fd2307
commit a19fe73afb

View File

@@ -767,6 +767,7 @@ def align_img(img, lm, lm3D, mask=None, target_size=224., rescale_factor=102.):
# calculate translation and scale factors using 5 facial landmarks and standard landmarks of a 3D face
t, s = POS(lm5p.transpose(), lm3D.transpose())
t = t.squeeze()
s = rescale_factor / s
# processing the image