MCPcopy Create free account

hub / github.com/PRBonn/SHINE_mapping / functions

Functions116 in github.com/PRBonn/SHINE_mapping

↓ 12 callersFunctionget_time
:return: get timing statistics
utils/tools.py:232
↓ 8 callersMethodquery_feature
(self, coord, faster = False)
model/feature_octree.py:237
↓ 8 callersMethodsdf
(self, sum_features)
model/decoder.py:49
↓ 7 callersMethodload
(self, config_file)
utils/config.py:206
↓ 6 callersFunctionfreeze_model
(model: nn.Module)
utils/tools.py:188
↓ 5 callersMethod_register_key_callback
(self, keys: List, callback: Callable)
utils/visualizer.py:107
↓ 4 callersFunctionget_gradient
(inputs, outputs)
utils/tools.py:175
↓ 4 callersMethodprint_detail
(self)
model/feature_octree.py:288
↓ 4 callersMethodrecon_bbx_mesh
(self, bbx, voxel_size, mesh_path, map_path, \ save_map = False, estimate_sem = False, estimate_normal
utils/mesher.py:253
↓ 4 callersMethodupdate
(self, scan, pose, mesh = None)
utils/visualizer.py:67
↓ 4 callersMethodupdate_view
(self)
utils/visualizer.py:57
↓ 3 callersMethodget_batch
(self)
dataset/lidar_dataset.py:401
↓ 3 callersMethodpause_view
(self)
utils/visualizer.py:61
↓ 3 callersMethodquery_points
query the sdf value, semantic label and marching cubes mask for points Args: coord: Nx3 torch tensor, the coordinates of all N (a
utils/mesher.py:33
↓ 3 callersMethodread_point_cloud
(self, filename: str)
dataset/lidar_dataset.py:283
↓ 3 callersMethodrecon_octree_mesh
(self, query_level, mc_res_m, mesh_path, map_path, \ save_map = False, estimate_sem
utils/mesher.py:294
↓ 3 callersFunctionsdf_bce_loss
(pred, label, sigma, weight, weighted=False, bce_reduction = "mean")
utils/loss.py:17
↓ 3 callersMethodsem_label_prob
(self, sum_features)
model/decoder.py:89
↓ 3 callersMethodstop
(self)
utils/visualizer.py:85
↓ 3 callersMethodtime_conditionded_sdf
(self, sum_features, ts)
model/decoder.py:65
↓ 2 callersMethodassign_to_bbx
assign the queried sdf, semantic label and marching cubes mask back to the 3D grids in the specified bounding box Args: sdf_pred:
utils/mesher.py:175
↓ 2 callersFunctionbatch_ray_rendering_loss
(x, y, d_meas, neus_on=True)
utils/loss.py:82
↓ 2 callersMethodestimate_vertices_sem
(self, mesh, verts, filter_free_space_vertices = True)
utils/mesher.py:224
↓ 2 callersMethodfilter_isolated_vertices
(self, mesh, filter_cluster_min_tri = 300)
utils/mesher.py:240
↓ 2 callersMethodget_indices
(self, coord)
model/feature_octree.py:199
↓ 2 callersMethodmc_mesh
use the marching cubes algorithm to get mesh vertices and faces Args: mc_sdf: a*b*c np.array, 3d grids of sign distance values
utils/mesher.py:200
↓ 2 callersFunctionnn_correspondance
for each vertex in verts2 find the nearest vertex in verts1 Args: nx3 np.array's scalar truncation_dist: points whose nearest nei
eval/eval_utils.py:110
↓ 2 callersMethodprocess_frame
(self, frame_id, incremental_on = False)
dataset/lidar_dataset.py:115
↓ 2 callersFunctionsdf_diff_loss
(pred, label, weight, scale, l2_loss=True)
utils/loss.py:6
↓ 2 callersMethodsem_label
(self, sum_features)
model/decoder.py:99
↓ 2 callersFunctionsetup_experiment
(config: SHINEConfig)
utils/tools.py:23
↓ 2 callersFunctionsetup_optimizer
(config: SHINEConfig, octree_feat, mlp_geo_param, mlp_sem_param, sigma_size)
utils/tools.py:57
↓ 2 callersMethodupdate_mesh
(self, mesh)
utils/visualizer.py:77
↓ 1 callersMethod_initialize_visualizer
(self)
utils/visualizer.py:94
↓ 1 callersMethod_register_key_callbacks
(self)
utils/visualizer.py:111
↓ 1 callersMethod_set_white_background
(self, vis)
utils/visualizer.py:123
↓ 1 callersMethod_update_geometries
(self, scan, pose, mesh = None)
utils/visualizer.py:165
↓ 1 callersMethod_update_mesh
(self, mesh)
utils/visualizer.py:145
↓ 1 callersMethod_update_pointcloud
(self, scan)
utils/visualizer.py:155
↓ 1 callersFunctioncal_feature_importance
(data: LiDARDataset, octree: FeatureOctree, mlp: Decoder, sigma, bs, down_rate=1, loss_reduction='mean',
utils/incre_learning.py:8
↓ 1 callersMethodcal_regularization
(self)
model/feature_octree.py:246
↓ 1 callersMethodcalculate_world_scale
(self)
utils/config.py:372
↓ 1 callersMethodconvertdata
(self, labelscan)
utils/semantic_kitti_utils.py:11
↓ 1 callersFunctioncrop_intersection
Get the cropped ground truth point cloud according to the intersection of the predicted mesh by different methods Args: file_gt: file
eval/eval_utils.py:180
↓ 1 callersFunctioncsv_odom_to_transforms
(path)
utils/pose.py:61
↓ 1 callersMethoddestroy_window
(self)
utils/visualizer.py:82
↓ 1 callersFunctioneval_mesh
Compute Mesh metrics between prediction and target. Opens the Meshs and runs the metrics Args: file_pred: file path of prediction (sh
eval/eval_utils.py:24
↓ 1 callersMethodgenerate_sdf_map
(self, coord, sdf_pred, mc_mask, map_path)
utils/mesher.py:152
↓ 1 callersMethodget_indices_fast
(self, coord)
model/feature_octree.py:267
↓ 1 callersMethodget_octree_nodes
(self, level)
model/feature_octree.py:94
↓ 1 callersMethodget_query_from_bbx
get grid query points inside a given bounding box (bbx) Args: bbx: open3d bounding box, in world coordinate system, with unit m
utils/mesher.py:110
↓ 1 callersMethodinterpolat
(self, x, level, polynomial_on = True)
model/feature_octree.py:172
↓ 1 callersMethodlist_duplicates
(self, seq)
model/feature_octree.py:257
↓ 1 callersFunctionload_focal_length
(filepath)
dataset/rgbd_to_kitti_format.py:119
↓ 1 callersFunctionload_from_json
Load a dictionary from a JSON filename. Args: filename: The filename to load from.
dataset/rgbd_to_kitti_format.py:110
↓ 1 callersFunctionload_poses
(posefile, with_head = False)
dataset/rgbd_to_kitti_format.py:123
↓ 1 callersMethodpreprocess_kitti
(self, points, z_th=-3.0, min_range=2.5)
dataset/lidar_dataset.py:334
↓ 1 callersMethodpreprocess_sem_kitti
(self, points, labels, min_range=2.75, filter_outlier = True, filter_moving = True)
dataset/lidar_dataset.py:341
↓ 1 callersMethodquery_feature_with_indices
(self, coord, hierarchical_indices)
model/feature_octree.py:222
↓ 1 callersFunctionread_calib_file
read calibration file (with the kitti format) returns -> dict calibration matrices as 4*4 numpy arrays
utils/pose.py:7
↓ 1 callersFunctionread_poses_file
read pose file (with the kitti format)
utils/pose.py:33
↓ 1 callersMethodread_semantic_point_label
(self, bin_filename: str, label_filename: str)
dataset/lidar_dataset.py:301
↓ 1 callersFunctionrgbd_to_kitti_format
(args)
dataset/rgbd_to_kitti_format.py:10
↓ 1 callersFunctionrun_shine_mapping_batch
()
shine_batch.py:23
↓ 1 callersFunctionrun_shine_mapping_incremental
()
shine_incre.py:22
↓ 1 callersMethodsample
(self, points_torch, sensor_origin_torch, normal_torch, sem_labe
utils/data_sampler.py:18
↓ 1 callersFunctionsave_checkpoint
( feature_octree, geo_decoder, sem_decoder, optimizer, run_path, checkpoint_name, iters )
utils/tools.py:200
↓ 1 callersFunctionsave_decoder
(geo_decoder, sem_decoder, run_path, checkpoint_name)
utils/tools.py:216
↓ 1 callersMethodset_zero
(self)
model/feature_octree.py:78
↓ 1 callersFunctionsetup_wandb
()
utils/tools.py:120
↓ 1 callersFunctionstep_lr_decay
( optimizer: Optimizer, learning_rate: float, iteration_number: int, steps: List, reduce:
utils/tools.py:135
↓ 1 callersMethodwrite_merged_pc
(self, out_path)
dataset/lidar_dataset.py:364
↓ 1 callersFunctionwrite_poses_kitti_format
(poses_mat, posefile)
dataset/rgbd_to_kitti_format.py:141
Method__getitem__
(self, index: int)
dataset/lidar_dataset.py:377
Method__init__
(self)
utils/config.py:7
Method__init__
(self, config: SHINEConfig)
utils/data_sampler.py:11
Method__init__
(self, config: SHINEConfig, octree: FeatureOctree, \ geo_decoder: Decoder, sem_decoder: Decoder)
utils/mesher.py:16
Method__init__
(self)
utils/visualizer.py:30
Method__init__
(self, labelscan)
utils/semantic_kitti_utils.py:7
Method__init__
(self, config: SHINEConfig, octree: FeatureOctree = None)
dataset/lidar_dataset.py:23
Method__init__
(self, config: SHINEConfig)
model/feature_octree.py:31
Method__init__
(self, config: SHINEConfig, is_geo_encoder = True, is_time_conditioned = False)
model/decoder.py:10
Method__len__
(self)
dataset/lidar_dataset.py:370
Method_next_frame
(self, vis)
utils/visualizer.py:131
Method_quit
(self, vis)
utils/visualizer.py:126
Method_set_black_background
(self, vis)
utils/visualizer.py:120
Method_start_stop
(self, vis)
utils/visualizer.py:134
Method_toggle_frame
(self, vis)
utils/visualizer.py:137
Method_toggle_map
(self, vis)
utils/visualizer.py:141
Method_toggle_view
(self, vis)
utils/visualizer.py:194
Functionalphanum_key
Turn a string into a list of string and number chunks. "z23a" -> ["z", 23, "a"]
dataset/rgbd_to_kitti_format.py:104
Methodclear_temp
(self)
model/feature_octree.py:107
Functioneval_depth
Computes 2d metrics between two depth maps Args: depth_pred: mxn np.array containing prediction depth_trgt: mxn np.array containi
eval/eval_utils.py:144
Methodforward
(self, x)
model/feature_octree.py:83
Methodforward
(self, feature)
model/decoder.py:43
Functionget_dict_values
(dictionary, keys, default=None)
model/feature_octree.py:17
Methodget_morton
(self, sample_points, level)
model/feature_octree.py:87
Functionget_random_rgb
(n)
utils/semantic_kitti_utils.py:35
Methodis_empty
(self)
model/feature_octree.py:103
Functionload_from_json
Load a dictionary from a JSON filename. Args: filename: The filename to load from.
utils/tools.py:239
next →1–100 of 116, ranked by callers