Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Hangz-nju-cuhk/Talking-Face_PC-AVS
/ functions
Functions
326 in github.com/Hangz-nju-cuhk/Talking-Face_PC-AVS
⨍
Functions
326
◇
Types & classes
66
↓ 16 callers
Method
__init__
(self, kernel, pad, upsample_factor=1)
models/networks/generator.py:132
↓ 15 callers
Method
select_frames
(self, in_obj_ts)
models/av_model.py:424
↓ 12 callers
Function
find_network_using_name
(target_network_name, filename)
models/networks/__init__.py:10
↓ 9 callers
Function
create_network
(cls, opt)
models/networks/__init__.py:36
↓ 8 callers
Method
generate_fake
(self, id_feature, pose_feature)
models/av_model.py:441
↓ 8 callers
Function
mkdir
(path)
scripts/prepare_testing_files.py:11
↓ 6 callers
Method
forward_feature
(self, x)
models/networks/encoder.py:82
↓ 6 callers
Method
load_network
(self, network, network_label, epoch_label)
models/av_model.py:765
↓ 6 callers
Method
load_separately
(self, network, network_label, opt)
models/av_model.py:738
↓ 6 callers
Method
merge_mouthpose
(self, mouth_feature, headpose_feature, embed_headpose=False)
models/av_model.py:454
↓ 6 callers
Method
save
(self)
util/html.py:52
↓ 5 callers
Function
_unsqueeze_ft
add new dimensions at the front and the tail
models/networks/sync_batchnorm/batchnorm.py:32
↓ 5 callers
Method
add_images
(self, ims, txts, links, width=512)
util/html.py:40
↓ 5 callers
Method
forward
(self, data, mode)
models/av_model.py:66
↓ 4 callers
Method
__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
models/networks/discriminator.py:156
↓ 4 callers
Method
__init__
(self, opt)
models/networks/encoder.py:51
↓ 4 callers
Method
_make_layer
(self, block, planes, blocks, stride=1)
models/networks/audio_network.py:53
↓ 4 callers
Method
downsample
(self, input)
models/networks/discriminator.py:42
↓ 4 callers
Method
encode_audiosync_feature
(self, spectrogram)
models/av_model.py:334
↓ 4 callers
Method
encode_noid_feature
(self, augmented)
models/av_model.py:413
↓ 4 callers
Method
l2_norm
(self, x)
models/networks/loss.py:157
↓ 3 callers
Method
__init__
(self, gan_mode, target_real_label=1.0, target_fake_label=0.0, tensor=torch.FloatTensor, opt=
models/networks/loss.py:12
↓ 3 callers
Method
__init__
(self, channel, reduction=8)
models/networks/audio_network.py:185
↓ 3 callers
Method
add_header
(self, str)
util/html.py:32
↓ 3 callers
Method
compute_sync_loss
(self, image_content_feature, audio_content_feature, G_losses, name='')
models/av_model.py:350
↓ 3 callers
Function
conv3x3
3x3 convolution with padding
models/networks/FAN_feature_extractor.py:7
↓ 3 callers
Method
discriminate_single
(self, single_image, netD)
models/av_model.py:712
↓ 3 callers
Method
encode_identity_feature
(self, input_img)
models/av_model.py:374
↓ 3 callers
Function
find_dataset_using_name
(dataset_name)
data/__init__.py:6
↓ 3 callers
Method
get
(self)
models/networks/sync_batchnorm/comm.py:22
↓ 3 callers
Function
get_out_channel
(layer)
models/networks/architecture.py:94
↓ 3 callers
Function
make_kernel
(k)
models/networks/generator.py:78
↓ 3 callers
Method
modify_commandline_options
(parser, is_train)
models/networks/base_network.py:10
↓ 3 callers
Function
proc_frames
(src_path, dst_path)
scripts/prepare_testing_files.py:16
↓ 3 callers
Method
put
(self, result)
models/networks/sync_batchnorm/comm.py:16
↓ 3 callers
Function
upfirdn2d
(input, kernel, up=1, down=1, pad=(0, 0))
models/networks/generator.py:66
↓ 2 callers
Method
__init__
(self)
models/networks/FAN_feature_extractor.py:108
↓ 2 callers
Method
_check_input_dim
(self, input)
models/networks/sync_batchnorm/batchnorm.py:194
↓ 2 callers
Function
_sum_ft
sum over the first and last dimention
models/networks/sync_batchnorm/batchnorm.py:27
↓ 2 callers
Method
audio_to_spectrogram
(self, wav)
config/AudioConfig.py:175
↓ 2 callers
Method
compute_GAN_Feat_loss
(self, pred_fake, pred_real)
models/av_model.py:595
↓ 2 callers
Method
convert_visuals_to_numpy
(self, visuals)
util/visualizer.py:156
↓ 2 callers
Method
discriminate
(self, input, fake_image, real_image, netD)
models/av_model.py:696
↓ 2 callers
Method
encode_ref_noid
(self, input_img)
models/av_model.py:388
↓ 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
models/networks/sync_batchnorm/replicate.py:19
↓ 2 callers
Function
find_model_using_name
(model_name)
models/__init__.py:3
↓ 2 callers
Function
fused_leaky_relu
(input, bias, negative_slope=0.2, scale=2 ** 0.5)
models/networks/generator.py:9
↓ 2 callers
Method
get_hop_size
(self)
config/AudioConfig.py:112
↓ 2 callers
Function
get_nonspade_norm_layer
(opt, norm_type='instance')
models/networks/architecture.py:92
↓ 2 callers
Method
get_target_tensor
(self, input, target_is_real)
models/networks/loss.py:34
↓ 2 callers
Method
get_zero_tensor
(self, input)
models/networks/loss.py:46
↓ 2 callers
Method
initialize
(self, opt)
data/base_dataset.py:16
↓ 2 callers
Method
l2_sim
(self, feature1, feature2)
models/networks/loss.py:161
↓ 2 callers
Method
load_img
(self, image_path, M=None, crop=True, crop_len=16)
data/voxtest_dataset.py:26
↓ 2 callers
Method
load_one_frame
(self, frame_ind, video_path, M=None, crop=True)
data/voxtest_dataset.py:136
↓ 2 callers
Method
load_pretrain
(self)
models/networks/encoder.py:77
↓ 2 callers
Method
loss
(self, input, target_is_real, for_discriminator=True)
models/networks/loss.py:52
↓ 2 callers
Function
mkdir
create a single empty directory if it didn't exist Parameters: path (str) -- a single directory path
models/networks/util.py:165
↓ 2 callers
Function
mkdir
(path)
util/util.py:128
↓ 2 callers
Method
option_file_path
(self, opt, makedir=False)
options/base_options.py:200
↓ 2 callers
Method
perspective_transform
(self, img, crop_size=224, pers_size=10, enlarge_size=-10)
data/base_dataset.py:63
↓ 2 callers
Method
read_audio
(self, audio_path)
config/AudioConfig.py:166
↓ 2 callers
Method
reset_parameters
(self)
models/networks/sync_batchnorm/batchnorm_reimpl.py:33
↓ 2 callers
Function
scatter
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 callers
Method
start_and_end_indices
(self, quantized, silence_threshold=2)
config/AudioConfig.py:92
↓ 2 callers
Method
to_Tensor
(self, img)
data/base_dataset.py:19
↓ 2 callers
Method
use_gpu
(self)
models/av_model.py:803
↓ 1 callers
Method
__data_parallel_replicate__
(self, ctx, copy_id)
models/networks/sync_batchnorm/batchnorm.py:85
↓ 1 callers
Method
__init__
(self, requires_grad=False)
models/networks/architecture.py:13
↓ 1 callers
Method
__init__
Args: master_callback: a callback to be invoked after having collected messages from slave devices.
models/networks/sync_batchnorm/comm.py:57
↓ 1 callers
Method
_amp_to_db
(self, x)
config/AudioConfig.py:153
↓ 1 callers
Method
_blur_and_sharp
(self, img)
data/base_dataset.py:36
↓ 1 callers
Method
_build_mel_basis
(self)
config/AudioConfig.py:147
↓ 1 callers
Method
_color_transfer
(self, img)
data/base_dataset.py:50
↓ 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.
models/networks/sync_batchnorm/batchnorm.py:118
↓ 1 callers
Method
_forward
(self, level, inp)
models/networks/FAN_feature_extractor.py:80
↓ 1 callers
Method
_generate_network
(self, level)
models/networks/FAN_feature_extractor.py:68
↓ 1 callers
Method
_linear_to_mel
(self, spectrogram)
config/AudioConfig.py:142
↓ 1 callers
Method
_lws_processor
(self)
config/AudioConfig.py:119
↓ 1 callers
Method
_normalize
(self, S)
config/AudioConfig.py:160
↓ 1 callers
Method
_reshape
(self, img, crop_size)
data/base_dataset.py:76
↓ 1 callers
Method
add_table
(self, border=1)
util/html.py:36
↓ 1 callers
Function
affine_align_3landmarks
(landmarks, M)
scripts/align_68.py:27
↓ 1 callers
Function
affine_align_img
(img, M, crop_size=224)
scripts/align_68.py:22
↓ 1 callers
Function
align_folder
(folder_path, folder_save_path)
scripts/align_68.py:46
↓ 1 callers
Function
atoi
(text)
util/util.py:133
↓ 1 callers
Method
audio_normalize
(self, samples, desired_rms=0.1, eps=1e-4)
config/AudioConfig.py:42
↓ 1 callers
Method
compute_D_input_nc
(self, opt)
models/networks/discriminator.py:99
↓ 1 callers
Method
compute_discriminator_loss
(self, input_img, real_image, augmented, spectrogram, netD, labels, lambda_D=1)
models/av_model.py:609
↓ 1 callers
Method
compute_encoder_loss
(self, input_img, real_image, spectrogram, labels)
models/av_model.py:299
↓ 1 callers
Method
compute_generator_loss
(self, input_img, real_image, augmented, spectrogram, netD, labels, no_ganFeat_
models/av_model.py:494
↓ 1 callers
Method
compute_pose_diff
(self, pose_feature, ref_noid_feature)
models/av_model.py:396
↓ 1 callers
Function
create_model
(opt)
models/__init__.py:31
↓ 1 callers
Method
create_single_discriminator
(self, opt)
models/networks/discriminator.py:34
↓ 1 callers
Method
divide_pred
(self, pred)
models/av_model.py:722
↓ 1 callers
Method
forward_feature
(self, input)
models/networks/vision_network.py:28
↓ 1 callers
Method
forward_feature
(self, x)
models/networks/encoder.py:25
↓ 1 callers
Method
frame2audio_indexs
(self, frame_inds)
data/voxtest_dataset.py:51
↓ 1 callers
Method
gather_options
(self)
options/base_options.py:150
↓ 1 callers
Method
generate_spectrogram_magphase
(self, audio)
config/AudioConfig.py:47
next →
1–100 of 326, ranked by callers