MCPcopy Create free account

hub / github.com/RozDavid/UnScene3D / functions

Functions1,028 in github.com/RozDavid/UnScene3D

⨍ Functions 1,028◇ Types & classes 235
↓ 2 callersFunctioncomputeIntersection
()
utils/votenet_utils/box_util.py:33
↓ 2 callersMethodcompute_averages
(self, avg_values, metric='ap')
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:228
↓ 2 callersFunctionconvert_oriented_box_to_trimesh_fmt
(box)
utils/votenet_utils/pc_util.py:407
↓ 2 callersFunctioncosine_sim
(feats_k, feats_q)
utils/freemask_utils.py:8
↓ 2 callersFunctioncrop
(points, x_min, y_min, z_min, x_max, y_max, z_max)
datasets/semseg.py:690
↓ 2 callersFunctiondraw_point_cloud
(coords, colors=None, label_text=None)
utils/pc_visualizations.py:21
↓ 2 callersFunctionelastic_distortion
Apply elastic distortion on sparse coordinate space. pointcloud: numpy array of (number of points, at least 3 spatial dims) granularity: size
datasets/semseg.py:651
↓ 2 callersFunctioneval_det
Generic functions to compute precision/recall for object detection for multiple classes. Input: pred_all: map of {img_id:
utils/votenet_utils/eval_det.py:168
↓ 2 callersFunctioneval_det_cls
Generic functions to compute precision/recall for object detection for a single class. Input: pred: map of {img_id: [(bbo
utils/votenet_utils/eval_det.py:77
↓ 2 callersMethodeval_step
(self, batch, batch_idx)
trainer/trainer.py:359
↓ 2 callersFunctionevaluate_matches
(matches)
benchmark/evaluate_semantic_instance.py:81
↓ 2 callersMethodexport
(self, pred_masks, scores, pred_classes, file_names, decoder_id)
trainer/trainer.py:168
↓ 2 callersFunctionflatten_dict
https://stackoverflow.com/questions/6027558/flatten-nested-dictionaries-compressing-keys
utils/utils.py:30
↓ 2 callersFunctionflip_in_center
(coordinates)
datasets/semseg.py:718
↓ 2 callersFunctionfmaxf
utils/cuda_utils/include/cutil_math.h:66
↓ 2 callersMethodforward
(self, x, point2segment=None, raw_coordinates=None, is_eval=False)
trainer/trainer.py:93
↓ 2 callersMethodgetNormalized
utils/cpp_utils/include/vec3.h:256
↓ 2 callersMethodget_full_cloud_by_scene_name
(self, scene_name)
pseudo_masks/datasets/s3dis.py:152
↓ 2 callersFunctionget_instance_freemasks
(list_freemasks, list_segments=None)
datasets/utils.py:480
↓ 2 callersFunctionget_instance_masks
(list_labels, task, list_segments=None, ignore_class_threshold=100, filter_out_classes=
datasets/utils.py:529
↓ 2 callersMethodget_loss
(self, loss, outputs, targets, indices, num_masks, mask_type, coords=None)
models/criterion.py:231
↓ 2 callersFunctionget_parameters
(cfg: DictConfig)
main_instance_segmentation.py:21
↓ 2 callersMethodget_transformation_matrix
(self)
pseudo_masks/datasets/voxelizer.py:44
↓ 2 callersFunctionheading2rotmat
(heading_angle)
utils/votenet_utils/pc_util.py:398
↓ 2 callersMethodhsv_to_rgb
(hsv)
utils/transforms.py:120
↓ 2 callersMethodinverse
utils/cuda_utils/include/cuda_SimpleMatrixUtil.h:756
↓ 2 callersMethodjoint_database
(self, train_modes=["train", "validation"])
datasets/preprocessing/base_preprocessing.py:151
↓ 2 callersFunctionl2_sim
(feats_k, feats_q)
utils/freemask_utils.py:20
↓ 2 callersMethodlength
utils/cpp_utils/include/vec3.h:187
↓ 2 callersFunctionload_checkpoint_with_missing_or_exsessive_keys
(cfg, model)
utils/utils.py:95
↓ 2 callersMethodload_data
(self, index, data_root=None)
pseudo_masks/datasets/dataset.py:189
↓ 2 callersMethodload_image
(self, path, target_shape, label=False)
pseudo_masks/datasets/scannet.py:118
↓ 2 callersMethodload_ply
(self, index)
pseudo_masks/datasets/s3dis.py:107
↓ 2 callersMethodload_rgb_data
(self, scene_name, info_dict, axis_alignment)
pseudo_masks/datasets/scannet.py:126
↓ 2 callersMethodload_scene_data
(self, index, data_root=None)
pseudo_masks/datasets/scannet.py:218
↓ 2 callersMethodload_scene_data
(self, index)
pseudo_masks/datasets/arkit.py:110
↓ 2 callersFunctionlog
utils/cpp_utils/include/vec3.h:417
↓ 2 callersFunctionmain
(min_vert_num=50)
utils/cpp_utils/test/test_segmentor.py:79
↓ 2 callersFunctionmake_pred_info
(pred: dict)
benchmark/evaluate_semantic_instance.py:262
↓ 2 callersFunctionnn_distance
Input: pc1: (B,N,C) torch tensor pc2: (B,M,C) torch tensor l1smooth: bool, whether to use l1smooth loss delta: sc
utils/votenet_utils/nn_distance.py:34
↓ 2 callersFunctionopt_n_threads
utils/pointops2/src/cuda_utils.h:11
↓ 2 callersFunctionplot_polys
(plist,scale=500.0)
utils/votenet_utils/box_util.py:256
↓ 2 callersFunctionpoint_cloud_to_bbox
Extract the axis aligned box from a pcl or batch of pcls Args: points: Nx3 points or BxNx3 output is 6 dim: xyz pos of center and
utils/votenet_utils/pc_util.py:346
↓ 2 callersFunctionpolygon_clip
Clip a polygon with another polygon. Ref: https://rosettacode.org/wiki/Sutherland-Hodgman_polygon_clipping#Python Args: subjectPolygon:
utils/votenet_utils/box_util.py:16
↓ 2 callersFunctionprint_metric_results
(avgs, metric='ap')
benchmark/evaluate_semantic_instance.py:402
↓ 2 callersMethodprint_results
(self, avgs, metric='ap')
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:304
↓ 2 callersFunctionrandom_around_points
( coordinates, color, normals, labels, rate=0.2, noise_rate=0, ignore_label=255 )
datasets/semseg.py:763
↓ 2 callersFunctionrandom_sampling
Input is NxC, output is num_samplexC
utils/votenet_utils/pc_util.py:35
↓ 2 callersMethodsave_visualizations
(self, target_full, full_res_coords, sorted_masks, sort_classes, file_name, origin
trainer/trainer.py:197
↓ 2 callersFunctionsegment_ids_to_mask
(selected_ids, unique_segments)
pseudo_masks/unscene3d_pseudo_main.py:254
↓ 2 callersFunctionsegment_scene
(coords, feats, batch, model, config, dataset, segment_ids, seg_connectivity, visualize=False, filter_large_se
pseudo_masks/unscene3d_pseudo_main.py:505
↓ 2 callersFunctionshift_scale_points
pred_xyz: B x N x 3 src_range: [[B x 3], [B x 3]] - min and max XYZ coords dst_range: [[B x 3], [B x 3]] - min and max XYZ coords
models/position_embedding.py:12
↓ 2 callersMethodsplitPointCloud
(self, cloud, size=50.0, stride=50, inner_core=-1)
datasets/semseg.py:280
↓ 2 callersMethodto_dict
(self)
benchmark/util_3d.py:105
↓ 2 callersFunctionvoxelize
(batch, ignore_label, voxel_size, probing, mode, task, ignore_class_threshold, filter_out_classes
datasets/utils.py:235
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos)
models/mask3d.py:511
↓ 2 callersMethodwrite_ply
(array, filepath)
utils/pc_utils.py:415
↓ 1 callersFunctionM
(axis, theta)
pseudo_masks/datasets/voxelizer.py:9
↓ 1 callersFunctionRAW2SCANNET
(label)
pseudo_masks/datasets/preprocess/scannet200_insseg.py:22
↓ 1 callersMethod__getitem__
(self, index)
pseudo_masks/datasets/dataset.py:140
↓ 1 callersMethod__init__
(self)
utils/utils.py:372
↓ 1 callersMethod__init__
(self, instance_id, instance_label, coords, cloud_instances, class_label=-1)
pseudo_masks/datasets/scannet_free.py:10
↓ 1 callersMethod__init__
(self, config=None, base_loss='bce', directions='xyz')
models/noise_robust_loss.py:107
↓ 1 callersMethod__init__
(self, in_channels, out_channels, config, D=3, out_fpn=False, **kwargs)
models/res16unet.py:22
↓ 1 callersMethod__init__
(self, in_channels, out_channels, config, D, **kwargs)
models/model.py:13
↓ 1 callersMethod__init__
(self, in_channels, out_channels, config, D=3, **kwargs)
models/resnet.py:18
↓ 1 callersMethod__init__
( self, inplanes, planes, stride=1, dilation=1, downsample=Non
models/modules/resnet_block.py:11
↓ 1 callersMethod__iter__
(self)
pseudo_masks/datasets/dataloader.py:30
↓ 1 callersMethod_augment_coords_to_feats
(self, coords, feats, labels=None)
pseudo_masks/datasets/dataset.py:297
↓ 1 callersMethod_buf_count_newlines_gen
(self, fname)
datasets/preprocessing/s3dis_preprocessing.py:72
↓ 1 callersMethod_dict_to_yaml
(cls, dictionary)
datasets/preprocessing/base_preprocessing.py:169
↓ 1 callersMethod_get_src_permutation_idx
(self, indices)
models/criterion.py:219
↓ 1 callersFunction_max_by_axis
(the_list)
models/misc.py:16
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
models/misc.py:76
↓ 1 callersMethod_parse_scene_subscene
(self, name)
datasets/preprocessing/freemask_preprocessing.py:239
↓ 1 callersMethod_parse_scene_subscene
(self, name)
datasets/preprocessing/arkit_preprocessing.py:168
↓ 1 callersMethod_parse_scene_subscene
(self, name)
datasets/preprocessing/scannet_preprocessing.py:248
↓ 1 callersMethod_remap_from_zero
(self, labels)
datasets/outdoor_semseg.py:177
↓ 1 callersMethod_reset_parameters
(self)
models/mask3d.py:506
↓ 1 callersMethod_reset_parameters
(self)
models/mask3d.py:563
↓ 1 callersMethod_reset_parameters
(self)
models/mask3d.py:628
↓ 1 callersMethod_select_correct_labels
(labels, num_labels)
datasets/semseg.py:577
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
models/mask3d.py:449
↓ 1 callersFunctionabs
utils/cpp_utils/include/vec3.h:392
↓ 1 callersFunctionadd_cutler_config
(cfg)
pseudo_masks/eval_cutler_projection_performance.py:27
↓ 1 callersFunctionadd_cutler_config
(cfg)
pseudo_masks/eval_cutler_images.py:29
↓ 1 callersFunctionaffinity
(sim, sigma)
pseudo_masks/eval_3duis_performance.py:91
↓ 1 callersFunctionarray_to_sequence
(batch_data)
pseudo_masks/eval_3duis_performance.py:99
↓ 1 callersFunctionarray_to_torch_sequence
(batch_data)
pseudo_masks/eval_3duis_performance.py:103
↓ 1 callersFunctionassign_instances_for_scan
(pred: dict, gt_file: str)
benchmark/evaluate_semantic_instance.py:275
↓ 1 callersMethodassign_instances_for_scan
(self, scene_id)
pseudo_masks/datasets/evaluation/evaluate_semantic_instance.py:246
↓ 1 callersFunctionassign_instances_for_scan_with_gt
(pred: dict, gt_ids)
benchmark/evaluate_semantic_instance.py:339
↓ 1 callersMethodaugment_individual_instance
( self, coordinates, color, normals, labels, oversampling=1.0 )
datasets/semseg.py:612
↓ 1 callersFunctionbatch_instances
(batch)
datasets/utils.py:221
↓ 1 callersFunctionbox3d_iou
Compute 3D bounding box IoU. Input: corners1: numpy array (8,3), assume up direction is negative Y corners2: numpy array (8,3),
utils/votenet_utils/box_util.py:92
↓ 1 callersMethodbuild_camera_matrix
Build the 3x3 camera matrix K using the given intrinsics. Equation 6.10 from HZ.
utils/pc_utils.py:134
↓ 1 callersFunctionbuild_graph
(feats, slc, points, center, source_num, cluster_idx, sink_factor=2, source_factor=2,
pseudo_masks/eval_3duis_performance.py:120
↓ 1 callersMethodcamera2world
Transform from camera coordinates (3D) to world coordinates (3D). Args: points_3d: Nx3 or Nx6 matrix of N points with XYZ or XYZRGB
utils/pc_utils.py:299
↓ 1 callersMethodchunks
Yield successive n-sized chunks from lst.
datasets/outdoor_semseg.py:76
↓ 1 callersFunctionclean_mesh
(scene_mesh)
utils/cpp_utils/test/test_segmentor.py:39
← previousnext →101–200 of 1,028, ranked by callers