MCPcopy Create free account

hub / github.com/MingwuZheng/ImFace / functions

Functions203 in github.com/MingwuZheng/ImFace

Method__init__
(self, mesh, points, normals=None, scans=None)
data_preprocess/surface_point_cloud.py:20
Method__init__
(self, file_name, to_name)
data_preprocess/cut_mesh.py:7
Method__iter__
(self)
run/train.py:24
Method__iter__
(self)
run/fit.py:27
Method__iter__
(self)
run/fit_one_sample.py:27
Method__len__
(self)
data_loader/NormalDataset.py:131
Method__len__
(self)
data_loader/OneSampleDataset.py:102
Functionbytes2human
(n)
utils/common.py:28
Functionclean_result_dirs
(config)
run/train.py:70
Functioncompute_chamfer_mesh
(recon_v, recon_f, gt_v, gt_f, num_pts=150000, factor_to_mm=100)
utils/common.py:67
Functioncond_mkdir
(path)
utils/fileio.py:52
Functioncond_mkdir
(path)
utils/common.py:17
Functionconvert_sdf_samples_with_color_to_ply
(sdf_3d, ply_filename_out, mask, voxel_grid_origin=np.array([-1, -1, -1]),
utils/summary.py:203
Functioncreate_grid
(N=256, mask_size=1., bbox_size=2.)
utils/geometry.py:200
Functioncreate_grid_2d
(N=256)
utils/geometry.py:217
Functioncrop_spherically
:param pcl: n*3 :param center: [x,y,z] :param r: range :return:
utils/geometry.py:21
Functiondict_to
(dict, device)
utils/common.py:22
Functionevaluate_imface
(path='/home/zhengmingwu_2020/ImFace/dataset/FacescapeAlign')
sketches/evaluate.py:147
Functionexp_se3
Exponential map from Lie algebra so3 to Lie group SO3. Modern Robotics Eqn 3.88. Args: S: (B,N,6) A screw axis of motion. theta: (
model/rigid.py:56
Functionfarthest_point_sample
Input: xyz: pointcloud data, [N, D] npoint: number of samples Return: centroids: sampled pointcloud index, [npoint, D
utils/geometry.py:176
Functionfilename_to_exp_type
(filename)
utils/common.py:75
Functionfirst_layer_sine_init
(m)
model/modules.py:328
Functionfit_decoder
(config, test_decoder, test_data, exp_embeddings, id_embeddings, path, device, fit_id_num)
utils/summary.py:211
Functionfit_one_decoder
(config, test_decoder, test_data, exp_embeddings, id_embeddings, path, device, fit_id_num)
utils/summary.py:247
Methodforward
:param x: (B,N,3*5) :param code: (B,N) :return: f(x-p)
model/imface.py:31
Methodforward
:param condition: (B,N,3) :param inputs: (B,N,K,O)
model/imface.py:62
Methodforward
:param keypoints: (B,K,3) or (K,3) :param xyz: (B,N,3) or (N,3) :param code: (B,D)
model/imface.py:88
Methodforward
(self, input_dict, exp_code=None, id_code=None)
model/imface.py:299
Methodforward
(self, input, params=None)
model/modules.py:25
Methodforward
(self, coords, params=None, **kwargs)
model/modules.py:102
Methodforward
(self, model_input, params=None)
model/modules.py:119
Methodforward
(self, coords)
model/modules.py:157
Methodforward
Args: z: Embedding. Input to hypernetwork. Could be output of "Autodecoder" Returns: params: OrderedDict. Can
model/modules.py:204
Functionfrom_homogenous
(v)
model/rigid.py:87
MethodgetFile
(self)
data_preprocess/cut_mesh.py:119
Functionget_coordinates
(config, test_decoder, data, exp_embeddings, id_embeddings, device, fit_id_num)
utils/summary.py:299
Functionget_jet_color
(array, clip_min=None, clip_max=None)
utils/visualization.py:87
Functionget_jet_color_corr
(array, p=0.5)
utils/visualization.py:112
Methodget_landmarks
(self, exp_embedding, id_embedding, int_idx=True)
model/imface.py:197
Functionget_mgrid
Generates a flattened grid of (x,y,...) coordinates in a range of -1 to 1.
utils/geometry.py:231
Functionget_point_color
(mesh_points, clip_min=None, clip_max=None)
utils/visualization.py:131
Methodget_template_kpts
(self)
data_loader/OneSampleDataset.py:68
Methodget_voxels
(self, voxel_resolution, use_depth_buffer=False, sample_count=11, pad=False, check_result=False,
data_preprocess/surface_point_cloud.py:88
Functiongrad_correction
(sdf, grad, pos_direction=None)
data_preprocess/surface_point_cloud.py:264
Functiongradient_3d_by_3d
(y, x)
model/diff_opts.py:52
Functiongradients
(y, x, grad_outputs=None)
model/diff_opts.py:46
Functiongradients_mse
(model_output, gt)
model/loss.py:31
Functionhessian
hessian of y wrt x y: shape (meta_batch_size, num_observations, channels) x: shape (meta_batch_size, num_observations, 2)
model/diff_opts.py:5
Methodhook_weight
(self, xyz, seg=0)
model/imface.py:212
Functionhypo_weight_loss
(model_output)
model/loss.py:20
Functioninit_weights_elu
(m)
model/modules.py:307
Functioninit_weights_normal
(m)
model/modules.py:294
Functioninit_weights_selu
(m)
model/modules.py:300
Functioninit_weights_trunc_normal
(m)
model/modules.py:278
Functioninit_weights_xavier
(m)
model/modules.py:314
Functionlaplace
(y, x)
model/diff_opts.py:27
Functionlaplace_mse
(model_output, gt)
model/loss.py:8
Functionlast_layer_sine_init
(m)
model/modules.py:336
Functionlatent_loss
(model_output)
model/loss.py:16
Functionlif_loss
(net_input, net_middle, net_out, losses)
model/loss.py:39
Functionlin2img
(tensor, image_resolution=None)
utils/visualization.py:45
Functionload_off
Reads vertices and faces from an off file. :param file: path to file to read :type file: str :return: vertices and faces as lists of
utils/fileio.py:154
Functionmap2color
(color_img, pcl)
utils/summary.py:279
Functionnbs_loss
(net_input, net_middle, net_out, losses)
model/loss.py:134
Functionon_complete
(*_)
data_preprocess/preprocess.py:183
Functionpositional_encoding
r"""Apply positional encoding to the input. Args: tensor (torch.Tensor): Input tensor to be positionally encoded. encoding_size (o
model/net_utils.py:5
Functionprincipal_analyze
(config, test_decoder, train_datas, path, device, n_components=4, shift_times=10)
utils/summary.py:314
Functionram_sample
(pos_tensor, neg_tensor, sample_num)
utils/sample.py:8
Functionrandom_sample
(pos_tensor, neg_tensor, sample_num)
utils/sample.py:31
Functionrandom_sample_full
(tensor, sample_num)
utils/sample.py:49
Functionrandom_sample_full_ram
(tensor, sample_num)
utils/sample.py:57
Functionrays_triangles_intersection
:param ray_origins: torch.Tensor, (n_rays, n_dimensions), (n, 3) :param ray_direction: torch.Tensor, (n_rays, n_dimensions), (n, 3) :pa
utils/geometry.py:140
Functionread_array
(path)
utils/fileio.py:75
Functionread_csv
(file_location)
utils/fileio.py:282
Functionread_dict
(path, key2int=True)
utils/fileio.py:63
Functionread_hdf5
Read a tensor, i.e. numpy array, from HDF5. :param file: path to file to read :type file: str :param key: key to read :type key:
utils/fileio.py:261
Functionread_pcl_txt
:param path: N rows, 3 columns, float txt :return: Nx3 pcl
utils/fileio.py:86
Methodregister
(self, module_name, module=None)
utils/registry.py:32
Methodregister_fn
(fn)
utils/registry.py:39
Functionsample_on_surface
(mesh, number_of_points=500000, surface_point_method='scan', sign_method='normal', scan_
data_preprocess/surface_point_cloud.py:291
Functionsample_sdf_near_surface
(mesh, number_of_points=500000, surface_point_method='scan', sign_method='normal',
data_preprocess/surface_point_cloud.py:275
Functionshow_contour_plot
(array_2d, mode='log')
utils/visualization.py:21
Functionshow_hist
(data)
utils/visualization.py:70
Functionshow_image
(img)
utils/visualization.py:65
Functionshow_roc
(score, label, color='r', lw=2)
utils/visualization.py:7
Functionshow_scatter
(pts)
utils/visualization.py:77
Functionshow_sdf_2d
(sdfs, w, h)
utils/visualization.py:57
Functionsine_init
(m)
model/modules.py:320
Functionsort_key
(key)
run/fit.py:57
Functionsymmetric_dirichlet_energy
(y, x)
model/diff_opts.py:82
Functiontest_decoder_on_trainset_2d
(config, test_decoder, train_data, test_obj_list, device, writer, batch_cnt)
utils/summary.py:150
Functiontexture_transfer
Use to map files with texture to files without texture mtl_name: files with texture obj_vertices: point cloud
utils/geometry.py:255
Functionto_homogenous
(v)
model/rigid.py:83
Functiontransformation_4d
(pcl, matrix_4x4)
utils/common.py:13
Functionwrite_color_obj_by_sdf_grid
(path, sdfs, resolution, clamp=0.003)
utils/fileio.py:246
Functionwrite_color_obj_by_sdf_point
(path, sdfs, points, clamp=0.003)
utils/fileio.py:254
Functionwrite_csv
(file_location, rows)
utils/fileio.py:288
Functionwrite_dict
(path, dict)
utils/fileio.py:57
Functionwrite_mtl
(vertices, faces, texture_vertices, to_name, to_path,
utils/fileio.py:349
Functionwrite_off
Writes the given vertices and faces to OFF. :param vertices: vertices as tuples of (x, y, z) coordinates :type vertices: [(float)] :
utils/fileio.py:113
← previousnext →101–200 of 203, ranked by callers