Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/OpenTalker/StyleHEAT
/ functions
Functions
1,100 in github.com/OpenTalker/StyleHEAT
⨍
Functions
1,100
◇
Types & classes
196
↓ 1 callers
Method
forward_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 callers
Method
forward_train
(self)
third_part/PerceptualSimilarity/models/dist_model.py:191
↓ 1 callers
Method
freeze_models
(self)
models/styleheat/styleheat.py:41
↓ 1 callers
Function
ganspace_edit
(latents, pca, edit_directions)
models/hfgi/editing.py:14
↓ 1 callers
Method
gather_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 callers
Function
gauss_fcn
(x)
third_part/decalib/utils/util.py:298
↓ 1 callers
Function
gaussian
(window_size, sigma)
third_part/decalib/utils/util.py:297
↓ 1 callers
Function
get_blocks
(num_layers)
models/e4e/helpers.py:30
↓ 1 callers
Function
get_cfg_defaults
Get a yacs CfgNode object with default values for my_project.
third_part/decalib/utils/config.py:117
↓ 1 callers
Function
get_colors
()
utils/common.py:35
↓ 1 callers
Function
get_config
(config_file)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_config.py:5
↓ 1 callers
Function
get_date_uid
Generate a unique id based on date. Returns: str: Return uid string, e.g. '20171122171307111552'.
utils/logging.py:9
↓ 1 callers
Function
get_delta
(pca, latent, idx, strength)
models/hfgi/editing.py:5
↓ 1 callers
Function
get_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 callers
Method
get_image_dir
Return the directory that stores images
third_part/Deep3DFaceRecon_pytorch/util/html.py:35
↓ 1 callers
Function
get_image_feature
(img_path, files_list, model_path, epoch, gpu_id)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval_ijbc.py:154
↓ 1 callers
Method
get_landmark
get landmark with dlib :return: np.array shape=(68, 2)
utils/video_preprocess/crop_videos.py:41
↓ 1 callers
Function
get_landmark_bbox
(lm, scale=1)
utils/landmark.py:4
↓ 1 callers
Function
get_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 callers
Function
get_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 callers
Function
get_mbf
(fp16, num_features)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/mobilefacenet.py:129
↓ 1 callers
Function
get_model_optimizer_and_scheduler_with_pretrain
(opt)
utils/trainer.py:45
↓ 1 callers
Function
get_network
(net_type: str)
loss/lpips.py:111
↓ 1 callers
Function
get_obj_by_name
Finds the python object with the given name.
dnnlib/util.py:287
↓ 1 callers
Function
get_optimizer
(opt_opt, net)
utils/trainer.py:184
↓ 1 callers
Function
get_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 callers
Function
get_option_setter
Return the static method <modify_commandline_options> of the model class.
third_part/Deep3DFaceRecon_pytorch/models/__init__.py:49
↓ 1 callers
Function
get_state_dict
(net_type: str = 'alex', version: str = '0.1')
loss/lpips.py:8
↓ 1 callers
Function
get_trainer
(opt, net_G, net_G_ema, opt_G, sch_G, train_dataset)
utils/trainer.py:36
↓ 1 callers
Method
get_video_index
(self, videos)
data/audio_dataset.py:86
↓ 1 callers
Method
get_video_index
(self, videos)
data/video_dataset.py:55
↓ 1 callers
Method
get_video_index
(self, videos)
data/vox_dataset.py:55
↓ 1 callers
Function
get_wra_data_path
(video_dir)
utils/video_preprocess/crop_videos.py:229
↓ 1 callers
Method
group_by_key
(self, video_list, key)
data/audio_dataset.py:94
↓ 1 callers
Method
group_by_key
(self, video_list, key)
data/video_dataset.py:64
↓ 1 callers
Method
group_by_key
(self, video_list, key)
data/vox_dataset.py:64
↓ 1 callers
Function
image2template_feature
(img_feats=None, templates=None, medias=None)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/eval_ijbc.py:212
↓ 1 callers
Function
image2template_feature
(img_feats=None, templates=None, medias=None)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_ijbc.py:99
↓ 1 callers
Function
img_padding
(img, box)
third_part/Deep3DFaceRecon_pytorch/util/preprocess.py:71
↓ 1 callers
Function
inference
(weight, name, img)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/inference.py:11
↓ 1 callers
Method
init_dataset
(self, val_targets, data_dir, image_size)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_callbacks.py:36
↓ 1 callers
Function
init_dist
r"""Initialize distributed training
utils/distributed.py:6
↓ 1 callers
Function
init_logging
(opt)
utils/logging.py:17
↓ 1 callers
Function
init_logging
(rank, models_root)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/utils/utils_logging.py:30
↓ 1 callers
Method
initialize
Define the common options that are used in both training and test.
third_part/Deep3DFaceRecon_pytorch/options/base_options.py:26
↓ 1 callers
Function
intuitive_edit
(generator, data)
inference.py:252
↓ 1 callers
Function
iresnet100
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:179
↓ 1 callers
Function
iresnet18
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:164
↓ 1 callers
Function
iresnet200
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:184
↓ 1 callers
Function
iresnet2060
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet2060.py:175
↓ 1 callers
Function
iresnet34
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:169
↓ 1 callers
Function
iresnet50
(pretrained=False, progress=True, **kwargs)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/backbones/iresnet.py:174
↓ 1 callers
Function
is_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 callers
Function
is_url
Determine whether the given object is a valid URL string.
dnnlib/util.py:378
↓ 1 callers
Function
lbs
Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB
third_part/decalib/models/lbs.py:140
↓ 1 callers
Method
load_PCA_model
(self)
loss/iterative_inversion_loss.py:66
↓ 1 callers
Method
load_checkpoint
(self, opt, path_dic)
models/styleheat/styleheat.py:46
↓ 1 callers
Method
load_checkpoint
(self, path_dic)
models/hfgi/hfgi.py:46
↓ 1 callers
Method
load_checkpoint
(self)
third_part/decalib/trainer.py:87
↓ 1 callers
Method
load_checkpoint
(self)
third_part/decalib/utils/trainer.py:78
↓ 1 callers
Function
load_flow
(path)
utils/flow_util.py:69
↓ 1 callers
Function
load_lm3d
(bfm_folder)
third_part/Deep3DFaceRecon_pytorch/util/load_mats.py:105
↓ 1 callers
Method
load_mask
(self, maskpath, h, w)
third_part/decalib/datasets/vox.py:153
↓ 1 callers
Method
load_mask
(self, maskpath, h, w)
third_part/decalib/datasets/train_datasets.py:192
↓ 1 callers
Method
load_mask
(self, maskpath, h, w)
third_part/decalib/datasets/ethnicity.py:114
↓ 1 callers
Method
load_mask
(self, maskpath, h, w)
third_part/decalib/datasets/vggface.py:109
↓ 1 callers
Method
load_mask
(self, maskpath, h, w)
third_part/decalib/datasets/vggface.py:218
↓ 1 callers
Method
load_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 callers
Method
load_next_video
(self, image_path=None, video_name=None, if_audio=True)
data/audio_dataset.py:270
↓ 1 callers
Function
load_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 callers
Method
load_weights
Function to load luatorch pretrained Args: path: path for the luatorch pretrained
third_part/decalib/utils/lossfunc.py:491
↓ 1 callers
Function
main
()
train.py:27
↓ 1 callers
Function
main
()
inference.py:314
↓ 1 callers
Function
main
(args)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/train.py:21
↓ 1 callers
Function
main
(args)
third_part/Deep3DFaceRecon_pytorch/models/arcface_torch/onnx_ijbc.py:169
↓ 1 callers
Function
makeColorwheel
()
utils/flow_util.py:93
↓ 1 callers
Function
make_cache_dir_path
(*paths: str)
dnnlib/util.py:124
↓ 1 callers
Function
make_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 callers
Function
make_logging_dir
r"""Create the logging directory Args: logdir (str): Log directory name
utils/logging.py:27
↓ 1 callers
Method
name
(self)
third_part/PerceptualSimilarity/models/dist_model.py:26
↓ 1 callers
Function
ndc_projection
(x=0.1, n=1.0, f=50.0)
third_part/Deep3DFaceRecon_pytorch/util/nvdiffrast.py:15
↓ 1 callers
Function
normalize_activation
(x, eps=1e-10)
loss/lpips.py:30
↓ 1 callers
Function
obtain_seq_index
(index, num_frames)
inference.py:149
↓ 1 callers
Method
obtain_seq_index
(self, index, num_frames)
data/audio_dataset.py:396
↓ 1 callers
Method
obtain_seq_index
(self, index, num_frames)
data/video_dataset.py:152
↓ 1 callers
Method
obtain_seq_index
(self, index, num_frames)
data/inference_dataset.py:155
↓ 1 callers
Method
obtain_seq_index
(self, index, num_frames)
data/vox_dataset.py:145
↓ 1 callers
Method
optimize_inverse
(self, x, save_path=None)
models/hfgi/hfgi.py:130
↓ 1 callers
Method
parallelize
(self, convert_sync_batchnorm=True)
third_part/Deep3DFaceRecon_pytorch/models/base_model.py:107
↓ 1 callers
Method
parse
Parse our options, create checkpoints directory suffix, and set up gpu device.
third_part/Deep3DFaceRecon_pytorch/options/base_options.py:121
↓ 1 callers
Function
parse_args
()
train.py:12
↓ 1 callers
Function
parse_args
()
inference.py:288
↓ 1 callers
Method
patch_extraction
(self, featmaps)
third_part/decalib/utils/lossfunc.py:388
↓ 1 callers
Method
patch_extraction
(self, featmaps)
third_part/decalib/utils/lossfunc.py:569
↓ 1 callers
Function
perspective_projection
(focal, center)
third_part/Deep3DFaceRecon_pytorch/models/bfm.py:11
↓ 1 callers
Function
plot_kpts
Draw 68 key points Args: image: the input image kpt: (68, 3).
third_part/decalib/utils/util.py:579
↓ 1 callers
Function
points2bbox
(points, points_scale=None)
third_part/decalib/utils/tensor_cropper.py:13
↓ 1 callers
Method
prepare
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 callers
Method
prepare_data
(self)
third_part/decalib/trainer.py:361
↓ 1 callers
Method
prepare_data
(self)
third_part/decalib/utils/trainer.py:247
← previous
next →
301–400 of 1,100, ranked by callers