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
get_scaling
(self)
scene/gaussian_model.py:100
Function
get_scannet_color_and_depth_intrinsic
(color_path)
utils/sai3d_utils.py:766
Function
get_scannet_depth
(color_path)
utils/sai3d_utils.py:757
Function
get_scannet_depth_mask
(color_path, base_dir, scene_id, mask_group_name)
utils/sai3d_utils.py:741
Function
get_scannet_mask
(color_path, base_dir, scene_id, mask_group_name)
utils/sai3d_utils.py:776
Function
get_scannet_pose
(color_path)
utils/sai3d_utils.py:752
Function
get_scannet_semantic_mask
(color_path, base_dir, scene_id, is_scannet200)
utils/sai3d_utils.py:786
Function
get_scannetpp_depth
(color_path, color_shape)
utils/sai3d_utils.py:825
Function
get_scannetpp_mask
(color_path, base_dir, scene_id, mask_group_name, color_shape)
utils/sai3d_utils.py:837
Function
get_scannetpp_poses_and_intrinsics
(base_dir, scene_id, freq)
utils/sai3d_utils.py:807
Method
get_seen_image
print images id which can see the points :param points_seen: (N, M), seen flag of all points in all views
merge_proj.py:503
Function
get_seg_label
(seg_ids, labels, points, voxel_size, max_label)
utils/sai3d_utils.py:918
Method
get_semantics
(self)
scene/gaussian_model.py:118
Function
get_similar_confidence_matrix_handle
(seg_neighbors, seg_ids, seg
utils/sai3d_utils.py:171
Function
get_splits
(base_dir, scannetpp)
utils/sai3d_utils.py:29
Method
get_superpoint_mask_ratio
(point_to_super, point_to_mask, num_masks=None, softmax=False)
merge_proj.py:100
Method
get_xyz
(self)
scene/gaussian_model.py:108
Function
grid_search_panoptic_partition
Runs a grid search on the partition parameters to find the best setup on a given sample `dataset[i_cloud]`. :param model: PanopticSegmentatio
ext/spt/utils/instance.py:1242
Method
gui_cb
(self,element)
scripts/image_encoding.py:85
Method
has_edge_attr
Whether the edges have features in `edge_attr`.
ext/spt/data/data.py:131
Method
has_edges
Whether the points have edges.
ext/spt/data/data.py:126
Method
has_instance_pred
Check whether `self` contains predicted data for panoptic segmentation `obj_index_pred`.
ext/spt/utils/output_panoptic.py:120
Method
has_multi_instance_pred
Check whether `self` contains predicted data for panoptic segmentation `obj_index_pred` as a list of results for performance compariso
ext/spt/utils/output_panoptic.py:127
Method
has_neighbors
Whether the points have neighbors.
ext/spt/data/data.py:121
Method
has_target
Check whether `self` contains target data for panoptic segmentation.
ext/spt/utils/output_panoptic.py:107
Method
has_target
Check whether `self` contains target data for semantic segmentation.
ext/spt/utils/output_semantic.py:47
Function
helper
(step)
utils/general_utils.py:47
Function
helper
(step)
gui/gs_renderer.py:30
Function
hex_to_tensor
(h)
ext/spt/utils/color.py:72
Function
histogram_to_atomic
Convert ground truth and predictions at a segment level (i.e. ground truth is 2D tensor carrying histogram of labels in each segment), to poin
ext/spt/utils/histogram.py:8
Function
host_data_root
Read the host machine's name and return the known $DATA_ROOT directory
ext/spt/utils/io.py:43
Method
indices
Returns the dense indices corresponding to the pointers.
ext/spt/data/csr.py:168
Function
init
(wish_host, wish_port)
gaussian_renderer/network_gui.py:27
Function
init_weights
Manual weight initialization. Allows setting specific init modes for certain modules. In particular, the linear and RPE layers are initialized
ext/spt/utils/nn.py:9
Method
initialize_options
(self)
scripts/setup_dependencies.py:50
Method
initialize_options
(self)
ext/spt/setup_dependencies.py:51
Method
instance_segmentation_oracle
Compute the oracle performance for instance segmentation. This is a proxy for the highest achievable performance with the cluster part
ext/spt/data/instance.py:736
Function
instantiate_callbacks
Instantiates callbacks from config.
ext/spt/utils/utils.py:99
Function
instantiate_loggers
Instantiates loggers from config.
ext/spt/utils/utils.py:118
Function
intrinsic_to_fov
根据相机内参计算视场角。 参数: f_x (float): x轴的焦距(像素单位) f_y (float): y轴的焦距(像素单位) width (int): 图像宽度(像素单位) height (int): 图像高度(像素单位) 返回:
gui/cam_utils.py:69
Method
intrinsics
(self)
gui/cam_utils.py:215
Function
inverse_sigmoid
(x)
utils/general_utils.py:18
Function
inverse_sigmoid
(x)
gui/gs_renderer.py:23
Method
is_cuda
Return True is one of the Data contains a CUDA Tensor.
ext/spt/data/nag.py:145
Method
is_index_value_key
Key name for is_index_value. This will be used as labels for serialization.
ext/spt/data/csr.py:405
Method
is_sub
Whether the points belong to a coarser super-graph.
ext/spt/data/data.py:116
Method
is_super
Whether the points are superpoints for a denser sub-graph.
ext/spt/data/data.py:111
Function
l1_loss
(network_output, gt)
utils/loss_utils.py:17
Function
l2_loss
(network_output, gt)
utils/loss_utils.py:20
Method
level_ratios
Ratios of number of nodes between consecutive partition levels. This can be useful for investigating how much each partition level 'co
ext/spt/data/nag.py:103
Function
linetimer
Decorating a function will log how long it took to execute each function call Usage: @linetimer() def foo(): pass "Cod
utils/linetimer.py:107
Function
listify_with_reference
listify `arg_ref` and the `args`, while ensuring that the length of `args` match the length of `arg_ref`. This is typically needed for parsing
ext/spt/utils/list.py:18
Method
load
Load ClusterBatch from an HDF5 file. See `Cluster.save` for writing such file. Options allow reading only part of the clusters.
ext/spt/data/cluster.py:215
Method
load
Read an HDF5 file and return its content as a Batch object. See Data.load()
ext/spt/data/data.py:1201
Method
load
Load CSRBatch from an HDF5 file. See `CSRData.save` for writing such file. Options allow reading only part of the clusters. :
ext/spt/data/csr.py:791
Method
load_gt_masks
(self, cameras)
gui/main.py:435
Function
log_hyperparameters
Controls which config parts are saved by lightning loggers. Additionally saves: - Number of model parameters
ext/spt/utils/utils.py:138
Function
loss_with_target_histogram
(criterion, pred, y_hist)
ext/spt/utils/loss.py:25
Function
make_checkpoint_file_search_widget
Generate an ipywidget for locally browsing a checkpoint file
ext/spt/utils/widgets.py:88
Function
make_device_widget
Generate an ipywidget for selecting the device on which to work
ext/spt/utils/widgets.py:53
Function
make_experiment_widgets
Generate two co-dependent ipywidgets for selecting the task and experiment from a predefined set of experiment configs.
ext/spt/utils/widgets.py:13
Function
make_split_widget
Generate an ipywidget for selecting the data split on which to work
ext/spt/utils/widgets.py:72
Function
mse
(img1, img2)
utils/image_utils.py:21
Method
multi_stage
If the semantic segmentation `logits` are stored in an enumerable, then the model output is multi-stage.
ext/spt/utils/output_semantic.py:54
Function
multiprocess_get_similar_confidence_matrix
:param seg_seen: (S, M), ratio of seen part of primitives in every view :param seg_neighbors: (S, S), 1 if two segs are neighbors :param
utils/sai3d_utils.py:239
Method
mvp
(self)
gui/cam_utils.py:220
Method
name
(self)
scripts/image_encoding.py:78
Method
natural_sort_key
Analyzes the key in a natural way to sort the strings that contain numbers. It converts the string into a list of mixed types
gui/main.py:1224
Method
neighbor_index
(self)
ext/spt/data/data.py:62
Function
new_export_res_to_class_agnostc_eval
(base_dir, scene_id, pol_name, semantic_name=None)
utils/sai3d_utils.py:1002
Method
norm_index
Index to be used for LayerNorm. :param mode: str Normalization mode. 'graph' will normalize per graph (i.e. per cloud
ext/spt/data/data.py:81
Method
num_batch_items
(self)
ext/spt/data/csr.py:585
Method
num_classes
Number for semantic classes in the output predictions.
ext/spt/utils/output_semantic.py:61
Method
num_clusters
(self)
ext/spt/data/cluster.py:59
Method
num_clusters
(self)
ext/spt/data/instance.py:148
Method
num_edges
Number for edges in the instance graph.
ext/spt/utils/output_panoptic.py:136
Method
num_edges
Overwrite the torch_geometric initial definition, which somehow returns incorrect results, like: data.num_edges != data.edge_index
ext/spt/data/data.py:161
Method
num_groups
(self)
ext/spt/data/csr.py:150
Method
num_items
(self)
ext/spt/data/csr.py:158
Method
num_levels
Number of levels of hierarchy in the nested graph.
ext/spt/data/nag.py:93
Method
num_nodes
Number for nodes/superpoints in the output predictions. By default, for a hierarchical partition, this means counting the number of le
ext/spt/utils/output_semantic.py:68
Method
num_obj
(self)
ext/spt/data/instance.py:156
Method
num_overlaps
(self)
ext/spt/data/instance.py:152
Method
num_points
(self)
ext/spt/data/cluster.py:63
Method
num_points
(self)
ext/spt/data/data.py:169
Method
num_points
Number of points/nodes in the lower-level graph.
ext/spt/data/nag.py:98
Method
num_settings
This assumes all items in `self.obj_index_pred` follow the output format of `InstancePartitioner._grid_forward()`.
ext/spt/utils/output_panoptic.py:551
Method
num_sub
(self)
ext/spt/data/data.py:177
Method
num_super
(self)
ext/spt/data/data.py:173
Method
num_values
(self)
ext/spt/data/csr.py:154
Method
obj
(self)
ext/spt/data/instance.py:112
Method
obj
InstanceData object indicating the instance indices for each node/point/superpoint in the Data.
ext/spt/data/data.py:51
Method
obj_move
(x, y, z)
gui/main.py:1028
Method
oneupSHdegree
(self)
scene/gaussian_model.py:128
Function
oracle_superpoint_clustering
Compute an oracle for superpoint clustering for instance and panoptic segmentation. This is a proxy for the highest achievable graph clusterin
ext/spt/utils/instance.py:680
Function
orbit_camera
(elevation, azimuth, radius=1, is_degree=True, target=None, opengl=True)
gui/cam_utils.py:126
Method
panoptic_pred
Return the predicted InstanceData, and the predicted instance semantic label and score, for a given batch item and a given partition s
ext/spt/utils/output_panoptic.py:557
Method
panoptic_segmentation_oracle
Compute the oracle performance for panoptic segmentation. This is a proxy for the highest achievable performance with the cluster part
ext/spt/data/instance.py:773
Method
perspective
(self)
gui/cam_utils.py:195
Method
pointer_key
Key name for pointers. This will be used as labels for serialization.
ext/spt/data/csr.py:389
← previous
next →
701–800 of 879, ranked by callers