Spaces:
Running
Running
Update avatar.py
Browse files
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 =
|
186 |
-
|
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)))
|