MCPcopy Create free account

hub / github.com/MingwuZheng/ImFace / functions

Functions203 in github.com/MingwuZheng/ImFace

⨍ Functions 203◇ Types & classes 20
↓ 9 callersFunctionconvert_sdf_samples_to_mesh
Convert sdf samples to .ply with color-coded template coordinates This function adapted from: https://github.com/RobotLocomotion/spartan
utils/summary.py:114
↓ 9 callersMethodinference_by_batch
(self, coords, exp_embedding, id_embedding, device, points_per_inference=163840, int_idx=True,
model/imface.py:223
↓ 7 callersMethodget_sdf_in_batches
(self, query_points, use_depth_buffer=False, sample_count=11, batch_size=1000000, r
data_preprocess/surface_point_cloud.py:69
↓ 6 callersMethodshow
(self)
data_preprocess/surface_point_cloud.py:184
↓ 6 callersFunctionwarp
:param xyz: (B,N,3) :param deformation: (B,N,6)/(B,N,3) :param warp_type: 'translation' or 'se3' :return: warped xyz
model/imface.py:109
↓ 4 callersMethod__init__
(self)
model/modules.py:40
↓ 4 callersMethodinference
(self, coords_ori, exp_embedding, id_embedding, int_idx=True, return_fix=False, return_coords=False)
model/imface.py:258
↓ 3 callersMethod__init__
(self, config, id_num, exp_num, kpt_num, template_kpts=None, initial_std=0.01)
model/imface.py:141
↓ 3 callersFunctionget_keypoints
(bnds, keypoint_type)
data_loader/NormalDataset.py:13
↓ 3 callersFunctionget_keypoints
(bnds, keypoint_type)
data_loader/OneSampleDataset.py:13
↓ 3 callersMethodget_template_kpts
(self)
data_loader/NormalDataset.py:97
↓ 3 callersFunctionnearest_neighbor
Find the nearest (Euclidean) neighbor in dst for each point in src Input: src: Nxm array of points dst: Nxm array of points
sketches/evaluate.py:54
↓ 2 callersFunction_register_generic
(module_dict, module_name, module)
utils/registry.py:4
↓ 2 callersFunctionbest_fit_transform
Calculates the least-squares best-fit transform that maps corresponding points A to B in m spatial dimensions Input: A: Nxm numpy array
sketches/evaluate.py:10
↓ 2 callersFunctioncompute_recon_error
(recon_path, gt_path, num_pts=150000, facetor_to_mm=100)
utils/summary.py:42
↓ 2 callersFunctionconvert_sdf_with_correspondence_color_to_ply
(model, exp_embedding, id_embedding, int_idx, sdf_3d, ply_filename_out,
utils/summary.py:193
↓ 2 callersMethodget_num_frequencies_nyquist
(self, samples)
model/modules.py:153
↓ 2 callersMethodget_random_surface_points
(self, count, use_scans=True)
data_preprocess/surface_point_cloud.py:28
↓ 2 callersMethodget_sdf
(self, query_points, use_depth_buffer=False, sample_count=11, return_gradients=False)
data_preprocess/surface_point_cloud.py:35
↓ 2 callersFunctionget_subdict
(dictionary, key=None)
model/modules.py:9
↓ 2 callersFunctionget_surface_point_cloud
(mesh, surface_point_method='scan', bounding_radius=None, scan_count=100, scan_res
data_preprocess/surface_point_cloud.py:199
↓ 2 callersFunctionicp
The Iterative Closest Point method: finds best-fit transform that maps points A on to points B Input: A: Nxm numpy array of source mD
sketches/evaluate.py:93
↓ 2 callersFunctionnorm_cdf
(x)
model/modules.py:250
↓ 2 callersMethodsample_on_surface
(self, number_of_points=500000, use_scans=True, sign_method='normal', normal_sample_
data_preprocess/surface_point_cloud.py:165
↓ 2 callersFunctionskew
Build a skew matrix ("cross product matrix") for vector w. Modern Robotics Eqn 3.30. Args: w: (B,N,3) A 3-vector Returns: W: (
model/rigid.py:4
↓ 2 callersFunctionwrite_color_obj
(path, xyz, color)
utils/fileio.py:234
↓ 1 callersMethod_get_item
(self, index)
data_loader/NormalDataset.py:134
↓ 1 callersMethod_get_item
(self, index)
data_loader/OneSampleDataset.py:105
↓ 1 callersMethod_get_nu_bnd
(self)
data_loader/NormalDataset.py:82
↓ 1 callersMethod_get_nu_bnd
(self)
data_loader/OneSampleDataset.py:56
↓ 1 callersMethod_load_data
(self, index)
data_loader/NormalDataset.py:102
↓ 1 callersMethod_load_data
(self)
data_loader/OneSampleDataset.py:73
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/modules.py:246
↓ 1 callersFunctioncheck_ray_triangle_intersection
Optimized to work for: >1 ray_origins 1 ray_direction multiplied to match the dimension of ray_origins 1 triangle Bas
utils/geometry.py:50
↓ 1 callersFunctionclean_result_dirs
(clean_path)
run/fit.py:92
↓ 1 callersFunctionclean_result_dirs
(clean_path)
run/fit_one_sample.py:80
↓ 1 callersFunctioncompute_chamfer
(recon_pts, gt_pts, f_score_threshold=0.01, factor_to_mm=100)
utils/common.py:45
↓ 1 callersFunctioncompute_chamfer_i3dmm
(recon_pts, gt_pts, f_score_threshold=0.01, facetor_to_mm=100)
utils/summary.py:20
↓ 1 callersFunctioncompute_chamfer_i3dmm
(recon_pts, gt_pts, f_score_threshold=0.01, facetor_to_mm=100)
sketches/evaluate.py:73
↓ 1 callersFunctioncreate_from_scans
(mesh, bounding_radius=1, scan_count=100, scan_resolution=400, calculate_normals=True)
data_preprocess/surface_point_cloud.py:228
↓ 1 callersFunctioncrop
(v, f)
data_preprocess/preprocess.py:19
↓ 1 callersFunctioncut
(vertices, mesh, circle, r)
data_preprocess/cut_mesh.py:255
↓ 1 callersFunctiondata_preprocess
(ply_path, save_path)
data_preprocess/preprocess.py:141
↓ 1 callersFunctiondelaunay_mesh_in_2d
(vertices)
utils/geometry.py:170
↓ 1 callersFunctiondelaunay_mesh_in_2d
(vertices)
data_preprocess/preprocess.py:91
↓ 1 callersFunctiondivergence
(y, x)
model/diff_opts.py:32
↓ 1 callersFunctionexp_so3
Exponential map from Lie algebra so3 to Lie group SO3. Modern Robotics Eqn 3.51, a.k.a. Rodrigues' formula. Args: w: (B,N,3) An axis of
model/rigid.py:38
↓ 1 callersFunctionfind_dot
(face, vertice, final_face, r, circle)
data_preprocess/cut_mesh.py:194
↓ 1 callersMethodfind_dot
(self, face, vertice, final_face)
data_preprocess/cut_mesh.py:60
↓ 1 callersFunctionfind_dot2
(face, vertice, final_face, r, circle)
data_preprocess/cut_mesh.py:145
↓ 1 callersMethodfind_dot2
(self, face, vertice, final_face)
data_preprocess/cut_mesh.py:11
↓ 1 callersMethodforward
(self, input)
model/modules.py:43
↓ 1 callersFunctionget_dataset
(config)
run/train.py:53
↓ 1 callersFunctionget_dataset
(config)
run/fit.py:56
↓ 1 callersFunctionget_dataset
(config)
run/fit_one_sample.py:56
↓ 1 callersFunctionget_equidistant_camera_angles
(count)
data_preprocess/surface_point_cloud.py:220
↓ 1 callersFunctionget_mesh
(test_decoder, exp_embedding, id_embedding, voxel_resolution=256, points_per_inference=163840)
utils/summary.py:288
↓ 1 callersFunctionget_mesh_color
(mesh_points, exp_embedding, id_embedding, model, config, int_idx)
utils/summary.py:87
↓ 1 callersMethodget_template_coords
(self, sample_subset, exp_embedding, id_embedding, int_idx=True)
model/imface.py:207
↓ 1 callersFunctionget_vertical2d
(triangle_id, vertices3d, mat2d, mat3d, face2d, face3d)
utils/geometry.py:263
↓ 1 callersFunctiongradient
(y, x, grad_outputs=None)
model/diff_opts.py:39
↓ 1 callersFunctionhidden_surface_remove
(vertices, triangle_idx, direction=None, epsilon=1e-6)
utils/geometry.py:103
↓ 1 callersFunctionhidden_surface_remove
(vertices, triangle_idx, direction=None, epsilon=1e-6)
data_preprocess/preprocess.py:41
↓ 1 callersFunctionhyper_bias_init
(m)
model/modules.py:233
↓ 1 callersFunctionhyper_weight_init
(m, in_features_main_net)
model/modules.py:223
↓ 1 callersMethodis_outside
(self, points)
data_preprocess/surface_point_cloud.py:189
↓ 1 callersFunctionjacobian
jacobian of y wrt x
model/diff_opts.py:65
↓ 1 callersFunctionload_config
(path)
run/fit.py:77
↓ 1 callersFunctionload_config
(path)
run/fit_one_sample.py:65
↓ 1 callersFunctionmain
(args)
run/train.py:83
↓ 1 callersFunctionmain
(args)
run/fit.py:103
↓ 1 callersFunctionmain
(args)
run/fit_one_sample.py:91
↓ 1 callersFunctionmake_contour_plot
(array_2d, mode='log')
utils/summary.py:63
↓ 1 callersFunctionprocess_normal
(v, f, des, landmark)
data_preprocess/preprocess.py:107
↓ 1 callersFunctionremove_and_watertight
(v, f)
data_preprocess/preprocess.py:96
↓ 1 callersFunctionremove_face_by_point_idx
(face_array_matrix, indicator)
utils/geometry.py:10
↓ 1 callersFunctionremove_face_by_point_idx
(face_array_matrix, indicator)
data_preprocess/preprocess.py:62
↓ 1 callersFunctionremove_vertices_from_mesh
:param vertices: (n, 3) float :param triangle_idx: (m, 3) int :param mask: (n,) where 1 indicates preserve :return: vertices, triangl
utils/geometry.py:120
↓ 1 callersFunctionremove_vertices_from_mesh
:param vertices: (n, 3) float :param triangle_idx: (m, 3) int :param mask: (n,) where 1 indicates preserve :return: vertices, triangl
data_preprocess/preprocess.py:72
↓ 1 callersFunctionrp_to_se3
Rotation and translation to homogeneous transform. Args: R: (B,N,3,3) An orthonormal rotation matrix. p: (B,N,3) A 3-vector representi
model/rigid.py:23
↓ 1 callersFunctionsample_from_mesh
(mesh, sample_point_count=10000000, calculate_normals=True)
data_preprocess/surface_point_cloud.py:250
↓ 1 callersMethodsample_sdf_near_surface
(self, number_of_points=500000, use_scans=True, sign_method='normal', normal_s
data_preprocess/surface_point_cloud.py:113
↓ 1 callersMethodsample_sdf_uni
(self, number_of_points=500000, sign_method='normal', normal_sample_count=11, return_gr
data_preprocess/surface_point_cloud.py:146
↓ 1 callersFunctionsample_uniform_points_in_sphere
(amount, radius=1.)
utils/geometry.py:35
↓ 1 callersFunctionwrite_tensorboard
(train_loss_epoch, writer, epoch)
run/train.py:28
↓ 1 callersFunctionwrite_tensorboard
(train_loss_epoch, writer, epoch)
run/fit.py:31
↓ 1 callersFunctionwrite_tensorboard
(train_loss_epoch, writer, epoch)
run/fit_one_sample.py:31
Method__getitem__
(self, index)
data_loader/NormalDataset.py:154
Method__getitem__
(self, index)
data_loader/OneSampleDataset.py:120
Method__init__
Loads a Wavefront OBJ file.
utils/fileio.py:8
Method__init__
(self, *args, **kwargs)
utils/registry.py:29
Method__init__
(self, root_path, ids, exps, sample_num, sample_func, keypoint_type='full', id2idx=None, exp2idx=None,
data_loader/NormalDataset.py:46
Method__init__
(self, root_path, sample_num, sample_func, keypoint_type='full')
data_loader/OneSampleDataset.py:46
Method__init__
(self, code_dim, in_features, out_features, hidden_features=32, num_hidden_layers=3, hyper_hi
model/imface.py:10
Method__init__
(self, fusion_number, condition_dim, weight_feature_dim=128)
model/imface.py:47
Method__init__
(self, embedding_dim, kpt_num, in_features, out_features, hidden_features=128, num_hidden_layers=3,
model/imface.py:73
Method__init__
(self, in_features, out_features, num_hidden_layers, hidden_features, outermost_linear=False,
model/modules.py:52
Method__init__
(self, out_features=1, type='sine', in_features=2, hidden_features=256, num_hidden_layers=3, **kwargs)
model/modules.py:113
Method__init__
(self, in_features, sidelength=None, fn_samples=None, use_nyquist=True)
model/modules.py:131
Method__init__
Args: hyper_in_features: In features of hypernetwork hyper_hidden_layers: Number of hidden layers in hypernetwork
model/modules.py:174
next →1–100 of 203, ranked by callers