MCPcopy Create free account

hub / github.com/RozDavid/UnScene3D / functions

Functions1,028 in github.com/RozDavid/UnScene3D

↓ 81 callersMethodjoin
utils/cpp_utils/include/segmentator.h:71
↓ 60 callersMethodto
(self, device)
models/misc.py:30
↓ 51 callersMethodsplit
(self, X, labels, groups)
utils/kfold.py:23
↓ 33 callersFunctionmake_float3
utils/cuda_utils/include/cutil_math.h:138
↓ 30 callersFunctionget_norm
(norm_type, n_channels, D, bn_momentum=0.1)
models/modules/common.py:20
↓ 29 callersMethodupdate
(self, val, n=1)
utils/utils.py:429
↓ 27 callersFunctionconv
( in_planes, out_planes, kernel_size, stride=1, dilation=1, bias=False, conv_type=
models/modules/common.py:125
↓ 26 callersFunctionmake_float2
utils/cuda_utils/include/cutil_math.h:91
↓ 25 callersFunctionmake_float4
utils/cuda_utils/include/cutil_math.h:205
↓ 24 callersMethod_make_layer
( self, block, planes, blocks, stride=1, dilation=1, n
models/resnet.py:96
↓ 23 callersFunctionmax
utils/cpp_utils/include/vec3.h:422
↓ 19 callersMethodtranspose
utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:356
↓ 18 callersMethodsize
utils/cpp_utils/include/segmentator.h:91
↓ 16 callersMethod_load_yaml
(cls, filepath)
datasets/preprocessing/base_preprocessing.py:181
↓ 16 callersMethod_save_yaml
(cls, path, file)
datasets/preprocessing/base_preprocessing.py:164
↓ 16 callersFunctionmake_int2
utils/cuda_utils/include/cutil_math.h:108
↓ 16 callersFunctionmake_int3
utils/cuda_utils/include/cutil_math.h:163
↓ 16 callersFunctionmake_int4
utils/cuda_utils/include/cutil_math.h:226
↓ 16 callersFunctionmin
utils/cpp_utils/include/vec3.h:438
↓ 15 callersMethoddot
utils/cpp_utils/include/vec3.h:155
↓ 13 callersFunctionmake_uint2
utils/cuda_utils/include/cutil_math.h:125
↓ 13 callersFunctionmake_uint3
utils/cuda_utils/include/cutil_math.h:184
↓ 13 callersFunctionmake_uint4
utils/cuda_utils/include/cutil_math.h:248
↓ 10 callersMethodbackward
input: grad_out: (m, c, nsample) output: (n, c), None
utils/pointops2/functions/pointops.py:64
↓ 10 callersFunctionclamp
utils/cpp_utils/include/vec3.h:454
↓ 10 callersMethodclip
(self, coords, center=None, trans_aug_ratio=None)
pseudo_masks/datasets/voxelizer.py:76
↓ 10 callersFunctionconv_tr
( in_planes, out_planes, kernel_size, upsample_stride=1, dilation=1, bias=False, c
models/modules/common.py:158
↓ 8 callersFunctiondot
utils/cuda_utils/include/cutil_math.h:1142
↓ 8 callersMethodfind
utils/cpp_utils/include/segmentator.h:63
↓ 7 callersMethod_read_json
(self, path)
datasets/preprocessing/rio_preprocessing.py:111
↓ 6 callersMethod_load_yaml
(filepath)
datasets/semseg.py:570
↓ 6 callersFunctioninitialize_data_loader
Initialize a data loader for a given dataset class.
pseudo_masks/datasets/dataset.py:495
↓ 6 callersFunctionload_dataset
Creates and returns an instance of the datasets given its name.
pseudo_masks/datasets/__init__.py:15
↓ 6 callersFunctionmake_crops
(batch)
datasets/utils.py:616
↓ 6 callersMethodnormalize
utils/cpp_utils/include/vec3.h:235
↓ 6 callersMethodsave_database
(self, database, mode)
datasets/preprocessing/base_preprocessing.py:146
↓ 5 callersMethod_load_yaml
(filepath)
datasets/outdoor_semseg.py:151
↓ 5 callersMethodadd
Computes the confusion matrix The shape of the confusion matrix is K x K, where K is the number of classes. Keyword argument
models/metrics/confusionmatrix.py:29
↓ 5 callersFunctionconvert_conv_type
(conv_type, kernel_size, D)
models/modules/common.py:80
↓ 5 callersFunctionload_ply_with_normals
(filepath)
utils/point_cloud_utils.py:23
↓ 5 callersFunctionload_state_with_same_shape
(model, weights, prefix='')
utils/utils.py:146
↓ 5 callersFunctionprecision_recall
(TP, FP, FN)
utils/votenet_utils/metric_util.py:95
↓ 5 callersMethodsegment_mesh
(self, scene_name, coords, axis_alignment, scene_mesh=None)
pseudo_masks/datasets/scannet.py:156
↓ 5 callersFunctionsingle_scene_precision_recall
Compute P and R for predicted bounding boxes. Ignores classes! Args: labels: (N x bbox) ground-truth bounding boxes (6 dims) pred
utils/votenet_utils/metric_util.py:57
↓ 4 callersMethod__init__
(self, config, hidden_dim, num_queries, num_heads, dim_feedforward, sample_sizes, shared_deco
models/mask3d.py:17
↓ 4 callersFunctionadd_datasets
(module)
pseudo_masks/datasets/__init__.py:5
↓ 4 callersMethodadd_gt
(self, instance_info, id)
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:365
↓ 4 callersMethodadd_prediction
(self, instance_info, id)
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:362
↓ 4 callersFunctioncalc_iou
Computes IoU of two axis aligned bboxes. Args: box_a, box_b: 6D of center and lengths Returns: iou
utils/votenet_utils/metric_util.py:101
↓ 4 callersFunctionclamp
utils/cuda_utils/include/cutil_math.h:1050
↓ 4 callersMethodcompute_color_mean_std
( self, train_database_path: str = "./data/processed/train_database.yaml", )
datasets/preprocessing/base_preprocessing.py:141
↓ 4 callersFunctioncompute_metric_averages
(avg_values, metric='ap')
benchmark/evaluate_semantic_instance.py:243
↓ 4 callersFunctioncreate_color_palette
()
benchmark/util.py:82
↓ 4 callersMethodevaluate
(self)
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:369
↓ 4 callersMethodfix_bugs_in_labels
(self)
datasets/preprocessing/base_preprocessing.py:138
↓ 4 callersMethodget_full_res_mask
(self, mask, inverse_map, point2segment_full, is_heatmap=False)
trainer/trainer.py:445
↓ 4 callersFunctionlength
utils/cuda_utils/include/cutil_math.h:1185
↓ 4 callersFunctionload_2d_model
Creates and returns an instance of the model given its class name.
models/encoders_2d/__init__.py:16
↓ 4 callersFunctionload_model
Creates and returns an instance of the model given its class name.
models/__init__.py:18
↓ 4 callersMethodmask_module
(self, query_feat, mask_features, mask_segments, num_pooling_steps, ret_attn_mask=True,
models/mask3d.py:407
↓ 4 callersFunctionpoly_area
Ref: http://stackoverflow.com/questions/24467972/calculate-area-of-polygon-given-x-y-coordinates
utils/votenet_utils/box_util.py:64
↓ 4 callersFunctionread_txt
Read txt file into lines.
utils/utils.py:445
↓ 4 callersMethodreset_permutation
(self)
pseudo_masks/datasets/dataloader.py:24
↓ 4 callersMethodvoxelize
(self, coords, feats, center=None, augment=True)
pseudo_masks/datasets/voxelizer.py:109
↓ 4 callersMethodwith_pos_embed
(self, tensor, pos)
models/mask3d.py:568
↓ 3 callersMethod__init__
data_paths: list of lists, [[str_path_to_input, str_path_to_label], [...]]
pseudo_masks/datasets/dataset.py:94
↓ 3 callersFunction_get_activation_fn
Return an activation function given a string
models/mask3d.py:654
↓ 3 callersFunction_remap_model_output
(output, labels)
utils/pc_visualizations.py:102
↓ 3 callersFunctionaggregate_features
(encoded_features, segment_ids, seg_connectivity, config)
pseudo_masks/unscene3d_pseudo_main.py:350
↓ 3 callersFunctionconvex_hull_intersection
Compute area of two convex hull's intersection area. p1,p2 are a list of (x,y) tuples of hull vertices. return a list of (x,y) for th
utils/votenet_utils/box_util.py:68
↓ 3 callersFunctioncutilDeviceSynchronize
utils/cuda_utils/include/cutil_inline_runtime.h:53
↓ 3 callersFunctionencode_scene_feats
(scene_batch, model, config, dataset, use_image_data=False, whiten=False, device='cuda')
pseudo_masks/unscene3d_pseudo_main.py:270
↓ 3 callersFunctionevaluate
(preds: dict, gt_path: str, output_file: str, dataset: str = "scannet", gt_dict: dict = None, print_scene_resu
benchmark/evaluate_semantic_instance.py:459
↓ 3 callersMethodget_mask_and_scores
(self, mask_cls, mask_pred, num_queries=100, num_classes=18, device=None)
trainer/trainer.py:456
↓ 3 callersFunctioninside
(p)
utils/votenet_utils/box_util.py:30
↓ 3 callersFunctionlerp
utils/cpp_utils/include/segmentator.h:105
↓ 3 callersMethodmap2color
(self, labels)
datasets/semseg.py:317
↓ 3 callersFunctionmax
utils/cuda_utils/include/cutil_math.h:71
↓ 3 callersFunctionmin
utils/cuda_utils/include/cutil_math.h:76
↓ 3 callersFunctionnormalize_mat
(A, eps=1e-5)
pseudo_masks/unscene3d_pseudo_main.py:82
↓ 3 callersMethodprepare_scene_data
(self, coords, feats, labels, instance_ids, segment_ids)
pseudo_masks/datasets/s3dis.py:127
↓ 3 callersFunctionrsqrtf
utils/cuda_utils/include/cutil_math.h:81
↓ 3 callersMethodsetZero
utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:1189
↓ 3 callersFunctionsqrt
utils/cpp_utils/include/vec3.h:402
↓ 2 callersFunction_ConvertSMVer2Cores_local
Beginning of GPU Architecture definitions
utils/cuda_utils/include/cutil_inline_runtime.h:93
↓ 2 callersMethod__init__
(self, dims3d, width, height, voxel_size, config, depth_min=0.1, depth_max=4.0)
utils/cuda_utils/raycast_image.py:129
↓ 2 callersMethod__init__
(self, in_channels, out_channels, config, D=3, **kwargs)
models/resunet.py:24
↓ 2 callersMethod__init__
(self, channel, reduction=16, D=-1)
models/modules/senet_block.py:9
↓ 2 callersFunction_hash_coords
(coords, coords_min, coords_dim)
utils/pc_utils.py:675
↓ 2 callersMethod_read_json
(self, path)
datasets/preprocessing/matterport_preprocessing.py:168
↓ 2 callersMethod_remap_from_zero
(self, labels)
datasets/semseg.py:598
↓ 2 callersMethod_remap_model_output
(self, output)
datasets/outdoor_semseg.py:184
↓ 2 callersMethod_select_correct_labels
(self, labels, num_labels)
datasets/outdoor_semseg.py:156
↓ 2 callersMethod_transform_points
Base/wrapper method for transforming points using R and T.
utils/pc_utils.py:278
↓ 2 callersMethodadd
utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:445
↓ 2 callersMethodadd_grid_locations
(self, location)
pseudo_masks/datasets/scannet_free.py:23
↓ 2 callersFunctionadd_models
(module)
models/__init__.py:6
↓ 2 callersFunctionadd_models
(module)
models/encoders_2d/__init__.py:6
↓ 2 callersFunctionbbox_corner_dist_measure
compute distance between box corners to replace iou Args: crnr1, crnr2: Nx3 points of box corners in camera axis (y points down)
utils/votenet_utils/pc_util.py:323
↓ 2 callersFunctionbox3d_vol
corners: (8,3) no assumption on axis direction
utils/votenet_utils/box_util.py:80
next →1–100 of 1,028, ranked by callers