MCPcopy Create free account

hub / github.com/EvelynFan/FaceFormer / functions

Functions33 in github.com/EvelynFan/FaceFormer

↓ 3 callersFunctionenc_dec_mask
(device, dataset, T, S)
faceformer.py:34
↓ 3 callersMethodpredict
(self, audio, template, one_hot)
faceformer.py:137
↓ 2 callersFunction_compute_mask_indices
( shape: Tuple[int, int], mask_prob: float, mask_length: int, attention_mask: Optional[torch.T
wav2vec.py:14
↓ 2 callersFunctionget_dataloaders
(args)
data_loader.py:96
↓ 2 callersFunctionget_slopes_power_of_2
(n)
faceformer.py:12
↓ 1 callersMethod__init__
(self, args)
faceformer.py:63
↓ 1 callersFunctioncount_parameters
(model)
main.py:103
↓ 1 callersFunctiongenerate_vertices_npy
(args,face_vert_mmap,data2array_verts)
vocaset/process_voca_data.py:15
↓ 1 callersFunctiongenerate_wav
(args,raw_audio)
vocaset/process_voca_data.py:27
↓ 1 callersFunctionget_slopes
(n)
faceformer.py:11
↓ 1 callersFunctioninit_biased_mask
(n_head, max_seq_len, period)
faceformer.py:10
↓ 1 callersFunctionlinear_interpolation
(features, input_fps, output_fps, output_len=None)
wav2vec.py:64
↓ 1 callersFunctionload_data
(args)
vocaset/process_voca_data.py:9
↓ 1 callersFunctionmain
()
render.py:129
↓ 1 callersFunctionmain
()
main.py:106
↓ 1 callersFunctionmain
()
demo.py:180
↓ 1 callersFunctionmain
()
vocaset/process_voca_data.py:35
↓ 1 callersFunctionread_data
(args)
data_loader.py:39
↓ 1 callersFunctionrender_mesh_helper
(args,mesh, t_center, rot=np.zeros(3), tex_img=None, z_offset=0)
render.py:27
↓ 1 callersFunctionrender_mesh_helper
(args,mesh, t_center, rot=np.zeros(3), tex_img=None, z_offset=0)
demo.py:63
↓ 1 callersFunctionrender_sequence
(args)
demo.py:141
↓ 1 callersFunctionrender_sequence_meshes
(args,sequence_vertices, template, out_path,predicted_vertices_path,vt, ft ,tex_img)
render.py:107
↓ 1 callersFunctiontest
(args, model, test_loader,epoch)
main.py:71
↓ 1 callersFunctiontest_model
(args)
demo.py:22
↓ 1 callersFunctiontrainer
(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
Methodforward
( self, input_values, dataset, attention_mask=None, output_attentions=
wav2vec.py:75
Methodforward
(self, x)
faceformer.py:58
Methodforward
(self, audio, template, vertice, one_hot, criterion,teacher_forcing=True)
faceformer.py:91