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
points
(self)
ext/spt/data/cluster.py:47
Method
pos
(self)
ext/spt/data/data.py:43
Method
pose
(self)
gui/cam_utils.py:170
Method
pred_res_mask
(self, prompt, width=512, height=512)
gui/main.py:1171
Function
predict_elevation
(pos_query)
ext/spt/utils/ground.py:119
Function
print_memory_size
(a)
ext/spt/utils/memory.py:8
Method
qvec2rotmat
(self)
scene/colmap_loader.py:69
Method
rand_poses
(batch=10, elevation_range=[0,0], azimuth_range=[0, 360], radius=1, is_degree=True, rotate_z=False)
gui/cam_utils.py:260
Function
readColmapSceneInfo
(path, images, eval, llffhold=8, load_img=False, load_sem=True)
scene/dataset_readers.py:149
Function
readNerfSyntheticInfo
(path, white_background, eval, extension=".png", load_img=True, load_sem=True)
scene/dataset_readers.py:242
Function
read_colmap_bin_array
Taken from https://github.com/colmap/colmap/blob/dev/scripts/python/read_dense.py :param path: path to the colmap binary file. :return:
scene/colmap_loader.py:273
Function
read_one_scan
(scannet_dir, scan_name, label_map_file)
ext/spt/utils/scannet.py:190
Function
read_one_test_scan
(scannet_dir, scan_name)
ext/spt/utils/scannet.py:209
Function
receive
()
gaussian_renderer/network_gui.py:71
Function
recenter_poses
Recenter poses around the origin.
utils/render_utils.py:43
Method
reindex_groups
Returns a copy of self with modified pointers to account for new groups. Affects the num_groups and the order of groups. Injects 0-len
ext/spt/data/csr.py:198
Method
remove_void
Return a new InstanceData with void clusters, objects and pairs removed. IMPORTANT: By convention, we assume `y ∈ [0, num_cla
ext/spt/data/instance.py:547
Method
render
( self, viewpoint_camera, scaling_modifier=1.0, bg_color=None, overrid
gui/gs_renderer.py:224
Function
render_net_image
(render_pkg, render_items, render_mode, camera)
utils/image_utils.py:46
Method
reset_opacity
(self)
scene/gaussian_model.py:274
Method
restore
(self, model_args, training_args)
scene/gaussian_model.py:80
Method
rgb
(self)
ext/spt/data/data.py:47
Function
rotmat2qvec
(R)
scene/colmap_loader.py:55
Method
run
(self)
scripts/setup_dependencies.py:54
Method
run
(self)
ext/spt/setup_dependencies.py:55
Function
run_command
Run a command-line process from Python and print its outputs in an online fashion. Credit: https://www.endpointdev.com/blog/2015/01/getting-r
ext/spt/utils/download.py:36
Method
sanitized_edge_affinities
Return the predicted and target edge affinities, along with masks indicating same-class and same-object edges. The output is sanitized
ext/spt/utils/output_panoptic.py:207
Method
save
(self, path)
scene/semantic_model.py:59
Method
save
Save Batch to HDF5 file. :param f: h5 file path of h5py.File or h5py.Group :param y_to_csr: bool Convert 'y' to CSR forma
ext/spt/data/data.py:1150
Method
save
Save CSRBatch to HDF5 file. :param f: h5 file path of h5py.File or h5py.Group :param fp_dtype: torch dtype Data type to w
ext/spt/data/csr.py:766
Function
save_label_ply
(point_xyz: np.ndarray, label: torch.Tensor, save_path: str)
utils/sai3d_utils.py:1055
Function
save_seg_to_point_obj_label
(args)
utils/sai3d_utils.py:145
Function
scannetpp_get_semantic_mask
(color_path, base_dir, scene_id, semantic_text, color_shape)
utils/sai3d_utils.py:849
Method
select
Returns a new Cluster with updated clusters and points, which indexes `self` using entries in `idx`. Supports torch and numpy fancy in
ext/spt/data/cluster.py:78
Method
select
Returns a new Data with updated clusters, which indexes `self` using entries in `idx`. Supports torch and numpy fancy indexing. `idx`
ext/spt/data/data.py:263
Method
select
Indexing CSRBatch format. Supports Numpy and torch indexing mechanisms. Since indexing breaks batching, this will return a CSRData
ext/spt/data/csr.py:739
Method
semantic_pred
(self)
ext/spt/data/data.py:58
Method
semantic_segmentation_oracle
Compute the oracle performance for semantic segmentation, when all clusters predict the dominant label among their points. This corres
ext/spt/data/instance.py:645
Method
semantic_target
Semantic target on the level-1 superpoint. Final semantic segmentation target are the label histogram of the first-level partition lo
ext/spt/utils/output_semantic.py:86
Function
send
(message_bytes, verify, metrics)
gaussian_renderer/network_gui.py:63
Method
set_point_prompt
(sender, app_data)
gui/main.py:928
Method
settings
This assumes all items in `self.obj_index_pred` follow the output format of `InstancePartitioner._grid_forward()`.
ext/spt/utils/output_panoptic.py:544
Method
show
See `src.visualization.show`.
ext/spt/data/data.py:850
Method
sizes
Returns the size of each group (i.e. the pointer jumps).
ext/spt/data/csr.py:162
Function
smooth_loss
(disp, img)
utils/loss_utils.py:28
Function
starmap_with_kwargs
By default, starmap only accepts args and not kwargs. This is a helper to get around this problem. :param fn: callable The function t
ext/spt/utils/multiprocessing.py:8
Function
strip_symmetric
(sym)
utils/general_utils.py:75
Function
strip_symmetric
(sym)
gui/gs_renderer.py:62
Method
sub
Cluster object indicating subpoint indices for each point.
ext/spt/data/data.py:67
Method
super_index
Index of the superpoint each point belongs to.
ext/spt/data/data.py:72
Method
superpoint_panoptic_pred
Panoptic predictions on the level-1 nodes. Returns the predicted semantic label and instance index for each superpoint, along with the
ext/spt/utils/output_panoptic.py:321
Function
task_wrapper
Optional decorator that wraps the task function in extra utilities. Makes multirun more resistant to failure. Utilities: - Calling the `
ext/spt/utils/utils.py:24
Function
timer
(f, *args, text='', text_size=64, **kwargs)
ext/spt/utils/time.py:8
Method
to
Move the CSRBatch to the specified device.
ext/spt/data/csr.py:595
Method
to_list
(self)
ext/spt/data/csr.py:672
Method
to_nag_list
(self, strict: bool = False)
ext/spt/data/nag.py:542
Function
torch_calcu_all_similar
:param labels0: (s1, lr-1), label distribuion of one seg :param labels1: (s2, lr-1), label distribution of neighbors of the seg above :r
utils/sai3d_utils.py:487
Function
torch_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:367
Function
torch_get_similar_confidence_matrix2
(seg_neighbors, seg_ids,
utils/sai3d_utils.py:303
Function
torch_to_numpy
Decorator intended for numpy-based functions to be fed and return torch.Tensor arguments. :param func: :return:
ext/spt/utils/tensor.py:210
Function
torch_world2cam_pixel
project N (1e6) points to M (1e3) images :param points_world: (N, 3) :param color_intrinsic, depth_intrinsic: (M, 3, 3) the intrinsics of col
utils/sai3d_utils.py:553
Function
transform_Image_to_tensor
(image)
utils/image_utils.py:147
Function
try_connect
(render_items)
gaussian_renderer/network_gui.py:45
Method
uncontract
(y)
utils/mesh_utils.py:193
Function
update
(*args)
ext/spt/utils/widgets.py:41
Method
update_learning_rate
Learning rate scheduling per step
scene/gaussian_model.py:230
Method
v_edge_attr
Vertical edge features.
ext/spt/data/data.py:77
Method
v_edge_keys
All keys starting with `v_edge_`.
ext/spt/data/data.py:156
Method
value_keys
List of names for each value. These will be used as labels for serialization.
ext/spt/data/csr.py:396
Function
visualize_poses
(poses, size=0.1,save_dir="logs/", file_name="test.glb")
gui/cam_utils.py:8
Function
visualize_poses_stabledreamfusion
(poses, dirs, size=0.1)
gui/cam_utils.py:35
Method
void_edge_mask
Returns a mask on the edges indicating those connecting two void nodes.
ext/spt/utils/output_panoptic.py:158
Method
void_mask
Returns a mask on the level-1 nodes indicating which is void. By convention, nodes/superpoints are void if they contain more than 50%
ext/spt/utils/output_semantic.py:95
Method
voxel_semantic_pred
Semantic predictions on the level-0 voxels. Final semantic segmentation predictions are the argmax of the first-level partition logit
ext/spt/utils/output_semantic.py:114
Function
world2cam_pixel
project N points to M images :param points_world: (N, 3) :param color_intrinsic, depth_intrinsic: (M, 3, 3) the intrinsics of color and depth
utils/sai3d_utils.py:518
Function
wrap
(cfg: DictConfig)
ext/spt/utils/utils.py:37
Function
wrapper
(*args, **kwargs)
utils/linetimer.py:147
Function
wrapper_torch_to_numba
(*args, **kwargs)
ext/spt/utils/tensor.py:219
Method
y
(self)
ext/spt/data/instance.py:136
← previous
801–879 of 879, ranked by callers