MCPcopy Create free account

hub / github.com/Hangz-nju-cuhk/Talking-Face_PC-AVS / functions

Functions326 in github.com/Hangz-nju-cuhk/Talking-Face_PC-AVS

↓ 16 callersMethod__init__
(self, kernel, pad, upsample_factor=1)
models/networks/generator.py:132
↓ 15 callersMethodselect_frames
(self, in_obj_ts)
models/av_model.py:424
↓ 12 callersFunctionfind_network_using_name
(target_network_name, filename)
models/networks/__init__.py:10
↓ 9 callersFunctioncreate_network
(cls, opt)
models/networks/__init__.py:36
↓ 8 callersMethodgenerate_fake
(self, id_feature, pose_feature)
models/av_model.py:441
↓ 8 callersFunctionmkdir
(path)
scripts/prepare_testing_files.py:11
↓ 6 callersMethodforward_feature
(self, x)
models/networks/encoder.py:82
↓ 6 callersMethodload_network
(self, network, network_label, epoch_label)
models/av_model.py:765
↓ 6 callersMethodload_separately
(self, network, network_label, opt)
models/av_model.py:738
↓ 6 callersMethodmerge_mouthpose
(self, mouth_feature, headpose_feature, embed_headpose=False)
models/av_model.py:454
↓ 6 callersMethodsave
(self)
util/html.py:52
↓ 5 callersFunction_unsqueeze_ft
add new dimensions at the front and the tail
models/networks/sync_batchnorm/batchnorm.py:32
↓ 5 callersMethodadd_images
(self, ims, txts, links, width=512)
util/html.py:40
↓ 5 callersMethodforward
(self, data, mode)
models/av_model.py:66
↓ 4 callersMethod__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
models/networks/discriminator.py:156
↓ 4 callersMethod__init__
(self, opt)
models/networks/encoder.py:51
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/networks/audio_network.py:53
↓ 4 callersMethoddownsample
(self, input)
models/networks/discriminator.py:42
↓ 4 callersMethodencode_audiosync_feature
(self, spectrogram)
models/av_model.py:334
↓ 4 callersMethodencode_noid_feature
(self, augmented)
models/av_model.py:413
↓ 4 callersMethodl2_norm
(self, x)
models/networks/loss.py:157
↓ 3 callersMethod__init__
(self, gan_mode, target_real_label=1.0, target_fake_label=0.0, tensor=torch.FloatTensor, opt=
models/networks/loss.py:12
↓ 3 callersMethod__init__
(self, channel, reduction=8)
models/networks/audio_network.py:185
↓ 3 callersMethodadd_header
(self, str)
util/html.py:32
↓ 3 callersMethodcompute_sync_loss
(self, image_content_feature, audio_content_feature, G_losses, name='')
models/av_model.py:350
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
models/networks/FAN_feature_extractor.py:7
↓ 3 callersMethoddiscriminate_single
(self, single_image, netD)
models/av_model.py:712
↓ 3 callersMethodencode_identity_feature
(self, input_img)
models/av_model.py:374
↓ 3 callersFunctionfind_dataset_using_name
(dataset_name)
data/__init__.py:6
↓ 3 callersMethodget
(self)
models/networks/sync_batchnorm/comm.py:22
↓ 3 callersFunctionget_out_channel
(layer)
models/networks/architecture.py:94
↓ 3 callersFunctionmake_kernel
(k)
models/networks/generator.py:78
↓ 3 callersMethodmodify_commandline_options
(parser, is_train)
models/networks/base_network.py:10
↓ 3 callersFunctionproc_frames
(src_path, dst_path)
scripts/prepare_testing_files.py:16
↓ 3 callersMethodput
(self, result)
models/networks/sync_batchnorm/comm.py:16
↓ 3 callersFunctionupfirdn2d
(input, kernel, up=1, down=1, pad=(0, 0))
models/networks/generator.py:66
↓ 2 callersMethod__init__
(self)
models/networks/FAN_feature_extractor.py:108
↓ 2 callersMethod_check_input_dim
(self, input)
models/networks/sync_batchnorm/batchnorm.py:194
↓ 2 callersFunction_sum_ft
sum over the first and last dimention
models/networks/sync_batchnorm/batchnorm.py:27
↓ 2 callersMethodaudio_to_spectrogram
(self, wav)
config/AudioConfig.py:175
↓ 2 callersMethodcompute_GAN_Feat_loss
(self, pred_fake, pred_real)
models/av_model.py:595
↓ 2 callersMethodconvert_visuals_to_numpy
(self, visuals)
util/visualizer.py:156
↓ 2 callersMethoddiscriminate
(self, input, fake_image, real_image, netD)
models/av_model.py:696
↓ 2 callersMethodencode_ref_noid
(self, input_img)
models/av_model.py:388
↓ 2 callersFunctionexecute_replication_callbacks
Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked w
models/networks/sync_batchnorm/replicate.py:19
↓ 2 callersFunctionfind_model_using_name
(model_name)
models/__init__.py:3
↓ 2 callersFunctionfused_leaky_relu
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
models/networks/generator.py:9
↓ 2 callersMethodget_hop_size
(self)
config/AudioConfig.py:112
↓ 2 callersFunctionget_nonspade_norm_layer
(opt, norm_type='instance')
models/networks/architecture.py:92
↓ 2 callersMethodget_target_tensor
(self, input, target_is_real)
models/networks/loss.py:34
↓ 2 callersMethodget_zero_tensor
(self, input)
models/networks/loss.py:46
↓ 2 callersMethodinitialize
(self, opt)
data/base_dataset.py:16
↓ 2 callersMethodl2_sim
(self, feature1, feature2)
models/networks/loss.py:161
↓ 2 callersMethodload_img
(self, image_path, M=None, crop=True, crop_len=16)
data/voxtest_dataset.py:26
↓ 2 callersMethodload_one_frame
(self, frame_ind, video_path, M=None, crop=True)
data/voxtest_dataset.py:136
↓ 2 callersMethodload_pretrain
(self)
models/networks/encoder.py:77
↓ 2 callersMethodloss
(self, input, target_is_real, for_discriminator=True)
models/networks/loss.py:52
↓ 2 callersFunctionmkdir
create a single empty directory if it didn't exist Parameters: path (str) -- a single directory path
models/networks/util.py:165
↓ 2 callersFunctionmkdir
(path)
util/util.py:128
↓ 2 callersMethodoption_file_path
(self, opt, makedir=False)
options/base_options.py:200
↓ 2 callersMethodperspective_transform
(self, img, crop_size=224, pers_size=10, enlarge_size=-10)
data/base_dataset.py:63
↓ 2 callersMethodread_audio
(self, audio_path)
config/AudioConfig.py:166
↓ 2 callersMethodreset_parameters
(self)
models/networks/sync_batchnorm/batchnorm_reimpl.py:33
↓ 2 callersFunctionscatter
r""" Slices tensors into approximately equal chunks and distributes them across given GPUs. Duplicates references to objects that are not
models/networks/sync_batchnorm/scatter_gather.py:5
↓ 2 callersMethodstart_and_end_indices
(self, quantized, silence_threshold=2)
config/AudioConfig.py:92
↓ 2 callersMethodto_Tensor
(self, img)
data/base_dataset.py:19
↓ 2 callersMethoduse_gpu
(self)
models/av_model.py:803
↓ 1 callersMethod__data_parallel_replicate__
(self, ctx, copy_id)
models/networks/sync_batchnorm/batchnorm.py:85
↓ 1 callersMethod__init__
(self, requires_grad=False)
models/networks/architecture.py:13
↓ 1 callersMethod__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
models/networks/sync_batchnorm/comm.py:57
↓ 1 callersMethod_amp_to_db
(self, x)
config/AudioConfig.py:153
↓ 1 callersMethod_blur_and_sharp
(self, img)
data/base_dataset.py:36
↓ 1 callersMethod_build_mel_basis
(self)
config/AudioConfig.py:147
↓ 1 callersMethod_color_transfer
(self, img)
data/base_dataset.py:50
↓ 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.
models/networks/sync_batchnorm/batchnorm.py:118
↓ 1 callersMethod_forward
(self, level, inp)
models/networks/FAN_feature_extractor.py:80
↓ 1 callersMethod_generate_network
(self, level)
models/networks/FAN_feature_extractor.py:68
↓ 1 callersMethod_linear_to_mel
(self, spectrogram)
config/AudioConfig.py:142
↓ 1 callersMethod_lws_processor
(self)
config/AudioConfig.py:119
↓ 1 callersMethod_normalize
(self, S)
config/AudioConfig.py:160
↓ 1 callersMethod_reshape
(self, img, crop_size)
data/base_dataset.py:76
↓ 1 callersMethodadd_table
(self, border=1)
util/html.py:36
↓ 1 callersFunctionaffine_align_3landmarks
(landmarks, M)
scripts/align_68.py:27
↓ 1 callersFunctionaffine_align_img
(img, M, crop_size=224)
scripts/align_68.py:22
↓ 1 callersFunctionalign_folder
(folder_path, folder_save_path)
scripts/align_68.py:46
↓ 1 callersFunctionatoi
(text)
util/util.py:133
↓ 1 callersMethodaudio_normalize
(self, samples, desired_rms=0.1, eps=1e-4)
config/AudioConfig.py:42
↓ 1 callersMethodcompute_D_input_nc
(self, opt)
models/networks/discriminator.py:99
↓ 1 callersMethodcompute_discriminator_loss
(self, input_img, real_image, augmented, spectrogram, netD, labels, lambda_D=1)
models/av_model.py:609
↓ 1 callersMethodcompute_encoder_loss
(self, input_img, real_image, spectrogram, labels)
models/av_model.py:299
↓ 1 callersMethodcompute_generator_loss
(self, input_img, real_image, augmented, spectrogram, netD, labels, no_ganFeat_
models/av_model.py:494
↓ 1 callersMethodcompute_pose_diff
(self, pose_feature, ref_noid_feature)
models/av_model.py:396
↓ 1 callersFunctioncreate_model
(opt)
models/__init__.py:31
↓ 1 callersMethodcreate_single_discriminator
(self, opt)
models/networks/discriminator.py:34
↓ 1 callersMethoddivide_pred
(self, pred)
models/av_model.py:722
↓ 1 callersMethodforward_feature
(self, input)
models/networks/vision_network.py:28
↓ 1 callersMethodforward_feature
(self, x)
models/networks/encoder.py:25
↓ 1 callersMethodframe2audio_indexs
(self, frame_inds)
data/voxtest_dataset.py:51
↓ 1 callersMethodgather_options
(self)
options/base_options.py:150
↓ 1 callersMethodgenerate_spectrogram_magphase
(self, audio)
config/AudioConfig.py:47
next →1–100 of 326, ranked by callers