MCPcopy Create free account

hub / github.com/YuQiao0303/PointDreamer / functions

Functions2,121 in github.com/YuQiao0303/PointDreamer

↓ 1 callersMethod_optimize_normal
(self, opt: th.optim.Optimizer)
models/DDNM/guided_diffusion/fp16_util.py:209
↓ 1 callersFunction_save_response_content
(response, destination, chunk_size=32768)
models/DDNM/datasets/utils.py:177
↓ 1 callersMethod_verify_classes
(self, classes)
models/DDNM/datasets/lsun.py:96
↓ 1 callersMethodaabb
get the axis-aligned bounding box of the mesh. Returns: Tuple[torch.Tensor]: the min xyz and max xyz of the mesh.
utils/mesh.py:439
↓ 1 callersFunctionaccimage_loader
(path)
models/DDNM/datasets/imagenet_subset.py:33
↓ 1 callersMethodadd_zeros
(self, vec)
models/DDNM/functions/svd_operators.py:1003
↓ 1 callersFunctionapply_texture2geo
reference: GET3D, utils_3d.py, savemeshtes2 :param v_list: :param f_list: :param uv_list: :param f_mat_id_list: :param mat_im
utils/other_utils.py:230
↓ 1 callersFunctionargs_to_dict
(args, keys)
models/DDNM/guided_diffusion/script_util.py:438
↓ 1 callersFunctionassert_direct_dimensions
models/POCO/eval/src/utils/libsimplify/simplify_mesh.cpp:12840
↓ 1 callersFunctionassign_face_view
(faces,neighbors,face_idxs,similarity,view_num)
pointdreamer/ours_utils.py:786
↓ 1 callersMethodauto_normal
auto calculate the vertex normals.
utils/mesh.py:460
↓ 1 callersMethodauto_size
auto resize the mesh. Args: bound (float, optional): resizing into ``[-bound, bound]^3``. Defaults to 0.9.
utils/mesh.py:449
↓ 1 callersMethodauto_uv
auto calculate the uv coordinates. Args: cache_path (str, optional): path to save/load the uv cache as a npz file, this can avoid
utils/mesh.py:485
↓ 1 callersFunctionaverage_pool
(input, indices)
models/POCO/lightconvpoint/nn/deprecated/pooling_old/average_pooling.py:5
↓ 1 callersFunctionaverage_pool
(input, indices)
models/POCO/lightconvpoint/nn/deprecated/pooling/average_pooling.py:5
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
models/DDNM/guided_diffusion/nn.py:42
↓ 1 callersFunctionbarycentric
models/POCO/eval/src/utils/libsimplify/Simplify.h:217
↓ 1 callersMethodbatched_index_select
Slicing of input with respect to the index tensor
models/POCO/lightconvpoint/nn/deprecated/upsample_nearest.py:39
↓ 1 callersFunctioncalc_accuracy
(num_true, num_predictions)
models/POCO/eval/eval_point2surf/evaluation.py:8
↓ 1 callersFunctioncalc_f1
(precision, recall)
models/POCO/eval/eval_point2surf/evaluation.py:31
↓ 1 callersFunctioncalc_precision
(num_true_pos, num_false_pos)
models/POCO/eval/eval_point2surf/evaluation.py:15
↓ 1 callersFunctioncalc_recall
(num_true_pos, num_false_neg)
models/POCO/eval/eval_point2surf/evaluation.py:23
↓ 1 callersFunctioncalculate_frechet_distance
Numpy implementation of the Frechet Distance. The Frechet distance between two multivariate Gaussians X_1 ~ N(mu_1, C_1) and X_2 ~ N(mu_2, C_2
utils/metric_utils/fid.py:31
↓ 1 callersFunctioncalculate_psnr_batch
:param gt_imgs: [N,H,W,C], uint8, 0-255 :param pred_imgs:[N,H,W,C], uint8, 0-255 :param border: :param use_sk: doesn't matter, same r
utils/metric_utils/psnr_ssmi.py:44
↓ 1 callersFunctioncalculate_ssim
calculate SSIM the same outputs as MATLAB's img1, img2: [0, 255], [H,W,C], uint8,
utils/metric_utils/psnr_ssmi.py:101
↓ 1 callersFunctioncalculate_ssim_batch
:param gt_imgs: [N,H,W,C], uint8, 0-255 :param pred_imgs:[N,H,W,C], uint8, 0-255 :param border: :param use_sk: set to True, much fast
utils/metric_utils/psnr_ssmi.py:76
↓ 1 callersFunctioncalculate_up_vector
(eye_position, target_position,world_up = None)
utils/camera_utils.py:104
↓ 1 callersFunctioncall_necessary
Check if all input files exist and at least one output file does not exist or is invalid. :param file_in: list of str or str :param file_
models/POCO/eval/eval_point2surf/file_utils.py:194
↓ 1 callersFunctionchamfer_distance_kdtree
KD-tree based implementation of the Chamfer distance. Args: points1 (numpy array): first point set points2 (numpy array): second
models/POCO/eval/src/common.py:71
↓ 1 callersFunctionchamfer_distance_naive
Naive implementation of the Chamfer distance. Args: points1 (numpy array): first point set points2 (numpy array): second point se
models/POCO/eval/src/common.py:50
↓ 1 callersFunctioncheck_overflow
(value)
models/DDNM/guided_diffusion/fp16_util.py:235
↓ 1 callersMethodcheck_triangles
(self, points, triangles)
models/POCO/eval/src/utils/libmesh/inside_mesh.py:129
↓ 1 callersFunctioncheck_voxel_occupied
(occupancy_grid)
models/POCO/eval/src/utils/voxels.py:252
↓ 1 callersFunctioncheck_voxel_unoccupied
(occupancy_grid)
models/POCO/eval/src/utils/voxels.py:268
↓ 1 callersFunctionchunks
Yield successive n-sized chunks from lst.
models/POCO/eval_scenenet.py:142
↓ 1 callersFunctioncolorize_one_mesh
( # name,root_path,cls_id, # load_exist_dense_img_path, coords
demo.py:38
↓ 1 callersMethodcompute_intersection_depth
(self, points, triangles)
models/POCO/eval/src/utils/libmesh/inside_mesh.py:75
↓ 1 callersFunctioncompute_iou
Computes the Intersection over Union (IoU) value for two sets of occupancy values. Args: occ1 (tensor): first set of occupancy values
models/POCO/eval/src/common.py:6
↓ 1 callersMethodcompute_mask
(self, pt, bs)
models/POCO/lightconvpoint/datasets/archives/s3dis_legacy.py:126
↓ 1 callersMethodcompute_mask
(self, pt, bs)
models/POCO/lightconvpoint/datasets/archives/s3dis.py:14
↓ 1 callersFunctioncompute_vertex_only_uv_mask
(face_vertex_idx, face_uvs_idx)
pointdreamer/unproject.py:17
↓ 1 callersFunctionconcat_txt_files
(files_in, file_out)
models/POCO/eval/eval_point2surf/file_utils.py:92
↓ 1 callersFunctionconstruct_subtree_double
Construct a sub tree over a range of data points. Params: pa : data points pidx : permutation index of data points no_dims: number of dime
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:978
↓ 1 callersFunctionconstruct_subtree_float
Construct a sub tree over a range of data points. Params: pa : data points pidx : permutation index of data points no_dims: number of dime
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:334
↓ 1 callersFunctionconstruct_tree_float
Construct a tree over data points. Params: pa : data points no_dims: number of dimensions n : number of data points bsp : number of p
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:390
↓ 1 callersFunctioncount_parameters
(model)
models/POCO/train.py:31
↓ 1 callersFunctioncount_parameters
(model)
models/POCO/generate.py:287
↓ 1 callersFunctioncreate_neighbors_tensor
:param N: :param adjacency_pairs: [M,2]. adjacency_pairs[i,0] and adjacency_pairs[i,1] are neighbors :return: [N,K], indicating the K nei
pointdreamer/ours_utils.py:713
↓ 1 callersFunctioncreate_node_double
Create a tree node. Params: start_idx : index of first data point to use n : number of data points ******************************************
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:1067
↓ 1 callersFunctioncreate_node_float
Create a tree node. Params: start_idx : index of first data point to use n : number of data points ******************************************
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:423
↓ 1 callersMethoddata_aug
(xyz, color, labels, idx, num_out)
models/POCO/lightconvpoint/datasets/helper_tool.py:80
↓ 1 callersFunctionddnm_diffusion
(x, model, b, eta, A_funcs, y, cls_fn=None, classes=None, config=None)
models/DDNM/functions/svd_ddnm.py:19
↓ 1 callersFunctionddnm_plus_diffusion
(x, model, b, eta, A_funcs, y, sigma_y, cls_fn=None, classes=None, config=None)
models/DDNM/functions/svd_ddnm.py:80
↓ 1 callersFunctiondefault_loader
(path)
models/DDNM/datasets/imagenet_subset.py:41
↓ 1 callersFunctiondelete_subtree_double
Delete subtree Params: root : root node of subtree to delete ************************************************/
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:1093
↓ 1 callersFunctiondelete_subtree_float
Delete subtree Params: root : root node of subtree to delete ************************************************/
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:449
↓ 1 callersFunctiondetect_abnormal_bright_spots_in_gray_img
:param img: [res,res], :param foreground_mask: :param min_for_norm: 1.0 :param max_for_norm: 3.0 :param edge_thresh: 50 # a brigh
utils/utils_2d.py:584
↓ 1 callersFunctiondilate_foreground
utils/utils_2d.py:850
↓ 1 callersFunctiondilate_torch_batch
dilate white pixels like cv2.dilate :param img_batch: [B, H, W] ,B indicate batch size. each element is either 0 or 1 :param kernel_size
utils/utils_2d.py:833
↓ 1 callersFunctiondisplay_CHW_RGB_img_np_cv2
(img)
utils/utils_2d.py:332
↓ 1 callersMethoddownload
(self)
models/DDNM/datasets/celeba.py:120
↓ 1 callersMethoddownload
(self)
models/POCO/lightconvpoint/datasets/semantic_kitti.py:76
↓ 1 callersMethoddownload
(self)
models/POCO/lightconvpoint/datasets/modelnet.py:138
↓ 1 callersMethoddownload
(self)
models/POCO/lightconvpoint/datasets/s3dis.py:878
↓ 1 callersMethoddownload
(self)
models/POCO/lightconvpoint/datasets/shapenet.py:104
↓ 1 callersFunctiondownload_file_from_google_drive
Download a Google Drive file from and place it in root. Args: file_id (str): id of file to be downloaded root (str): Directory t
models/DDNM/datasets/utils.py:134
↓ 1 callersMethoddumpkvs
(self)
models/DDNM/guided_diffusion/logger.py:355
↓ 1 callersMethodencode_rgb_feature
(self,p_features)
models/TextureField/TF_Network.py:71
↓ 1 callersMethodeval_fid
(self,gt_root_path,pred_root_path)
data/run_evaluation.py:174
↓ 1 callersMethodeval_lpips
(self,gt_root_path,pred_root_path)
data/run_evaluation.py:141
↓ 1 callersMethodeval_mesh
Evaluates a mesh. Args: mesh (trimesh): mesh which should be evaluated pointcloud_tgt (numpy array): target point clo
models/POCO/eval/src/eval.py:37
↓ 1 callersMethodeval_pointcloud
Evaluates a point cloud. Args: pointcloud (numpy array): predicted point cloud pointcloud_tgt (numpy array): target p
models/POCO/eval/src/eval.py:91
↓ 1 callersMethodeval_psnr
(self,gt_root_path,pred_root_path)
data/run_evaluation.py:217
↓ 1 callersMethodeval_ssim
(self,gt_root_path,pred_root_path)
data/run_evaluation.py:252
↓ 1 callersMethodextra_repr
(self)
models/DDNM/datasets/vision.py:54
↓ 1 callersFunctionfibonacci_sphere
(samples, radius)
utils/camera_utils.py:86
↓ 1 callersFunctionfix_K_camera
Fix camera projection matrix. This changes a camera projection matrix that maps to [0, img_size] x [0, img_size] to one that maps to [-1, 1] x
models/POCO/eval/src/common.py:286
↓ 1 callersFunctionfix_Rt_camera
Fixes Rt camera matrix. Args: Rt (tensor): Rt camera matrix loc (tensor): location scale (float): scale
models/POCO/eval/src/common.py:263
↓ 1 callersMethodfixed_normalization
(self, pts, radius=None)
models/POCO/lightconvpoint/nn/conv_fkaconv.py:53
↓ 1 callersMethodforward
(self, x)
models/DDNM/guided_diffusion/nn.py:13
↓ 1 callersMethodforward
(self, x)
models/DDNM/guided_diffusion/unet.py:100
↓ 1 callersMethodforward_with_features
(self, x, pos, support_points, indices)
models/POCO/lightconvpoint/nn/deprecated/module.py:12
↓ 1 callersMethodfree_all
Frees all allocated memory chunks */
models/POCO/lightconvpoint/src/nanoflann.hpp:592
↓ 1 callersMethodgenerate_grid_features
(self, p, c)
models/TextureField/convonet.py:498
↓ 1 callersFunctiongetGaussianKernel
(ksize, sigma=0)
utils/utils_2d.py:743
↓ 1 callersFunctionget_bbox_center_vectors
(box)
utils/vtk_basic.py:715
↓ 1 callersFunctionget_beta_schedule
(beta_schedule, *, beta_start, beta_end, num_diffusion_timesteps)
models/DDNM/guided_diffusion/diffusion.py:46
↓ 1 callersFunctionget_bounding_box_double
Get the bounding box of a set of points Params: pa : data points pidx : permutation index of data points no_dims: number of dimensions
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:812
↓ 1 callersFunctionget_bounding_box_float
Get the bounding box of a set of points Params: pa : data points pidx : permutation index of data points no_dims: number of dimensions
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:168
↓ 1 callersFunctionget_box_corners
Convert box center and vectors to the corner-form :param center: :param vectors: :return: corner points and faces related to the box
utils/vtk_basic.py:728
↓ 1 callersMethodget_category
(self, f_id)
models/POCO/datasets/abc_test.py:35
↓ 1 callersMethodget_category
(self, f_id)
models/POCO/datasets/real_world.py:32
↓ 1 callersMethodget_category
(self, f_id)
models/POCO/datasets/thingi10k_test.py:34
↓ 1 callersMethodget_category
(self, f_id)
models/POCO/datasets/famous_test.py:34
↓ 1 callersMethodget_category
(self, f_id)
models/POCO/datasets/shapenet.py:120
↓ 1 callersFunctionget_ckpt_path
(name, root=None, check=False, prefix='exp')
models/DDNM/functions/ckpt_util.py:55
↓ 1 callersMethodget_class_name
(self, f_id)
models/POCO/datasets/abc.py:46
↓ 1 callersFunctionget_conv
Get a convolutional layer by name. # Arguments conv_name: string. The name of the convolutional layer.
models/POCO/lightconvpoint/utils/network.py:5
↓ 1 callersFunctionget_cube_offset_double
Get squared distance from point to cube in specified dimension Params: dim : dimension point_coord : cartesian coordinates of point bbox :
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:1159
↓ 1 callersFunctionget_cube_offset_float
Get squared distance from point to cube in specified dimension Params: dim : dimension point_coord : cartesian coordinates of point bbox :
models/POCO/eval/src/utils/libkdtree/pykdtree/_kdtree_core.c:515
← previousnext →701–800 of 2,121, ranked by callers