MCPcopy Create free account

hub / github.com/Songluchuan/TDMM-LM_data / functions

Functions434 in github.com/Songluchuan/TDMM-LM_data

↓ 1 callersMethodscheduler_step
(self)
tools/smirk_inverse/src/base_trainer.py:103
↓ 1 callersMethodset_freeze_status
(self, config, batch_idx, epoch_idx)
tools/smirk_inverse/src/base_trainer.py:258
↓ 1 callersFunctionset_rasterizer
(type = 'pytorch3d')
tools/spectre_inverse/src/utils/renderer.py:24
↓ 1 callersMethodsetup_losses
(self)
tools/smirk_inverse/src/base_trainer.py:77
↓ 1 callersMethodstep1
(self, batch)
tools/smirk_inverse/src/smirk_trainer.py:34
↓ 1 callersMethodstep2
(self, encoder_output, batch, batch_idx, phase='train')
tools/smirk_inverse/src/smirk_trainer.py:184
↓ 1 callersMethodswitch_to_deploy
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:125
↓ 1 callersMethodtraining_step
(self, batch, batch_nb)
tools/spectre_inverse/src/utils/trainer.py:101
↓ 1 callersFunctiontransform_mat
Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of
tools/spectre_inverse/src/models/lbs.py:308
↓ 1 callersFunctiontransform_mat
Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of
tools/smirk_inverse/src/FLAME/lbs.py:308
↓ 1 callersFunctiontriangle_area
(vertices)
tools/smirk_inverse/src/utils/masking.py:41
↓ 1 callersMethodunfreeze_encoder
(self)
tools/smirk_inverse/src/smirk_trainer.py:339
↓ 1 callersMethodvalidation_step
(self)
tools/spectre_inverse/src/utils/trainer.py:194
↓ 1 callersFunctionvertices2joints
Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array tha
tools/spectre_inverse/src/models/lbs.py:230
↓ 1 callersFunctionvertices2joints
Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array tha
tools/smirk_inverse/src/FLAME/lbs.py:230
Method__getitem__
(self, index)
tools/spectre_inverse/datasets/datasets.py:48
Method__getitem__
(self, index)
tools/spectre_inverse/datasets/build_datasets.py:58
Method__getitem__
(self, index)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:37
Method__getitem__
(self, index)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:93
Method__getitem__
(self, index)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:137
Method__getitem__
(self, index)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:192
Method__getitem__
(self, index)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:244
Method__getitem__
(self, index)
tools/smirk_inverse/datasets/base_dataset.py:102
Method__getitem_aux__
(self, index)
tools/smirk_inverse/datasets/ffhq_dataset.py:14
Method__getitem_aux__
(self, index)
tools/smirk_inverse/datasets/celeba_dataset.py:15
Method__getitem_aux__
(self, index)
tools/smirk_inverse/datasets/mead_dataset.py:17
Method__getitem_aux__
(self, index)
tools/smirk_inverse/datasets/mead_sides_dataset.py:15
Method__init__
(self)
tools/spectre_inverse/utils/run_av_hubert.py:15
Method__init__
(self, model, config=None, device='cuda:0')
tools/spectre_inverse/src/trainer_spectre.py:27
Method__init__
(self, config=None, device='cuda')
tools/spectre_inverse/src/spectre.py:32
Method__init__
use fixed raster_settings for rendering faces
tools/spectre_inverse/src/utils/renderer.py:40
Method__init__
(self, crop_size, scale=[1,1], trans_scale = 0.)
tools/spectre_inverse/src/utils/tensor_cropper.py:86
Method__init__
(self, **kwargs)
tools/spectre_inverse/src/utils/util.py:568
Method__init__
(self, model, config=None, device='cuda:0')
tools/spectre_inverse/src/utils/trainer.py:44
Method__init__
(self, outsize, cfg)
tools/spectre_inverse/src/models/encoders.py:9
Method__init__
(self, **kwargs)
tools/spectre_inverse/src/models/FLAME.py:33
Method__init__
(self, config)
tools/spectre_inverse/src/models/FLAME.py:225
Method__init__
(self)
tools/spectre_inverse/src/models/expression_loss.py:24
Method__init__
(self, block, layers, num_classes=1000)
tools/spectre_inverse/src/models/resnet.py:24
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
tools/spectre_inverse/src/models/resnet.py:85
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
tools/spectre_inverse/src/models/resnet.py:128
Method__init__
(self, in_channels, out_channels)
tools/spectre_inverse/src/models/resnet.py:197
Method__init__
(self, in_channels, out_channels)
tools/spectre_inverse/src/models/resnet.py:215
Method__init__
(self, in_channels, out_channels)
tools/spectre_inverse/src/models/resnet.py:256
Method__init__
(self, data_list, landmarks_path, cfg, test=False)
tools/spectre_inverse/datasets/datasets.py:12
Method__init__
(self, data_list, cfg, test=False)
tools/spectre_inverse/datasets/build_datasets.py:13
Method__init__
(self, eps=1e-7)
tools/spectre_inverse/6DRepNet/sixdrepnet/loss.py:8
Method__init__
(self, data_dir, filename_path, transform, img_ext='.jpg', annot_ext='.mat', image_mode='RGB')
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:25
Method__init__
(self, data_dir, filename_path, transform, img_ext='.jpg', annot_ext='.txt', image_mode='RGB')
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:80
Method__init__
(self, data_dir, filename_path, transform, img_ext='.jpg', annot_ext='.txt', image_mode='RGB')
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:124
Method__init__
(self, data_dir, filename_path, transform, image_mode='RGB', train_mode=True)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:178
Method__init__
(self, data_dir, filename_path, transform, img_ext='.jpg', annot_ext='.mat', image_mode='RGB')
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:232
Method__init__
Constructs the SixDRepNet instance with all necessary attributes. Parameters ---------- gpu:id : int
tools/spectre_inverse/6DRepNet/sixdrepnet/regressor.py:16
Method__init__
(self, block, layers, fc_layers=1)
tools/spectre_inverse/6DRepNet/sixdrepnet/model.py:49
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, pad
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:18
Method__init__
(self, input_channels, internal_neurons)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/se_block.py:9
Method__init__
(self, config)
tools/smirk_inverse/src/smirk_trainer.py:13
Method__init__
(self, config)
tools/smirk_inverse/src/base_trainer.py:13
Method__init__
(self, n_shape=300)
tools/smirk_inverse/src/smirk_encoder.py:49
Method__init__
(self, n_exp=50)
tools/smirk_inverse/src/smirk_encoder.py:77
Method__init__
(self, n_exp=50, n_shape=300)
tools/smirk_inverse/src/smirk_encoder.py:114
Method__init__
(self, in_channels=3, out_channels=1, init_features=16, res_blocks=3)
tools/smirk_inverse/src/smirk_generator.py:8
Method__init__
(self, render_full_head=False, obj_filename='assets/head_template.obj')
tools/smirk_inverse/src/renderer/renderer.py:50
Method__init__
(self, **kwargs)
tools/smirk_inverse/src/FLAME/FLAME.py:40
Method__init__
(self, flame_model_path='assets/FLAME2020/generic_model.pkl', flame_lmk_embedding_path='asse
tools/smirk_inverse/src/FLAME/FLAME.py:51
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1)
tools/smirk_inverse/src/models/MICA/arcface.py:35
Method__init__
(self, pretrained_path=None, **kwargs)
tools/smirk_inverse/src/models/MICA/arcface.py:165
Method__init__
(self, z_dim, map_hidden_dim, map_output_dim, hidden=2)
tools/smirk_inverse/src/models/MICA/mica.py:15
Method__init__
(self)
tools/smirk_inverse/src/losses/VGGPerceptualLoss.py:9
Method__init__
(self)
tools/smirk_inverse/src/losses/ExpressionLoss.py:27
Method__init__
(self, inplanes, planes, stride=1, downsample=None)
tools/smirk_inverse/src/losses/resnet.py:14
Method__init__
(self, inplanes, planes, stride=1, downsample=None, emoca_specific=False)
tools/smirk_inverse/src/losses/resnet.py:46
Method__init__
(self, dataset_sizes, dataset_ratios, batch_size, n_samples)
tools/smirk_inverse/datasets/mixed_dataset_sampler.py:8
Method__init__
(self, data_list, config, test=False)
tools/smirk_inverse/datasets/base_dataset.py:30
Method__init__
(self, data_list, config, test=False)
tools/smirk_inverse/datasets/lrs3_dataset.py:9
Method__init__
(self, data_list, config, test=False)
tools/smirk_inverse/datasets/ffhq_dataset.py:9
Method__init__
(self, data_list, config, test=False)
tools/smirk_inverse/datasets/celeba_dataset.py:7
Method__init__
(self, data_list, config, test=False)
tools/smirk_inverse/datasets/mead_dataset.py:10
Method__init__
(self, data_list, config, test=False)
tools/smirk_inverse/datasets/mead_sides_dataset.py:8
Method__iter__
(self)
tools/smirk_inverse/datasets/mixed_dataset_sampler.py:28
Method__len__
(self)
tools/spectre_inverse/datasets/datasets.py:45
Method__len__
(self)
tools/spectre_inverse/datasets/build_datasets.py:55
Method__len__
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:74
Method__len__
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:118
Method__len__
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:173
Method__len__
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:226
Method__len__
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:302
Method__len__
(self)
tools/smirk_inverse/datasets/mixed_dataset_sampler.py:53
Method__len__
(self)
tools/smirk_inverse/datasets/base_dataset.py:98
Method__len__
(self)
tools/smirk_inverse/datasets/lrs3_dataset.py:13
Method__len__
(self)
tools/smirk_inverse/datasets/celeba_dataset.py:12
Method__len__
(self)
tools/smirk_inverse/datasets/mead_dataset.py:14
Method__len__
(self)
tools/smirk_inverse/datasets/mead_sides_dataset.py:12
Method_vertices2landmarks
Calculates landmarks by barycentric interpolation Input: vertices: torch.tensor NxVx3, dtype = torch.float32
tools/spectre_inverse/src/models/FLAME.py:137
Method_vertices2landmarks
Calculates landmarks by barycentric interpolation Input: vertices: torch.tensor NxVx3, dtype = torch.float32
tools/smirk_inverse/src/FLAME/FLAME.py:163
Methodadd_SHlight
sh_coeff: [bz, 9, 3]
tools/smirk_inverse/src/renderer/renderer.py:209
Methodadd_pointlight
vertices: [bz, nv, 3] lights: [bz, nlight, 6] returns: shading: [bz, nv, 3]
tools/smirk_inverse/src/renderer/renderer.py:224
Functionangle2matrix
get rotation matrix from three rotation angles(degree). right-handed. Args: angles: [batch_size, 3] tensor containing X, Y, and Z angles.
tools/spectre_inverse/src/utils/util.py:447
Functionbatch_axis2euler
(theta)
tools/spectre_inverse/src/utils/rotation_converter.py:325
Functionbatch_euler2axis
(r)
tools/spectre_inverse/src/utils/rotation_converter.py:300
← previousnext →201–300 of 434, ranked by callers