MCPcopy Create free account

hub / github.com/Atrovast/THGS / functions

Functions879 in github.com/Atrovast/THGS

Methodcallback_3ddel
(sender, app_data)
gui/main.py:736
Methodcallback_3dseg
(sender, app_data)
gui/main.py:727
Methodcallback_camera_drag_pan
(sender, app_data)
gui/main.py:1011
Methodcallback_camera_drag_rotate_or_draw_mask
(sender, app_data)
gui/main.py:991
Methodcallback_camera_wheel_scale
(sender, app_data)
gui/main.py:1002
Methodcallback_change_mode
(sender, app_data)
gui/main.py:775
Methodcallback_click_handler
(sender, app_data)
gui/main.py:969
Methodcallback_key_press
(sender, app_data)
gui/main.py:1035
Methodcallback_load_new_eval_data
(sender, app_data, user_data)
gui/main.py:612
Methodcallback_load_new_scene
(sender, app_data, user_data)
gui/main.py:591
Methodcallback_retrieval
(sender, app_data)
gui/main.py:716
Methodcallback_retrieval_exit
(sender, app_data)
gui/main.py:751
Methodcallback_retrieval_reset
(sender, app_data)
gui/main.py:745
Methodcallback_save_current_view
(sender, app_data)
gui/main.py:648
Methodcallback_set_color_overlay_ratio
(sender, app_data)
gui/main.py:880
Methodcallback_set_fovy
(sender, app_data)
gui/main.py:787
Methodcallback_set_gaussain_scale
(sender, app_data)
gui/main.py:800
Methodcallback_set_mouse_loc
(sender, app_data)
gui/main.py:1021
Methodcallback_set_seed
(sender, app_data)
gui/main.py:578
Methodcallback_set_topk
(sender, app_data)
gui/main.py:695
Methodcallback_setattr
(sender, app_data, user_data)
gui/main.py:567
Methodcallback_update_clip_mask
(sender, app_data)
gui/main.py:682
Methodcallback_update_res_prompt
(sender, app_data)
gui/main.py:946
Methodcallback_update_sim_binary
(sender, app_data)
gui/main.py:904
Methodcallback_update_sim_color
(sender, app_data)
gui/main.py:894
Methodcampos
(self)
gui/cam_utils.py:165
Methodcapture
(self)
scene/gaussian_model.py:63
Methodchange_save_path
(sender, app_data)
gui/main.py:570
Functionchange_title
(chooser)
ext/spt/utils/widgets.py:128
Functioncolormap
(img, cmap='jet')
utils/general_utils.py:163
Functioncompute_mask_ratio
(refer_mask, mask)
utils/image_utils.py:79
Functioncompute_panoptic_metrics_s3dis_6fold
Helper function to compute the semantic, instance, panoptic segmentation metrics of a model on a S3DIS 6-fold, for given instance graph and pa
ext/spt/utils/instance.py:944
Methodcompute_relative_gs_index
(self)
gui/main.py:341
Methodcompute_sdf_perframe
compute per frame sdf
utils/mesh_utils.py:197
Functioncompute_semantic_metrics_s3dis_6fold
Helper function to compute the semantic segmentation metrics of a model on a S3DIS 6-fold. :param fold_ckpt: dict Dictionary with S3D
ext/spt/utils/semantic.py:81
Methodcompute_similarity
(self, semantic_feature)
utils/vlm_utils.py:34
Methodcompute_unbounded_tsdf
Fusion all frames, perform adaptive sdf_funcation on the contract spaces.
utils/mesh_utils.py:210
Functionconstruct_saving_name
(args)
utils/sai3d_utils.py:50
Methodcontract
(x)
utils/mesh_utils.py:189
Methodcpu
Move the CSRData to the CPU.
ext/spt/data/csr.py:137
Functioncreate_rotation_matrix_from_direction_vector_batch
(direction_vectors)
utils/general_utils.py:138
Methodcuda
Move the NAG with all Data in it to CUDA.
ext/spt/data/data.py:204
Methodcuda
Move the CSRData to the first available GPU.
ext/spt/data/csr.py:141
Functiondated_dir
Returns a directory path in root, named based on the current date and time.
ext/spt/utils/io.py:31
Methoddebug
(self)
ext/spt/utils/output_panoptic.py:58
Methoddebug
(self)
ext/spt/data/cluster.py:135
Methoddebug
(self)
ext/spt/data/instance.py:611
Functiondecorator
(func)
utils/linetimer.py:145
Methoddensify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size)
scene/gaussian_model.py:467
Methoddetach
Extend `torch_geometric.Data.detach` to handle Cluster and InstanceData attributes.
ext/spt/data/data.py:180
Methoddetach
Detach all tensors in the CSRBatch.
ext/spt/data/csr.py:588
Methoddevice
Device of the first-encountered tensor in 'self'.
ext/spt/data/data.py:209
Methoddevice
(self)
ext/spt/data/csr.py:146
Functiondownload_url
Download the content of an URL to a specific folder. :param url: string :param folder: string :param log: bool If `False`, will n
ext/spt/utils/download.py:8
Functiondraw_adj_distribuution
(adj, save_path)
utils/sai3d_utils.py:129
Methodedge_affinity_pred
Simply applies a sigmoid on `edge_affinity_logits` to produce the actual affinity predictions to be used for superpoint graph clusteri
ext/spt/utils/output_panoptic.py:150
Methodedge_keys
All keys starting with `edge_`, apart from `edge_index` and `edge_attr`.
ext/spt/data/data.py:136
Methodedit_delete
(self)
gui/main.py:421
Methodembedding_dim
(self)
scripts/image_encoding.py:82
Methodencode_text
(self, text)
utils/vlm_utils.py:26
Functionenforce_tags
Prompts user to input tags from command line if no tags are provided in config.
ext/spt/utils/rich_utils.py:79
Functionexplode_transform
Extracts a flattened list of transforms from a Compose or from a list of transforms.
ext/spt/transforms/__init__.py:175
Functionexport_res_to_class_agnostc_eval
(base_dir, scene_id, pol_name, semantic_name=None)
utils/sai3d_utils.py:965
Functionexport_small_gt_to_class_agnostic_eval
(base_dir, scene_id)
utils/sai3d_utils.py:1017
Methodextract
(self, args)
arguments/__init__.py:63
Functionfast_zeros
Same as torch.zeros but relies numpy on CPU. This may be x40 faster when manipulating large tensors on CPU. ``` from time import time
ext/spt/utils/tensor.py:275
Functionfilter_by_population
filter out small groups of points
utils/sai3d_utils.py:870
Methodfinetune_sh_setup
(self, training_args)
scene/gaussian_model.py:178
Methodforward
(self, x)
scene/semantic_model.py:8
Methodforward
(self, semantic_features)
scene/semantic_model.py:45
Methodforward
(self, x: torch.Tensor)
lpipsPyTorch/modules/networks.py:53
Methodforward
(self, x: torch.Tensor, y: torch.Tensor)
lpipsPyTorch/modules/lpips.py:30
Functionfourier_position_encoder
Heuristic: keeping ```f_min = 1 / f_max``` ensures that roughly 50% of the encoding dimensions are untouched and free to use. This is imp
ext/spt/utils/encoding.py:7
Methodfull_res_panoptic_pred
Panoptic predictions on the full-resolution input point cloud. Returns the predicted semantic label and instance index for each point,
ext/spt/utils/output_panoptic.py:430
Methodfull_res_semantic_pred
Semantic predictions on the full-resolution input point cloud. Final semantic segmentation predictions are the argmax of the
ext/spt/utils/output_semantic.py:139
Functiongarbage_collection_cuda
Garbage collection Torch (CUDA) memory.
ext/spt/utils/memory.py:57
Functiongaussian_3d_coeff
(xyzs, covs)
gui/gs_renderer.py:65
Functiongenerate_random_bbox_data
( num_img=1, num_classes=1, height=128, width=128, h_split=1,
ext/spt/utils/instance.py:43
Functiongenerate_random_segment_data
Generate multiple images with random ground truth and predicted instance and semantic segmentation data. To make the images realistic, and to
ext/spt/utils/instance.py:300
Functiongeom_transform_points
(points, transf_matrix)
utils/graphics_utils.py:22
FunctiongetWorld2View
(R, t)
utils/graphics_utils.py:31
Methodget_base_class
Helps `self.from_list()` and `self.to_list()` identify which classes to use for batch collation and un-collation.
ext/spt/data/instance.py:98
Methodget_batch_class
Helps `self.from_list()` and `self.to_list()` identify which classes to use for batch collation and un-collation.
ext/spt/data/cluster.py:40
Methodget_batch_class
Helps `self.from_list()` and `self.to_list()` identify which classes to use for batch collation and un-collation.
ext/spt/data/instance.py:105
Methodget_cam_location
(self)
gui/cam_utils.py:254
Functionget_expon_lr_func
( lr_init, lr_final, lr_delay_steps=0, lr_delay_mult=1.0, max_steps=1000000 )
gui/gs_renderer.py:26
Methodget_features
(self)
scene/gaussian_model.py:112
Functionget_matterport_color
(color_path)
utils/sai3d_utils.py:656
Functionget_matterport_depth
(color_path)
utils/sai3d_utils.py:661
Functionget_matterport_intrinsic
(color_path)
utils/sai3d_utils.py:678
Functionget_matterport_mask
(base_dir, scene_id, color_name, mask_name, region_id='region4')
utils/sai3d_utils.py:688
Functionget_matterport_pose
(color_path)
utils/sai3d_utils.py:683
Functionget_matterport_semantic_mask
(color_path, semantic)
utils/sai3d_utils.py:703
Functionget_metric_value
Safely retrieves value of the metric logged in LightningModule.
ext/spt/utils/utils.py:181
Methodget_opacity
(self)
scene/gaussian_model.py:122
Functionget_points_from_openscene_pth
(base_dir, scene_id)
utils/sai3d_utils.py:712
Functionget_points_from_ply
(ply_path)
utils/sai3d_utils.py:21
Functionget_pylogger
Initializes multi-GPU-friendly python command line logger.
ext/spt/utils/pylogger.py:6
Methodget_relevancy
(self, embed: torch.Tensor, positive_id: int)
scripts/image_encoding.py:95
Methodget_rotation
(self)
scene/gaussian_model.py:104
← previousnext →601–700 of 879, ranked by callers