Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Atrovast/THGS
/ functions
Functions
879 in github.com/Atrovast/THGS
⨍
Functions
879
◇
Types & classes
117
Method
callback_3ddel
(sender, app_data)
gui/main.py:736
Method
callback_3dseg
(sender, app_data)
gui/main.py:727
Method
callback_camera_drag_pan
(sender, app_data)
gui/main.py:1011
Method
callback_camera_drag_rotate_or_draw_mask
(sender, app_data)
gui/main.py:991
Method
callback_camera_wheel_scale
(sender, app_data)
gui/main.py:1002
Method
callback_change_mode
(sender, app_data)
gui/main.py:775
Method
callback_click_handler
(sender, app_data)
gui/main.py:969
Method
callback_key_press
(sender, app_data)
gui/main.py:1035
Method
callback_load_new_eval_data
(sender, app_data, user_data)
gui/main.py:612
Method
callback_load_new_scene
(sender, app_data, user_data)
gui/main.py:591
Method
callback_retrieval
(sender, app_data)
gui/main.py:716
Method
callback_retrieval_exit
(sender, app_data)
gui/main.py:751
Method
callback_retrieval_reset
(sender, app_data)
gui/main.py:745
Method
callback_save_current_view
(sender, app_data)
gui/main.py:648
Method
callback_set_color_overlay_ratio
(sender, app_data)
gui/main.py:880
Method
callback_set_fovy
(sender, app_data)
gui/main.py:787
Method
callback_set_gaussain_scale
(sender, app_data)
gui/main.py:800
Method
callback_set_mouse_loc
(sender, app_data)
gui/main.py:1021
Method
callback_set_seed
(sender, app_data)
gui/main.py:578
Method
callback_set_topk
(sender, app_data)
gui/main.py:695
Method
callback_setattr
(sender, app_data, user_data)
gui/main.py:567
Method
callback_update_clip_mask
(sender, app_data)
gui/main.py:682
Method
callback_update_res_prompt
(sender, app_data)
gui/main.py:946
Method
callback_update_sim_binary
(sender, app_data)
gui/main.py:904
Method
callback_update_sim_color
(sender, app_data)
gui/main.py:894
Method
campos
(self)
gui/cam_utils.py:165
Method
capture
(self)
scene/gaussian_model.py:63
Method
change_save_path
(sender, app_data)
gui/main.py:570
Function
change_title
(chooser)
ext/spt/utils/widgets.py:128
Function
colormap
(img, cmap='jet')
utils/general_utils.py:163
Function
compute_mask_ratio
(refer_mask, mask)
utils/image_utils.py:79
Function
compute_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
Method
compute_relative_gs_index
(self)
gui/main.py:341
Method
compute_sdf_perframe
compute per frame sdf
utils/mesh_utils.py:197
Function
compute_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
Method
compute_similarity
(self, semantic_feature)
utils/vlm_utils.py:34
Method
compute_unbounded_tsdf
Fusion all frames, perform adaptive sdf_funcation on the contract spaces.
utils/mesh_utils.py:210
Function
construct_saving_name
(args)
utils/sai3d_utils.py:50
Method
contract
(x)
utils/mesh_utils.py:189
Method
cpu
Move the CSRData to the CPU.
ext/spt/data/csr.py:137
Function
create_rotation_matrix_from_direction_vector_batch
(direction_vectors)
utils/general_utils.py:138
Method
cuda
Move the NAG with all Data in it to CUDA.
ext/spt/data/data.py:204
Method
cuda
Move the CSRData to the first available GPU.
ext/spt/data/csr.py:141
Function
dated_dir
Returns a directory path in root, named based on the current date and time.
ext/spt/utils/io.py:31
Method
debug
(self)
ext/spt/utils/output_panoptic.py:58
Method
debug
(self)
ext/spt/data/cluster.py:135
Method
debug
(self)
ext/spt/data/instance.py:611
Function
decorator
(func)
utils/linetimer.py:145
Method
densify_and_prune
(self, max_grad, min_opacity, extent, max_screen_size)
scene/gaussian_model.py:467
Method
detach
Extend `torch_geometric.Data.detach` to handle Cluster and InstanceData attributes.
ext/spt/data/data.py:180
Method
detach
Detach all tensors in the CSRBatch.
ext/spt/data/csr.py:588
Method
device
Device of the first-encountered tensor in 'self'.
ext/spt/data/data.py:209
Method
device
(self)
ext/spt/data/csr.py:146
Function
download_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
Function
draw_adj_distribuution
(adj, save_path)
utils/sai3d_utils.py:129
Method
edge_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
Method
edge_keys
All keys starting with `edge_`, apart from `edge_index` and `edge_attr`.
ext/spt/data/data.py:136
Method
edit_delete
(self)
gui/main.py:421
Method
embedding_dim
(self)
scripts/image_encoding.py:82
Method
encode_text
(self, text)
utils/vlm_utils.py:26
Function
enforce_tags
Prompts user to input tags from command line if no tags are provided in config.
ext/spt/utils/rich_utils.py:79
Function
explode_transform
Extracts a flattened list of transforms from a Compose or from a list of transforms.
ext/spt/transforms/__init__.py:175
Function
export_res_to_class_agnostc_eval
(base_dir, scene_id, pol_name, semantic_name=None)
utils/sai3d_utils.py:965
Function
export_small_gt_to_class_agnostic_eval
(base_dir, scene_id)
utils/sai3d_utils.py:1017
Method
extract
(self, args)
arguments/__init__.py:63
Function
fast_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
Function
filter_by_population
filter out small groups of points
utils/sai3d_utils.py:870
Method
finetune_sh_setup
(self, training_args)
scene/gaussian_model.py:178
Method
forward
(self, x)
scene/semantic_model.py:8
Method
forward
(self, semantic_features)
scene/semantic_model.py:45
Method
forward
(self, x: torch.Tensor)
lpipsPyTorch/modules/networks.py:53
Method
forward
(self, x: torch.Tensor, y: torch.Tensor)
lpipsPyTorch/modules/lpips.py:30
Function
fourier_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
Method
full_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
Method
full_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
Function
garbage_collection_cuda
Garbage collection Torch (CUDA) memory.
ext/spt/utils/memory.py:57
Function
gaussian_3d_coeff
(xyzs, covs)
gui/gs_renderer.py:65
Function
generate_random_bbox_data
( num_img=1, num_classes=1, height=128, width=128, h_split=1,
ext/spt/utils/instance.py:43
Function
generate_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
Function
geom_transform_points
(points, transf_matrix)
utils/graphics_utils.py:22
Function
getWorld2View
(R, t)
utils/graphics_utils.py:31
Method
get_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
Method
get_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
Method
get_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
Method
get_cam_location
(self)
gui/cam_utils.py:254
Function
get_expon_lr_func
( lr_init, lr_final, lr_delay_steps=0, lr_delay_mult=1.0, max_steps=1000000 )
gui/gs_renderer.py:26
Method
get_features
(self)
scene/gaussian_model.py:112
Function
get_matterport_color
(color_path)
utils/sai3d_utils.py:656
Function
get_matterport_depth
(color_path)
utils/sai3d_utils.py:661
Function
get_matterport_intrinsic
(color_path)
utils/sai3d_utils.py:678
Function
get_matterport_mask
(base_dir, scene_id, color_name, mask_name, region_id='region4')
utils/sai3d_utils.py:688
Function
get_matterport_pose
(color_path)
utils/sai3d_utils.py:683
Function
get_matterport_semantic_mask
(color_path, semantic)
utils/sai3d_utils.py:703
Function
get_metric_value
Safely retrieves value of the metric logged in LightningModule.
ext/spt/utils/utils.py:181
Method
get_opacity
(self)
scene/gaussian_model.py:122
Function
get_points_from_openscene_pth
(base_dir, scene_id)
utils/sai3d_utils.py:712
Function
get_points_from_ply
(ply_path)
utils/sai3d_utils.py:21
Function
get_pylogger
Initializes multi-GPU-friendly python command line logger.
ext/spt/utils/pylogger.py:6
Method
get_relevancy
(self, embed: torch.Tensor, positive_id: int)
scripts/image_encoding.py:95
Method
get_rotation
(self)
scene/gaussian_model.py:104
← previous
next →
601–700 of 879, ranked by callers