Spaces:
Running
on
A10G
Running
on
A10G
Update src/generate_facerender_batch.py
Browse files
src/generate_facerender_batch.py
CHANGED
@@ -6,7 +6,7 @@ import torch
|
|
6 |
import scipy.io as scio
|
7 |
|
8 |
def get_facerender_data(coeff_path, pic_path, first_coeff_path, audio_path,
|
9 |
-
batch_size, camera_yaw_list, camera_pitch_list, camera_roll_list,
|
10 |
expression_scale=1.0, still_mode = False):
|
11 |
|
12 |
semantic_radius = 13
|
|
|
6 |
import scipy.io as scio
|
7 |
|
8 |
def get_facerender_data(coeff_path, pic_path, first_coeff_path, audio_path,
|
9 |
+
batch_size, camera_yaw_list=[0], camera_pitch_list=[0], camera_roll_list=[0],
|
10 |
expression_scale=1.0, still_mode = False):
|
11 |
|
12 |
semantic_radius = 13
|