ykani commited on
Commit
0f9ddb3
·
unverified ·
1 Parent(s): 549cd6f

fix: ensure upper bond does not go below zero in landmark extraction (#329)

Browse files
Files changed (1) hide show
  1. musetalk/utils/preprocessing.py +2 -1
musetalk/utils/preprocessing.py CHANGED
@@ -118,7 +118,8 @@ def get_landmark_and_bbox(img_list,upperbondrange =0):
118
  if upperbondrange != 0:
119
  half_face_coord[1] = upperbondrange+half_face_coord[1] #手动调整 + 向下(偏29) - 向上(偏28)
120
  half_face_dist = np.max(face_land_mark[:,1]) - half_face_coord[1]
121
- upper_bond = half_face_coord[1]-half_face_dist
 
122
 
123
  f_landmark = (np.min(face_land_mark[:, 0]),int(upper_bond),np.max(face_land_mark[:, 0]),np.max(face_land_mark[:,1]))
124
  x1, y1, x2, y2 = f_landmark
 
118
  if upperbondrange != 0:
119
  half_face_coord[1] = upperbondrange+half_face_coord[1] #手动调整 + 向下(偏29) - 向上(偏28)
120
  half_face_dist = np.max(face_land_mark[:,1]) - half_face_coord[1]
121
+ min_upper_bond = 0
122
+ upper_bond = max(min_upper_bond, half_face_coord[1] - half_face_dist)
123
 
124
  f_landmark = (np.min(face_land_mark[:, 0]),int(upper_bond),np.max(face_land_mark[:, 0]),np.max(face_land_mark[:,1]))
125
  x1, y1, x2, y2 = f_landmark