MCPcopy Create free account

hub / github.com/Elsaam2y/DINet_optimized / functions

Functions151 in github.com/Elsaam2y/DINet_optimized

↓ 9 callersMethod__init__
(self, in_channel_image, in_channel_audio, out_dim)
models/Syncnet.py:242
↓ 9 callersMethodget
(self)
sync_batchnorm/comm.py:32
↓ 8 callersMethod__init__
(self, para_ch, feature_ch)
models/DINet.py:236
↓ 8 callersMethod__init__
(self, in_channel_image, in_channel_audio, out_dim)
models/old/Syncnet_BN.py:254
↓ 8 callersMethod__init__
(self, in_channel_image, in_channel_audio, out_dim)
models/old/Syncnet_halfBN.py:239
↓ 6 callersMethodparse_args
(self)
config/config.py:87
↓ 5 callersFunction_unsqueeze_ft
add new dimensions at the front and the tail
sync_batchnorm/batchnorm.py:56
↓ 5 callersFunctionget_scheduler
scheduler in training stage
utils/training_utils.py:6
↓ 5 callersFunctionupdate_learning_rate
(scheduler, optimizer)
utils/training_utils.py:32
↓ 3 callersMethodcompute_audio_feature
(self, audio_path)
utils/deep_speech.py:66
↓ 3 callersFunctioncompute_crop_radius
judge if crop face and compute crop radius
utils/data_processing.py:28
↓ 3 callersMethodput
(self, result)
sync_batchnorm/comm.py:26
↓ 2 callersFunction_sum_ft
sum over the first and last dimention
sync_batchnorm/batchnorm.py:51
↓ 2 callersFunctionconvert_model
Traverse the input module and its child recursively and replace all instance of torch.nn.modules.batchnorm.BatchNorm*N*d to Synchronized
sync_batchnorm/batchnorm.py:418
↓ 2 callersFunctionexecute_replication_callbacks
Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked w
sync_batchnorm/replicate.py:27
↓ 2 callersFunctionload_landmark_openface
load openface landmark from .csv file
utils/data_processing.py:7
↓ 1 callersMethod__data_parallel_replicate__
(self, ctx, copy_id)
sync_batchnorm/batchnorm.py:149
↓ 1 callersMethod__init__
(self)
utils/wav2vecDS.py:25
↓ 1 callersMethod__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
sync_batchnorm/comm.py:69
↓ 1 callersMethod__init__
(self, in_features, out_features, kernel_size=4, pool=False)
models/Discriminator.py:6
↓ 1 callersMethod_compute_mean_std
Compute the mean and standard-deviation with sum and square-sum. This method also maintains the moving average on the master device.
sync_batchnorm/batchnorm.py:182
↓ 1 callersMethod_prepare_deepspeech_net
(self, deepspeech_pb_path)
utils/deep_speech.py:20
↓ 1 callersMethodconv_audio_to_deepspeech_input_vector
( self, audio, sample_rate, num_cepstrum, num_context )
utils/deep_speech.py:32
↓ 1 callersFunctioncrop_face_according_openfaceLM
crop face according to openface landmark
data_processing.py:76
↓ 1 callersFunctionextract_audio
extract audio files from videos
data_processing.py:14
↓ 1 callersFunctionextract_deep_speech
extract deep speech feature
data_processing.py:32
↓ 1 callersFunctionextract_frames_from_video
(video_path, save_dir)
inference.py:34
↓ 1 callersFunctionextract_video_frame
extract video frames from videos
data_processing.py:50
↓ 1 callersFunctiongenerate_training_json
(crop_face_dir, deep_speech_dir, clip_length, res_json_path)
data_processing.py:164
↓ 1 callersFunctionget_data
(json_name, augment_num)
dataset/dataset_DINet_frame.py:10
↓ 1 callersFunctionget_data
(json_name, augment_num)
dataset/dataset_DINet_clip.py:10
↓ 1 callersMethodget_target_tensor
(self, input, target_is_real)
utils/training_utils.py:52
↓ 1 callersFunctionmake_coordinate_grid_3d
generate 3D coordinate grid
models/DINet.py:13
↓ 1 callersMethodmapping
(self, array)
utils/wav2vecDS.py:32
↓ 1 callersMethodregister_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
sync_batchnorm/comm.py:86
↓ 1 callersMethodreset_parameters
(self)
sync_batchnorm/batchnorm_reimpl.py:44
↓ 1 callersMethodreset_running_stats
(self)
sync_batchnorm/batchnorm_reimpl.py:40
↓ 1 callersMethodrun_master
Main entry for the master device in each forward pass. The messages were first collected from each devices (including the master devi
sync_batchnorm/comm.py:104
↓ 1 callersMethodrun_slave
(self, msg)
sync_batchnorm/comm.py:51
Method__getitem__
(self, index)
dataset/dataset_DINet_frame.py:34
Method__getitem__
(self, index)
dataset/dataset_DINet_clip.py:34
Method__getstate__
(self)
sync_batchnorm/comm.py:80
Method__init__
(self, model_path)
utils/deep_speech.py:11
Method__init__
(self, use_lsgan=True, target_real_label=1.0, target_fake_label=0.0)
utils/training_utils.py:43
Method__init__
(self)
utils/wav2vec.py:23
Method__init__
(self, input_dim, hidden_dim)
utils/wav2vecDS.py:8
Method__init__
(self, path_json, augment_num, mouth_region_size)
dataset/dataset_DINet_frame.py:24
Method__init__
(self, path_json, augment_num, mouth_region_size)
dataset/dataset_DINet_clip.py:24
Method__init__
(self)
sync_batchnorm/comm.py:21
Method__init__
(self, num_features, eps=1e-5, momentum=0.1)
sync_batchnorm/batchnorm_reimpl.py:28
Method__init__
( self, num_features, eps=1e-5, momentum=0.1, affine=True, tra
sync_batchnorm/batchnorm.py:66
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/DINet.py:36
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/DINet.py:77
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1)
models/DINet.py:118
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/DINet.py:142
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1, stride=2)
models/DINet.py:166
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/DINet.py:190
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1)
models/DINet.py:213
Method__init__
(self, source_channel, ref_channel, audio_channel)
models/DINet.py:281
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/Syncnet.py:10
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/Syncnet.py:47
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/Syncnet.py:84
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1, groups=1)
models/Syncnet.py:106
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/Syncnet.py:130
Method__init__
(self, in_features, out_features, groups=1, kernel_size=3, padding=1)
models/Syncnet.py:151
Method__init__
(self, in_channel, out_dim)
models/Syncnet.py:173
Method__init__
(self, in_channel, out_dim)
models/Syncnet.py:212
Method__init__
(self, pretrain_path)
models/Syncnet.py:273
Method__init__
( self, num_channels, block_expansion=64, num_blocks=4, max_features=512 )
models/Discriminator.py:27
Method__init__
(self, requires_grad=False)
models/VGG19.py:11
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_BN.py:10
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_BN.py:51
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_BN.py:92
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1, groups=1)
models/old/Syncnet_BN.py:116
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_BN.py:142
Method__init__
(self, in_features, out_features, groups=1, kernel_size=3, padding=1)
models/old/Syncnet_BN.py:165
Method__init__
(self, in_channel, out_dim)
models/old/Syncnet_BN.py:189
Method__init__
(self, in_channel, out_dim)
models/old/Syncnet_BN.py:228
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_halfBN.py:11
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_halfBN.py:48
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_halfBN.py:89
Method__init__
(self, in_features, out_features, kernel_size=3, padding=1, groups=1)
models/old/Syncnet_halfBN.py:111
Method__init__
(self, in_features, out_features, kernel_size, padding)
models/old/Syncnet_halfBN.py:137
Method__init__
(self, in_features, out_features, groups=1, kernel_size=3, padding=1)
models/old/Syncnet_halfBN.py:158
Method__init__
(self, in_channel, out_dim)
models/old/Syncnet_halfBN.py:178
Method__init__
(self, in_channel, out_dim)
models/old/Syncnet_halfBN.py:213
Method__init__
(self)
config/config.py:5
Method__init__
(self)
config/config.py:84
Method__init__
(self)
config/config.py:202
Method__len__
(self)
dataset/dataset_DINet_frame.py:98
Method__len__
(self)
dataset/dataset_DINet_clip.py:135
Method__setstate__
(self, state)
sync_batchnorm/comm.py:83
Method_check_input_dim
(self, input)
sync_batchnorm/batchnorm.py:273
Method_check_input_dim
(self, input)
sync_batchnorm/batchnorm.py:336
Method_check_input_dim
(self, input)
sync_batchnorm/batchnorm.py:398
Method_data_parallel_master
Reduce the sum and square-sum, compute the statistics, and broadcast it.
sync_batchnorm/batchnorm.py:159
MethodassertTensorClose
(self, x, y)
sync_batchnorm/unittest.py:17
Methodcompute_audio_feature
(self, audio_path, chunk_duration=10)
utils/wav2vec.py:29
Methodforward
(self, input, target_is_real)
utils/training_utils.py:59
Methodforward
(self, x)
utils/wav2vecDS.py:15
next →1–100 of 151, ranked by callers