TDN-M commited on
Commit
61f4eff
·
verified ·
1 Parent(s): acdb381

Update avatar.py

Browse files
Files changed (1) hide show
  1. avatar.py +4 -4
avatar.py CHANGED
@@ -181,9 +181,9 @@ class Avatar:
181
  return boxes
182
 
183
 
184
- def create_face_detection_results(self, full_frames,save_result=True):
185
- detector = FACE_DETECTION.FaceAlignment(FACE_DETECTION.LandmarksType._2D,
186
- flip_input=False, device=self.device)
187
  images=full_frames
188
  while 1:
189
  predictions = []
@@ -503,7 +503,7 @@ class Avatar:
503
  self.output_video_path+self.output_video_name
504
  )
505
  self.image_frame_num_current=0
506
- def text_to_lip_video_parallel(self, input_text, pre_base_video_frames,pre_face_detect_results):
507
 
508
  mel_chunks_from_audio, audio_segment = self.create_mel_from_audio(input_text)
509
  print(str(len(self.face_detect_img_results)))
 
181
  return boxes
182
 
183
 
184
+ def create_face_detection_results(self, full_frames, save_result=True):
185
+ detector = FaceAlignment(LandmarksType._2D,
186
+ flip_input=False, device=self.device)
187
  images=full_frames
188
  while 1:
189
  predictions = []
 
503
  self.output_video_path+self.output_video_name
504
  )
505
  self.image_frame_num_current=0
506
+ def text_to_lip_video_parallel(self, input_text, pre_base_video_frames, pre_face_detect_results, progress_bar=None):
507
 
508
  mel_chunks_from_audio, audio_segment = self.create_mel_from_audio(input_text)
509
  print(str(len(self.face_detect_img_results)))