(self, face_images)
| 393 | return psi_list |
| 394 | |
| 395 | def prepare_driving_video(self, face_images): |
| 396 | print("Prepare driving video...") |
| 397 | pipeline = self.get_pipeline() |
| 398 | f_img_np = (face_images * 255).byte().numpy() |
| 399 | |
| 400 | out_list = [] |
| 401 | for f_img in f_img_np: |
| 402 | i_d = self.prepare_src_image(f_img) |
| 403 | d_info = pipeline.get_kp_info(i_d) |
| 404 | out_list.append(d_info) |
| 405 | |
| 406 | return out_list |
| 407 | |
| 408 | def calc_fe(_, x_d_new, eyes, eyebrow, wink, pupil_x, pupil_y, mouth, eee, woo, smile, |
| 409 | rotate_pitch, rotate_yaw, rotate_roll): |
no test coverage detected