MCPcopy Create free account

hub / github.com/RozDavid/UnScene3D / functions

Functions1,028 in github.com/RozDavid/UnScene3D

⨍ Functions 1,028◇ Types & classes 235
↓ 1 callersMethodclean_mesh
(self, scene_mesh)
pseudo_masks/datasets/arkit.py:61
↓ 1 callersMethodclean_segments
(self, comps, min_vert_num=500)
pseudo_masks/datasets/arkit.py:89
↓ 1 callersFunctionconfig
()
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:407
↓ 1 callersFunctionconfig
()
pseudo_masks/datasets/evaluation/evaluate_semantic_label.py:132
↓ 1 callersFunctionconvert_box_to_trimesh_fmt
(box)
utils/votenet_utils/pc_util.py:370
↓ 1 callersMethodconvert_to_ply
Convert Stanford3DDataset to PLY format that is compatible with Synthia dataset. Assumes file structure as given by the dataset. Outpu
pseudo_masks/datasets/preprocess/stanford.py:140
↓ 1 callersMethodcreate_label_database
(self, git_repo)
datasets/preprocessing/freemask_preprocessing.py:55
↓ 1 callersMethodcreate_label_database
(self)
datasets/preprocessing/s3dis_preprocessing.py:60
↓ 1 callersMethodcreate_label_database
(self, config_file)
datasets/preprocessing/semantic_kitti_preprocessing.py:46
↓ 1 callersMethodcreate_label_database
(self)
datasets/preprocessing/arkit_preprocessing.py:45
↓ 1 callersMethodcreate_label_database
(self)
datasets/preprocessing/stpls3d_preprocessing.py:69
↓ 1 callersMethodcreate_label_database
(self, git_repo)
datasets/preprocessing/scannet_preprocessing.py:51
↓ 1 callersMethodcreate_poisson_felzenswalb_oversegmentation
(cls, points, colors, config, mesh_file)
pseudo_masks/datasets/preprocess/stanford.py:81
↓ 1 callersFunctioncrop_region
(points, cluster_label, curr_cluster, growth_region=0.)
pseudo_masks/eval_3duis_performance.py:45
↓ 1 callersFunctioncross
utils/cpp_utils/include/segmentator.h:99
↓ 1 callersMethodcross
utils/cpp_utils/include/vec3.h:159
↓ 1 callersFunctiondemean
removes mean of tensor channels
pseudo_masks/unscene3d_pseudo_main.py:156
↓ 1 callersFunctiondemo_nn_distance
()
utils/votenet_utils/nn_distance.py:63
↓ 1 callersFunctiondo_felzenszwalb
(vertices, colors, faces, num_points)
utils/cpp_utils/test/test_segmentor.py:28
↓ 1 callersMethoddo_weight_init
(self, weight_init_name)
models/modules/helpers_3detr.py:104
↓ 1 callersMethoddo_weight_init
(self, weight_init_name)
models/modules/3detr_helpers.py:104
↓ 1 callersMethodelastic_distortion
Apply elastic distortion on sparse coordinate space. pointcloud: numpy array of (number of points, at least 3 spatial dims) granu
utils/transforms.py:207
↓ 1 callersMethodencode
(v)
utils/kfold.py:29
↓ 1 callersMethodeval_instance_epoch_end
(self)
trainer/trainer.py:785
↓ 1 callersMethodeval_instance_step
(self, output, target_low_res, target_full_res, inverse_maps, file_names, full_res_
trainer/trainer.py:479
↓ 1 callersMethodevaluate_confusion
(self, output_file=None)
pseudo_masks/datasets/evaluation/evaluate_semantic_label.py:104
↓ 1 callersMethodevaluate_matches
(self, matches)
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:73
↓ 1 callersFunctionexport_ids
(filename, ids)
pseudo_masks/datasets/evaluation/scannet_benchmark_utils/util_3d.py:32
↓ 1 callersFunctionexport_ids
(filename, ids)
benchmark/util_3d.py:32
↓ 1 callersMethodextract_instance_from_file
(self, sample_from_database)
datasets/preprocessing/base_preprocessing.py:114
↓ 1 callersMethodextrinsics2RT
Convert extrinsics matrix to separate rotation matrix R and translation vector T.
utils/pc_utils.py:149
↓ 1 callersFunctionfloorf
utils/cuda_utils/include/cutil_math.h:1222
↓ 1 callersFunctionfminf
utils/cuda_utils/include/cutil_math.h:61
↓ 1 callersMethodforward
(self, x)
models/modules/helpers_3detr.py:110
↓ 1 callersMethodforward
(self, x)
models/modules/3detr_helpers.py:110
↓ 1 callersMethodforward_attention
Extracts both key and query features from the last block, scale them to output resolution and return them
models/encoders_2d/dino.py:67
↓ 1 callersMethodforward_descriptor
(self, input_images)
models/encoders_2d/dino.py:29
↓ 1 callersMethodforward_post
(self, tgt, tgt_mask = None, tgt_key_padding_mask= None,
models/mask3d.py:514
↓ 1 callersMethodforward_post
(self, tgt, memory, memory_mask = None, memory_key_padding_mask = No
models/mask3d.py:571
↓ 1 callersMethodforward_post
(self, tgt)
models/mask3d.py:636
↓ 1 callersMethodforward_pre
(self, tgt, tgt_mask= None, tgt_key_padding_mask = None,
models/mask3d.py:526
↓ 1 callersMethodforward_pre
(self, tgt, memory, memory_mask = None, memory_key_padding_mask = None
models/mask3d.py:585
↓ 1 callersMethodforward_pre
(self, tgt)
models/mask3d.py:642
↓ 1 callersFunctionfreemask_voxelize
(batch, ignore_label, voxel_size, mode, ignore_class_threshold)
datasets/utils.py:370
↓ 1 callersMethodgenerate_grid_targets
(self, coords, labels, instance_ids, segment_ids=None)
pseudo_masks/datasets/scannet_free.py:103
↓ 1 callersFunctionget_affinity_matrix
(feats, tau=0.15, eps=1e-5, normalize_sim=True, similarity_metric='cos')
pseudo_masks/unscene3d_pseudo_main.py:89
↓ 1 callersFunctionget_cluster_saliency
(x, out, grad_point, corr_points=None)
pseudo_masks/eval_3duis_performance.py:66
↓ 1 callersFunctionget_evenly_distributed_colors
(count: int)
trainer/trainer.py:32
↓ 1 callersMethodget_fourier_embeddings
(self, xyz, num_channels=None, input_range=None)
models/position_embedding.py:128
↓ 1 callersMethodget_instance_info
:param xyz: (n, 3) :param instance_ids: (n), int, (1~nInst, -1) :return: instance_num, dict
pseudo_masks/datasets/dataset.py:310
↓ 1 callersMethodget_instance_verts
(self, mesh_vert_instances, instance_id)
pseudo_masks/datasets/evaluation/scannet_benchmark_utils/util_3d.py:105
↓ 1 callersMethodget_instance_verts
(self, mesh_vert_instances, instance_id)
benchmark/util_3d.py:99
↓ 1 callersFunctionget_iou
Calculate the Intersection over Union (IoU) of two 2D bounding boxes. Parameters ---------- bb1 : dict Keys: {'x1', 'x2', 'y
utils/votenet_utils/box_util.py:120
↓ 1 callersMethodget_iou
(self, label_id, confusion)
pseudo_masks/datasets/evaluation/evaluate_semantic_label.py:65
↓ 1 callersFunctionget_iou_main
(get_iou_func, args)
utils/votenet_utils/eval_det.py:74
↓ 1 callersMethodget_label_id
(self, instance_id)
pseudo_masks/datasets/evaluation/scannet_benchmark_utils/util_3d.py:102
↓ 1 callersMethodget_label_id
(self, instance_id)
benchmark/util_3d.py:96
↓ 1 callersFunctionget_masked_affinity_matrix
(painting, feats, mask)
pseudo_masks/unscene3d_pseudo_main.py:122
↓ 1 callersFunctionget_models
Returns a tuple of sample models.
models/__init__.py:13
↓ 1 callersFunctionget_models
Returns a tuple of sample models.
models/encoders_2d/__init__.py:12
↓ 1 callersMethodget_pos_encs
(self, coords)
models/mask3d.py:183
↓ 1 callersFunctionget_salient_areas
(second_smallest_vec)
pseudo_masks/unscene3d_pseudo_main.py:149
↓ 1 callersMethodget_sine_embeddings
xyz: B x N x 3 num_channels: int - number of output feature dim input_range: [[B x 3], [B x 3]] - min and max XYZ coords
models/position_embedding.py:74
↓ 1 callersFunctiongraph_cut
(G)
pseudo_masks/eval_3duis_performance.py:171
↓ 1 callersFunctionhandle_scene
(index, dataset, model, output_base_path, z_max=0.25, roi_size = 0., refine=True)
pseudo_masks/eval_3duis_performance.py:187
↓ 1 callersMethodhash
(self, time, batch)
utils/utils.py:499
↓ 1 callersFunctionhuber_loss
Args: error: Torch tensor (d1,d2,...,dk) Returns: loss: Torch tensor (d1,d2,...,dk) x = error = pred - gt or dist(pred,g
utils/votenet_utils/nn_distance.py:15
↓ 1 callersMethodimage_summary
Log a list of images.
utils/votenet_utils/tf_logger.py:26
↓ 1 callersMethodinitialize_filter
(self, NetClass, in_nchannel, out_nchannel, config)
models/wrapper.py:18
↓ 1 callersFunctioninitialize_models
(config, device)
pseudo_masks/unscene3d_pseudo_main.py:36
↓ 1 callersMethodinvert
utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:368
↓ 1 callersFunctionis_dist_avail_and_initialized
()
models/misc.py:106
↓ 1 callersFunctionkinectDepthToSkeleton
utils/cuda_utils/include/cudaUtil.h:103
↓ 1 callersFunctionkinectProjToCameraZ
utils/cuda_utils/include/cudaUtil.h:109
↓ 1 callersFunctionload_backbone_checkpoint_with_missing_or_exsessive_keys
(cfg, model)
utils/utils.py:58
↓ 1 callersMethodload_intrinsics
(self, scene_name, coords)
pseudo_masks/datasets/scannet.py:199
↓ 1 callersFunctionload_matrix_from_txt
(path, shape=(4, 4))
utils/utils.py:194
↓ 1 callersMethodload_mesh
(self, index)
pseudo_masks/datasets/arkit.py:95
↓ 1 callersFunctionload_obj_with_normals
(filepath)
utils/point_cloud_utils.py:37
↓ 1 callersFunctionload_ply
(filepath)
utils/point_cloud_utils.py:9
↓ 1 callersMethodload_ply_w_path
(self, filepath, scene_name)
pseudo_masks/datasets/dataset.py:205
↓ 1 callersMethodload_self_train_masks
(self, idx, points, freemasks, drop_original=False)
datasets/freemask_semseg.py:224
↓ 1 callersMethodload_unsupervised_scene_data
(self, index)
pseudo_masks/datasets/scannet_free.py:271
↓ 1 callersMethodload_world_pointcloud
(self, filename)
pseudo_masks/datasets/dataset.py:423
↓ 1 callersFunctionmain
(cfg: DictConfig)
main_instance_segmentation.py:110
↓ 1 callersFunctionmain
(config)
pseudo_masks/eval_freemask_gt_performance.py:22
↓ 1 callersFunctionmain
(config)
pseudo_masks/unscene3d_pseudo_main.py:533
↓ 1 callersFunctionmain
(config)
pseudo_masks/eval_3duis_performance.py:351
↓ 1 callersFunctionmain
(config)
pseudo_masks/eval_cutler_projection_performance.py:63
↓ 1 callersFunctionmain
(config)
pseudo_masks/eval_cutler_images.py:65
↓ 1 callersFunctionmain
(config)
pseudo_masks/freemask_main.py:27
↓ 1 callersFunctionmain
()
pseudo_masks/datasets/evaluation/evaluate_semantic_label.py:140
↓ 1 callersFunctionmake_pcd
(coords, feats)
pseudo_masks/datasets/s3dis.py:192
↓ 1 callersFunctionmanhattan
(a, b)
pseudo_masks/eval_3duis_performance.py:95
↓ 1 callersFunctionmatrix_nms
(cate_labels, seg_masks, sum_masks, cate_scores, sigma=2.0, kernel='gaussian', eps=10e-9, nms_thr=0.5)
utils/pc_utils.py:724
↓ 1 callersMethodmemory_efficient_forward
More memory-friendly matching
models/matcher.py:98
↓ 1 callersFunctionmkdir_p
(path)
utils/utils.py:435
↓ 1 callersMethodnetwork_initialization
(self, in_channels, out_channels, config, D)
models/resnet.py:27
↓ 1 callersFunctionnms_2d
(boxes, overlap_threshold)
utils/votenet_utils/nms.py:13
↓ 1 callersFunctionnms_2d_faster
(boxes, overlap_threshold, old_type=False)
utils/votenet_utils/nms.py:44
← previousnext →201–300 of 1,028, ranked by callers