MCPcopy Create free account

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

Functions434 in github.com/Songluchuan/TDMM-LM_data

Functionbatch_euler2matrix
(r)
tools/spectre_inverse/src/utils/rotation_converter.py:303
Functionbatch_matrix2axis
(rot_mats)
tools/spectre_inverse/src/utils/rotation_converter.py:315
Functionbatch_orth_proj
X is N x num_pquaternion_to_angle_axisoints x 3
tools/spectre_inverse/src/utils/rotation_converter.py:332
Functionbatch_orth_proj
orthgraphic projection X: 3d vertices, [bz, n_point, 3] camera: scale and translation, [bz, 3], [scale, tx, ty]
tools/spectre_inverse/src/utils/util.py:283
Functionbinary_erosion
(tensor, kernel_size=5)
tools/spectre_inverse/src/utils/util.py:478
Functionbuild_questionnaire
(config)
tools/spectre_inverse/datasets/build_datasets.py:399
Functionbuild_questionnaire_TCDTIMIT
(config)
tools/spectre_inverse/datasets/build_datasets.py:365
Functionbuild_train_val
(config, is_train=True)
tools/spectre_inverse/datasets/build_datasets.py:179
Functionbuild_train_val_MEAD
(config)
tools/spectre_inverse/datasets/build_datasets.py:295
Functionbuild_train_val_MEAD_txt
(config=None)
tools/spectre_inverse/datasets/build_datasets.py:246
Functionbuild_train_val_TCD
(config=None)
tools/spectre_inverse/datasets/build_datasets.py:432
Functioncheck_mkdir
(path)
tools/spectre_inverse/src/utils/util.py:527
Functioncheck_mkdirlist
(pathlist)
tools/spectre_inverse/src/utils/util.py:532
Functioncollate_fn
(batch)
tools/smirk_inverse/datasets/data_utils.py:46
Methodcollate_fn
(batch)
tools/spectre_inverse/src/trainer_spectre.py:684
Functioncompute_euler_angles_from_rotation_matrices
(rotation_matrices)
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:165
Functioncompute_rotation_matrix_from_ortho6d
(poses)
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:146
Functioncopy_state_dict
(cur_state_dict, pre_state_dict, prefix='', load_name=None)
tools/spectre_inverse/src/utils/util.py:507
Methodcount_parameters
(model)
tools/spectre_inverse/src/trainer_spectre.py:95
Functioncreate_RepVGG_A0
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:198
Functioncreate_RepVGG_A1
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:202
Functioncreate_RepVGG_A2
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:206
Functioncreate_RepVGG_B0
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:210
Functioncreate_RepVGG_B1
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:214
Functioncreate_RepVGG_B1g2
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:218
Functioncreate_RepVGG_B1g4
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:222
Functioncreate_RepVGG_B2
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:227
Functioncreate_RepVGG_B2g2
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:231
Functioncreate_RepVGG_B2g4
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:235
Functioncreate_RepVGG_B3
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:240
Functioncreate_RepVGG_B3g2
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:244
Functioncreate_RepVGG_B3g4
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:248
Functioncreate_RepVGG_D2se
(deploy=False)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:252
Functiondeg2rad
Function that converts angles from degrees to radians. See :class:`~torchgeometry.DegToRad` for details. Args: tensor (Tensor): Tens
tools/spectre_inverse/src/utils/rotation_converter.py:44
Functiondict2obj
(d)
tools/spectre_inverse/src/utils/util.py:555
Functiondict_tensor2npy
(tensor_dict)
tools/spectre_inverse/src/utils/util.py:581
Functiondraw_axis
(img, yaw, pitch, roll, tdx=None, tdy=None, size = 100)
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:56
Functionextract_audio
(video_path)
tools/spectre_inverse/utils/extract_frames_and_audio.py:26
Functionextract_wav
(videopath)
tools/spectre_inverse/utils/extract_wavs_LRS3.py:12
Functionface_vertices
:param vertices: [batch size, number of vertices, 3] :param faces: [batch size, number of faces, 3] :return: [batch size, number of face
tools/spectre_inverse/src/utils/util.py:229
Functionfind_dynamic_lmk_idx_and_bcoords
Compute the faces, barycentric coordinates for the dynamic landmarks To do so, we first compute the rotation of the neck around the y-axis
tools/spectre_inverse/src/models/lbs.py:34
Functionfind_dynamic_lmk_idx_and_bcoords
Compute the faces, barycentric coordinates for the dynamic landmarks To do so, we first compute the rotation of the neck around the y-axis
tools/smirk_inverse/src/FLAME/lbs.py:34
Methodfit
(self)
tools/spectre_inverse/src/trainer_spectre.py:307
Functionflip_image
purpose: flip a image given by src_image and the 2d keypoints flip_mode: 0: horizontal flip >0:
tools/spectre_inverse/src/utils/util.py:488
Methodforward
(self, vertices, faces, attributes=None, h=None, w=None)
tools/spectre_inverse/src/utils/renderer.py:56
Methodforward
-- Texture Rendering vertices: [batch_size, V, 3], vertices in world space, for calculating normals, then shading transformed
tools/spectre_inverse/src/utils/renderer.py:138
Methodforward
(self, inputs)
tools/spectre_inverse/src/models/encoders.py:31
Methodforward
(self, inputs)
tools/spectre_inverse/src/models/encoders.py:80
Methodforward
Input: shape_params: N X number of shape parameters expression_params: N X number of expression parameter
tools/spectre_inverse/src/models/FLAME.py:175
Methodforward
texcode: [batchsize, n_tex] texture: [bz, 3, 256, 256], range: 0-1
tools/spectre_inverse/src/models/FLAME.py:256
Methodforward
(self, inputs)
tools/spectre_inverse/src/models/expression_loss.py:37
Methodforward
(self, x)
tools/spectre_inverse/src/models/resnet.py:64
Methodforward
(self, x)
tools/spectre_inverse/src/models/resnet.py:98
Methodforward
(self, x)
tools/spectre_inverse/src/models/resnet.py:138
Methodforward
(self, x)
tools/spectre_inverse/src/models/resnet.py:208
Methodforward
(self, x)
tools/spectre_inverse/src/models/resnet.py:222
Methodforward
(self, x1, x2)
tools/spectre_inverse/src/models/resnet.py:240
Methodforward
(self, x)
tools/spectre_inverse/src/models/resnet.py:260
Methodforward
(self, m1, m2)
tools/spectre_inverse/6DRepNet/sixdrepnet/loss.py:12
Methodforward
(self, x)
tools/spectre_inverse/6DRepNet/sixdrepnet/model.py:36
Methodforward
(self, x)
tools/spectre_inverse/6DRepNet/sixdrepnet/model.py:95
Methodforward
(self, inputs)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:48
Methodforward
(self, x)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:182
Methodforward
(self, inputs)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/se_block.py:15
Methodforward
(self, img)
tools/smirk_inverse/src/smirk_encoder.py:34
Methodforward
(self, img)
tools/smirk_inverse/src/smirk_encoder.py:66
Methodforward
(self, img)
tools/smirk_inverse/src/smirk_encoder.py:95
Methodforward
(self, img)
tools/smirk_inverse/src/smirk_encoder.py:123
Methodforward
(self, x)
tools/smirk_inverse/src/smirk_generator.py:51
Methodforward
Forward function (with skip connections)
tools/smirk_inverse/src/smirk_generator.py:175
Methodforward
(self, vertices, cam_params, **landmarks)
tools/smirk_inverse/src/renderer/renderer.py:100
Methodforward
(self, x)
tools/smirk_inverse/src/models/MICA/arcface.py:51
Methodforward
(self, x)
tools/smirk_inverse/src/models/MICA/arcface.py:147
Methodforward
(self, images)
tools/smirk_inverse/src/models/MICA/arcface.py:177
Methodforward
(self, z)
tools/smirk_inverse/src/models/MICA/mica.py:34
Methodforward
(self, x, y)
tools/smirk_inverse/src/losses/VGGPerceptualLoss.py:23
Methodforward
(self, gen, tar, use_mean=True, metric='l2')
tools/smirk_inverse/src/losses/ExpressionLoss.py:48
Methodforward
(self, x)
tools/smirk_inverse/src/losses/resnet.py:24
Methodforward
(self, x)
tools/smirk_inverse/src/losses/resnet.py:65
Methodforward
(self, x)
tools/smirk_inverse/src/losses/resnet.py:137
Methodforward2
(self, inputs)
tools/spectre_inverse/src/models/expression_loss.py:32
Functionfreeze_module
(module, module_name=None)
tools/smirk_inverse/src/utils/utils.py:46
Functiongaussian_blur
(x, kernel_size=(3,3), sigma=(0.8,0.8))
tools/spectre_inverse/src/utils/util.py:373
Functiongenerate_triangles
(h, w, margin_x=2, margin_y=5, mask = None)
tools/spectre_inverse/src/utils/util.py:211
FunctiongetDataset
(dataset, data_dir, filename_list, transformations, train_mode = True)
tools/spectre_inverse/6DRepNet/sixdrepnet/datasets.py:306
Functionget_LRS3_test
(config)
tools/smirk_inverse/datasets/lrs3_dataset.py:71
Functionget_R
Get rotation matrix from three rotation angles (radians). right-handed. Args: angles: [3,]. x, y, z angles Returns: R: [3, 3]
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:192
Methodget_custom_L2
(self)
tools/spectre_inverse/6DRepNet/sixdrepnet/backbone/repvgg.py:68
Methodget_landmarks
Input: shape_params: N X number of shape parameters expression_params: N X number of ex
tools/smirk_inverse/src/FLAME/FLAME.py:201
Functionget_pose_params_from_mat
(mat_path)
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:90
Functionget_pt2d_from_mat
(mat_path)
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:110
Functionget_ypr_from_mat
(mat_path)
tools/spectre_inverse/6DRepNet/sixdrepnet/utils.py:100
Functionimage_to_tensor
Converts a numpy image to a tensor.
tools/smirk_inverse/src/utils/utils.py:36
Functionkaiming_leaky_init
(m)
tools/smirk_inverse/src/models/MICA/mica.py:8
Functionl2_distance
(verts1, verts2)
tools/spectre_inverse/src/utils/lossfunc.py:7
Functionlandmark_loss
(predicted_landmarks, landmarks_gt, weight=1.)
tools/spectre_inverse/src/utils/lossfunc.py:29
Functionlaplacian
(x)
tools/spectre_inverse/src/utils/util.py:438
Functionload_ResNet101Model
()
tools/spectre_inverse/src/models/resnet.py:179
Functionload_ResNet152Model
()
tools/spectre_inverse/src/models/resnet.py:184
Functionload_ResNet50Model
()
tools/spectre_inverse/src/models/resnet.py:174
← previousnext →301–400 of 434, ranked by callers