zejunyang commited on
Commit
bf49b76
·
1 Parent(s): 9a70da5
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -122,6 +122,7 @@ def audio2video(input_audio, ref_img, headpose_video=None, size=512, steps=25, l
122
  print('=====1======')
123
 
124
  ref_image_np = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
 
125
  ref_image_np = crop_face(ref_image_np, lmk_extractor)
126
  if ref_image_np is None:
127
  return None, Image.fromarray(ref_img)
 
122
  print('=====1======')
123
 
124
  ref_image_np = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
125
+ print('=====1======', ref_img.shape, ref_image_np.shape)
126
  ref_image_np = crop_face(ref_image_np, lmk_extractor)
127
  if ref_image_np is None:
128
  return None, Image.fromarray(ref_img)