MCPcopy Create free account

hub / github.com/OpenTalker/StyleHEAT / functions

Functions1,100 in github.com/OpenTalker/StyleHEAT

↓ 1 callersMethodforward_backward
Partial fc forward and backward with model parallel label: tensor Label tensor on each rank(GPU) features: tenso
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/partial_fc.py:159
↓ 1 callersMethodforward_train
(self)
third_part/PerceptualSimilarity/models/dist_model.py:191
↓ 1 callersMethodfreeze_models
(self)
models/styleheat/styleheat.py:41
↓ 1 callersFunctionganspace_edit
(latents, pca, edit_directions)
models/hfgi/editing.py:14
↓ 1 callersMethodgather_options
Initialize our parser with basic options(only once). Add additional model-specific and dataset-specific options. These options are def
third_part/Deep3DFaceRecon_pytorch/options/base_options.py:51
↓ 1 callersFunctiongauss_fcn
(x)
third_part/decalib/utils/util.py:298
↓ 1 callersFunctiongaussian
(window_size, sigma)
third_part/decalib/utils/util.py:297
↓ 1 callersFunctionget_blocks
(num_layers)
models/e4e/helpers.py:30
↓ 1 callersFunctionget_cfg_defaults
Get a yacs CfgNode object with default values for my_project.
third_part/decalib/utils/config.py:117
↓ 1 callersFunctionget_colors
()
utils/common.py:35
↓ 1 callersFunctionget_config
(config_file)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_config.py:5
↓ 1 callersFunctionget_date_uid
Generate a unique id based on date. Returns: str: Return uid string, e.g. '20171122171307111552'.
utils/logging.py:9
↓ 1 callersFunctionget_delta
(pca, latent, idx, strength)
models/hfgi/editing.py:5
↓ 1 callersFunctionget_gaussian_kernel2d
r"""Function that returns Gaussian filter matrix coefficients. Args: kernel_size (Tuple[int, int]): filter sizes in the x and y direction
third_part/decalib/utils/util.py:332
↓ 1 callersMethodget_image_dir
Return the directory that stores images
third_part/Deep3DFaceRecon_pytorch/util/html.py:35
↓ 1 callersFunctionget_image_feature
(img_path, files_list, model_path, epoch, gpu_id)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval_ijbc.py:154
↓ 1 callersMethodget_landmark
get landmark with dlib :return: np.array shape=(68, 2)
utils/video_preprocess/crop_videos.py:41
↓ 1 callersFunctionget_landmark_bbox
(lm, scale=1)
utils/landmark.py:4
↓ 1 callersFunctionget_laplacian_kernel2d
r"""Function that returns Gaussian filter matrix coefficients. Args: kernel_size (int): filter size should be odd. Returns:
third_part/decalib/utils/lossfunc.py:287
↓ 1 callersFunctionget_laplacian_kernel2d
r"""Function that returns Gaussian filter matrix coefficients. Args: kernel_size (int): filter size should be odd. Returns:
third_part/decalib/utils/util.py:400
↓ 1 callersFunctionget_mbf
(fp16, num_features)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:129
↓ 1 callersFunctionget_model_optimizer_and_scheduler_with_pretrain
(opt)
utils/trainer.py:45
↓ 1 callersFunctionget_network
(net_type: str)
loss/lpips.py:111
↓ 1 callersFunctionget_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:287
↓ 1 callersFunctionget_optimizer
(opt_opt, net)
utils/trainer.py:184
↓ 1 callersFunctionget_optimizer_for_params
r"""Return the scheduler object. Args: opt_opt (obj): Config for the specific optimization module (gen/dis). params (obj): Parame
utils/trainer.py:188
↓ 1 callersFunctionget_option_setter
Return the static method <modify_commandline_options> of the model class.
third_part/Deep3DFaceRecon_pytorch/models/__init__.py:49
↓ 1 callersFunctionget_state_dict
(net_type: str = 'alex', version: str = '0.1')
loss/lpips.py:8
↓ 1 callersFunctionget_trainer
(opt, net_G, net_G_ema, opt_G, sch_G, train_dataset)
utils/trainer.py:36
↓ 1 callersMethodget_video_index
(self, videos)
data/audio_dataset.py:86
↓ 1 callersMethodget_video_index
(self, videos)
data/video_dataset.py:55
↓ 1 callersMethodget_video_index
(self, videos)
data/vox_dataset.py:55
↓ 1 callersFunctionget_wra_data_path
(video_dir)
utils/video_preprocess/crop_videos.py:229
↓ 1 callersMethodgroup_by_key
(self, video_list, key)
data/audio_dataset.py:94
↓ 1 callersMethodgroup_by_key
(self, video_list, key)
data/video_dataset.py:64
↓ 1 callersMethodgroup_by_key
(self, video_list, key)
data/vox_dataset.py:64
↓ 1 callersFunctionimage2template_feature
(img_feats=None, templates=None, medias=None)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval_ijbc.py:212
↓ 1 callersFunctionimage2template_feature
(img_feats=None, templates=None, medias=None)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_ijbc.py:99
↓ 1 callersFunctionimg_padding
(img, box)
third_part/Deep3DFaceRecon_pytorch/util/preprocess.py:71
↓ 1 callersFunctioninference
(weight, name, img)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/inference.py:11
↓ 1 callersMethodinit_dataset
(self, val_targets, data_dir, image_size)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:36
↓ 1 callersFunctioninit_dist
r"""Initialize distributed training
utils/distributed.py:6
↓ 1 callersFunctioninit_logging
(opt)
utils/logging.py:17
↓ 1 callersFunctioninit_logging
(rank, models_root)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_logging.py:30
↓ 1 callersMethodinitialize
Define the common options that are used in both training and test.
third_part/Deep3DFaceRecon_pytorch/options/base_options.py:26
↓ 1 callersFunctionintuitive_edit
(generator, data)
inference.py:252
↓ 1 callersFunctioniresnet100
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:179
↓ 1 callersFunctioniresnet18
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:164
↓ 1 callersFunctioniresnet200
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:184
↓ 1 callersFunctioniresnet2060
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet2060.py:175
↓ 1 callersFunctioniresnet34
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:169
↓ 1 callersFunctioniresnet50
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:174
↓ 1 callersFunctionis_top_level_function
Determine whether the given object is a top-level function, i.e., defined at module scope using 'def'.
dnnlib/util.py:312
↓ 1 callersFunctionis_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:378
↓ 1 callersFunctionlbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
third_part/decalib/models/lbs.py:140
↓ 1 callersMethodload_PCA_model
(self)
loss/iterative_inversion_loss.py:66
↓ 1 callersMethodload_checkpoint
(self, opt, path_dic)
models/styleheat/styleheat.py:46
↓ 1 callersMethodload_checkpoint
(self, path_dic)
models/hfgi/hfgi.py:46
↓ 1 callersMethodload_checkpoint
(self)
third_part/decalib/trainer.py:87
↓ 1 callersMethodload_checkpoint
(self)
third_part/decalib/utils/trainer.py:78
↓ 1 callersFunctionload_flow
(path)
utils/flow_util.py:69
↓ 1 callersFunctionload_lm3d
(bfm_folder)
third_part/Deep3DFaceRecon_pytorch/util/load_mats.py:105
↓ 1 callersMethodload_mask
(self, maskpath, h, w)
third_part/decalib/datasets/vox.py:153
↓ 1 callersMethodload_mask
(self, maskpath, h, w)
third_part/decalib/datasets/train_datasets.py:192
↓ 1 callersMethodload_mask
(self, maskpath, h, w)
third_part/decalib/datasets/ethnicity.py:114
↓ 1 callersMethodload_mask
(self, maskpath, h, w)
third_part/decalib/datasets/vggface.py:109
↓ 1 callersMethodload_mask
(self, maskpath, h, w)
third_part/decalib/datasets/vggface.py:218
↓ 1 callersMethodload_networks
Load all the networks from the disk. Parameters: epoch (int) -- current epoch; used in the file name '%s_net_%s.pth' % (epoch, na
third_part/Deep3DFaceRecon_pytorch/models/base_model.py:244
↓ 1 callersMethodload_next_video
(self, image_path=None, video_name=None, if_audio=True)
data/audio_dataset.py:270
↓ 1 callersFunctionload_state_dict
Set parameters converted from Caffe models authors of VGGFace2 provide. See https://www.robots.ox.ac.uk/~vgg/data/vgg_face2/. Arguments:
third_part/decalib/models/frnet.py:156
↓ 1 callersMethodload_weights
Function to load luatorch pretrained Args: path: path for the luatorch pretrained
third_part/decalib/utils/lossfunc.py:491
↓ 1 callersFunctionmain
()
train.py:27
↓ 1 callersFunctionmain
()
inference.py:314
↓ 1 callersFunctionmain
(args)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/train.py:21
↓ 1 callersFunctionmain
(args)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_ijbc.py:169
↓ 1 callersFunctionmakeColorwheel
()
utils/flow_util.py:93
↓ 1 callersFunctionmake_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callersFunctionmake_coordinate_grid
r"""obtain coordinate grid with the same size as the flow filed. Args: flow (tensor): Flow field obtained by the model Returns:
utils/flow_util.py:20
↓ 1 callersFunctionmake_logging_dir
r"""Create the logging directory Args: logdir (str): Log directory name
utils/logging.py:27
↓ 1 callersMethodname
(self)
third_part/PerceptualSimilarity/models/dist_model.py:26
↓ 1 callersFunctionndc_projection
(x=0.1, n=1.0, f=50.0)
third_part/Deep3DFaceRecon_pytorch/util/nvdiffrast.py:15
↓ 1 callersFunctionnormalize_activation
(x, eps=1e-10)
loss/lpips.py:30
↓ 1 callersFunctionobtain_seq_index
(index, num_frames)
inference.py:149
↓ 1 callersMethodobtain_seq_index
(self, index, num_frames)
data/audio_dataset.py:396
↓ 1 callersMethodobtain_seq_index
(self, index, num_frames)
data/video_dataset.py:152
↓ 1 callersMethodobtain_seq_index
(self, index, num_frames)
data/inference_dataset.py:155
↓ 1 callersMethodobtain_seq_index
(self, index, num_frames)
data/vox_dataset.py:145
↓ 1 callersMethodoptimize_inverse
(self, x, save_path=None)
models/hfgi/hfgi.py:130
↓ 1 callersMethodparallelize
(self, convert_sync_batchnorm=True)
third_part/Deep3DFaceRecon_pytorch/models/base_model.py:107
↓ 1 callersMethodparse
Parse our options, create checkpoints directory suffix, and set up gpu device.
third_part/Deep3DFaceRecon_pytorch/options/base_options.py:121
↓ 1 callersFunctionparse_args
()
train.py:12
↓ 1 callersFunctionparse_args
()
inference.py:288
↓ 1 callersMethodpatch_extraction
(self, featmaps)
third_part/decalib/utils/lossfunc.py:388
↓ 1 callersMethodpatch_extraction
(self, featmaps)
third_part/decalib/utils/lossfunc.py:569
↓ 1 callersFunctionperspective_projection
(focal, center)
third_part/Deep3DFaceRecon_pytorch/models/bfm.py:11
↓ 1 callersFunctionplot_kpts
Draw 68 key points Args: image: the input image kpt: (68, 3).
third_part/decalib/utils/util.py:579
↓ 1 callersFunctionpoints2bbox
(points, points_scale=None)
third_part/decalib/utils/tensor_cropper.py:13
↓ 1 callersMethodprepare
get sampled class centers for cal softmax. label: tensor Label tensor on each rank. optimizer: opt O
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/partial_fc.py:139
↓ 1 callersMethodprepare_data
(self)
third_part/decalib/trainer.py:361
↓ 1 callersMethodprepare_data
(self)
third_part/decalib/utils/trainer.py:247
← previousnext →301–400 of 1,100, ranked by callers