Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/EvelynFan/FaceFormer
/ functions
Functions
33 in github.com/EvelynFan/FaceFormer
⨍
Functions
33
◇
Types & classes
4
↓ 3 callers
Function
enc_dec_mask
(device, dataset, T, S)
faceformer.py:34
↓ 3 callers
Method
predict
(self, audio, template, one_hot)
faceformer.py:137
↓ 2 callers
Function
_compute_mask_indices
( shape: Tuple[int, int], mask_prob: float, mask_length: int, attention_mask: Optional[torch.T
wav2vec.py:14
↓ 2 callers
Function
get_dataloaders
(args)
data_loader.py:96
↓ 2 callers
Function
get_slopes_power_of_2
(n)
faceformer.py:12
↓ 1 callers
Method
__init__
(self, args)
faceformer.py:63
↓ 1 callers
Function
count_parameters
(model)
main.py:103
↓ 1 callers
Function
generate_vertices_npy
(args,face_vert_mmap,data2array_verts)
vocaset/process_voca_data.py:15
↓ 1 callers
Function
generate_wav
(args,raw_audio)
vocaset/process_voca_data.py:27
↓ 1 callers
Function
get_slopes
(n)
faceformer.py:11
↓ 1 callers
Function
init_biased_mask
(n_head, max_seq_len, period)
faceformer.py:10
↓ 1 callers
Function
linear_interpolation
(features, input_fps, output_fps, output_len=None)
wav2vec.py:64
↓ 1 callers
Function
load_data
(args)
vocaset/process_voca_data.py:9
↓ 1 callers
Function
main
()
render.py:129
↓ 1 callers
Function
main
()
main.py:106
↓ 1 callers
Function
main
()
demo.py:180
↓ 1 callers
Function
main
()
vocaset/process_voca_data.py:35
↓ 1 callers
Function
read_data
(args)
data_loader.py:39
↓ 1 callers
Function
render_mesh_helper
(args,mesh, t_center, rot=np.zeros(3), tex_img=None, z_offset=0)
render.py:27
↓ 1 callers
Function
render_mesh_helper
(args,mesh, t_center, rot=np.zeros(3), tex_img=None, z_offset=0)
demo.py:63
↓ 1 callers
Function
render_sequence
(args)
demo.py:141
↓ 1 callers
Function
render_sequence_meshes
(args,sequence_vertices, template, out_path,predicted_vertices_path,vt, ft ,tex_img)
render.py:107
↓ 1 callers
Function
test
(args, model, test_loader,epoch)
main.py:71
↓ 1 callers
Function
test_model
(args)
demo.py:22
↓ 1 callers
Function
trainer
(args, train_loader, dev_loader, model, optimizer, criterion, epoch=100)
main.py:15
Method
__getitem__
Returns one data pair (source and target).
data_loader.py:22
Method
__init__
(self, config)
wav2vec.py:73
Method
__init__
(self, data,subjects_dict,data_type="train")
data_loader.py:15
Method
__init__
(self, d_model, dropout=0.1, period=25, max_seq_len=600)
faceformer.py:46
Method
__len__
(self)
data_loader.py:36
Method
forward
( self, input_values, dataset, attention_mask=None, output_attentions=
wav2vec.py:75
Method
forward
(self, x)
faceformer.py:58
Method
forward
(self, audio, template, vertice, one_hot, criterion,teacher_forcing=True)
faceformer.py:91