Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Elsaam2y/DINet_optimized
/ functions
Functions
151 in github.com/Elsaam2y/DINet_optimized
⨍
Functions
151
◇
Types & classes
61
↓ 9 callers
Method
__init__
(self, in_channel_image, in_channel_audio, out_dim)
models/Syncnet.py:242
↓ 9 callers
Method
get
(self)
sync_batchnorm/comm.py:32
↓ 8 callers
Method
__init__
(self, para_ch, feature_ch)
models/DINet.py:236
↓ 8 callers
Method
__init__
(self, in_channel_image, in_channel_audio, out_dim)
models/old/Syncnet_BN.py:254
↓ 8 callers
Method
__init__
(self, in_channel_image, in_channel_audio, out_dim)
models/old/Syncnet_halfBN.py:239
↓ 6 callers
Method
parse_args
(self)
config/config.py:87
↓ 5 callers
Function
_unsqueeze_ft
add new dimensions at the front and the tail
sync_batchnorm/batchnorm.py:56
↓ 5 callers
Function
get_scheduler
scheduler in training stage
utils/training_utils.py:6
↓ 5 callers
Function
update_learning_rate
(scheduler, optimizer)
utils/training_utils.py:32
↓ 3 callers
Method
compute_audio_feature
(self, audio_path)
utils/deep_speech.py:66
↓ 3 callers
Function
compute_crop_radius
judge if crop face and compute crop radius
utils/data_processing.py:28
↓ 3 callers
Method
put
(self, result)
sync_batchnorm/comm.py:26
↓ 2 callers
Function
_sum_ft
sum over the first and last dimention
sync_batchnorm/batchnorm.py:51
↓ 2 callers
Function
convert_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 callers
Function
execute_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 callers
Function
load_landmark_openface
load openface landmark from .csv file
utils/data_processing.py:7
↓ 1 callers
Method
__data_parallel_replicate__
(self, ctx, copy_id)
sync_batchnorm/batchnorm.py:149
↓ 1 callers
Method
__init__
(self)
utils/wav2vecDS.py:25
↓ 1 callers
Method
__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
sync_batchnorm/comm.py:69
↓ 1 callers
Method
__init__
(self, in_features, out_features, kernel_size=4, pool=False)
models/Discriminator.py:6
↓ 1 callers
Method
_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 callers
Method
_prepare_deepspeech_net
(self, deepspeech_pb_path)
utils/deep_speech.py:20
↓ 1 callers
Method
conv_audio_to_deepspeech_input_vector
( self, audio, sample_rate, num_cepstrum, num_context )
utils/deep_speech.py:32
↓ 1 callers
Function
crop_face_according_openfaceLM
crop face according to openface landmark
data_processing.py:76
↓ 1 callers
Function
extract_audio
extract audio files from videos
data_processing.py:14
↓ 1 callers
Function
extract_deep_speech
extract deep speech feature
data_processing.py:32
↓ 1 callers
Function
extract_frames_from_video
(video_path, save_dir)
inference.py:34
↓ 1 callers
Function
extract_video_frame
extract video frames from videos
data_processing.py:50
↓ 1 callers
Function
generate_training_json
(crop_face_dir, deep_speech_dir, clip_length, res_json_path)
data_processing.py:164
↓ 1 callers
Function
get_data
(json_name, augment_num)
dataset/dataset_DINet_frame.py:10
↓ 1 callers
Function
get_data
(json_name, augment_num)
dataset/dataset_DINet_clip.py:10
↓ 1 callers
Method
get_target_tensor
(self, input, target_is_real)
utils/training_utils.py:52
↓ 1 callers
Function
make_coordinate_grid_3d
generate 3D coordinate grid
models/DINet.py:13
↓ 1 callers
Method
mapping
(self, array)
utils/wav2vecDS.py:32
↓ 1 callers
Method
register_slave
Register an slave device. Args: identifier: an identifier, usually is the device id. Returns: a `SlavePipe` obj
sync_batchnorm/comm.py:86
↓ 1 callers
Method
reset_parameters
(self)
sync_batchnorm/batchnorm_reimpl.py:44
↓ 1 callers
Method
reset_running_stats
(self)
sync_batchnorm/batchnorm_reimpl.py:40
↓ 1 callers
Method
run_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 callers
Method
run_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
Method
assertTensorClose
(self, x, y)
sync_batchnorm/unittest.py:17
Method
compute_audio_feature
(self, audio_path, chunk_duration=10)
utils/wav2vec.py:29
Method
forward
(self, input, target_is_real)
utils/training_utils.py:59
Method
forward
(self, x)
utils/wav2vecDS.py:15
next →
1–100 of 151, ranked by callers