MCPcopy Create free account

hub / github.com/Atrovast/THGS / functions

Functions879 in github.com/Atrovast/THGS

Methodpoints
(self)
ext/spt/data/cluster.py:47
Methodpos
(self)
ext/spt/data/data.py:43
Methodpose
(self)
gui/cam_utils.py:170
Methodpred_res_mask
(self, prompt, width=512, height=512)
gui/main.py:1171
Functionpredict_elevation
(pos_query)
ext/spt/utils/ground.py:119
Functionprint_memory_size
(a)
ext/spt/utils/memory.py:8
Methodqvec2rotmat
(self)
scene/colmap_loader.py:69
Methodrand_poses
(batch=10, elevation_range=[0,0], azimuth_range=[0, 360], radius=1, is_degree=True, rotate_z=False)
gui/cam_utils.py:260
FunctionreadColmapSceneInfo
(path, images, eval, llffhold=8, load_img=False, load_sem=True)
scene/dataset_readers.py:149
FunctionreadNerfSyntheticInfo
(path, white_background, eval, extension=".png", load_img=True, load_sem=True)
scene/dataset_readers.py:242
Functionread_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
Functionread_one_scan
(scannet_dir, scan_name, label_map_file)
ext/spt/utils/scannet.py:190
Functionread_one_test_scan
(scannet_dir, scan_name)
ext/spt/utils/scannet.py:209
Functionreceive
()
gaussian_renderer/network_gui.py:71
Functionrecenter_poses
Recenter poses around the origin.
utils/render_utils.py:43
Methodreindex_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
Methodremove_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
Methodrender
( self, viewpoint_camera, scaling_modifier=1.0, bg_color=None, overrid
gui/gs_renderer.py:224
Functionrender_net_image
(render_pkg, render_items, render_mode, camera)
utils/image_utils.py:46
Methodreset_opacity
(self)
scene/gaussian_model.py:274
Methodrestore
(self, model_args, training_args)
scene/gaussian_model.py:80
Methodrgb
(self)
ext/spt/data/data.py:47
Functionrotmat2qvec
(R)
scene/colmap_loader.py:55
Methodrun
(self)
scripts/setup_dependencies.py:54
Methodrun
(self)
ext/spt/setup_dependencies.py:55
Functionrun_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
Methodsanitized_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
Methodsave
(self, path)
scene/semantic_model.py:59
Methodsave
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
Methodsave
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
Functionsave_label_ply
(point_xyz: np.ndarray, label: torch.Tensor, save_path: str)
utils/sai3d_utils.py:1055
Functionsave_seg_to_point_obj_label
(args)
utils/sai3d_utils.py:145
Functionscannetpp_get_semantic_mask
(color_path, base_dir, scene_id, semantic_text, color_shape)
utils/sai3d_utils.py:849
Methodselect
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
Methodselect
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
Methodselect
Indexing CSRBatch format. Supports Numpy and torch indexing mechanisms. Since indexing breaks batching, this will return a CSRData
ext/spt/data/csr.py:739
Methodsemantic_pred
(self)
ext/spt/data/data.py:58
Methodsemantic_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
Methodsemantic_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
Functionsend
(message_bytes, verify, metrics)
gaussian_renderer/network_gui.py:63
Methodset_point_prompt
(sender, app_data)
gui/main.py:928
Methodsettings
This assumes all items in `self.obj_index_pred` follow the output format of `InstancePartitioner._grid_forward()`.
ext/spt/utils/output_panoptic.py:544
Methodshow
See `src.visualization.show`.
ext/spt/data/data.py:850
Methodsizes
Returns the size of each group (i.e. the pointer jumps).
ext/spt/data/csr.py:162
Functionsmooth_loss
(disp, img)
utils/loss_utils.py:28
Functionstarmap_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
Functionstrip_symmetric
(sym)
utils/general_utils.py:75
Functionstrip_symmetric
(sym)
gui/gs_renderer.py:62
Methodsub
Cluster object indicating subpoint indices for each point.
ext/spt/data/data.py:67
Methodsuper_index
Index of the superpoint each point belongs to.
ext/spt/data/data.py:72
Methodsuperpoint_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
Functiontask_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
Functiontimer
(f, *args, text='', text_size=64, **kwargs)
ext/spt/utils/time.py:8
Methodto
Move the CSRBatch to the specified device.
ext/spt/data/csr.py:595
Methodto_list
(self)
ext/spt/data/csr.py:672
Methodto_nag_list
(self, strict: bool = False)
ext/spt/data/nag.py:542
Functiontorch_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
Functiontorch_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
Functiontorch_get_similar_confidence_matrix2
(seg_neighbors, seg_ids,
utils/sai3d_utils.py:303
Functiontorch_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
Functiontorch_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
Functiontransform_Image_to_tensor
(image)
utils/image_utils.py:147
Functiontry_connect
(render_items)
gaussian_renderer/network_gui.py:45
Methoduncontract
(y)
utils/mesh_utils.py:193
Functionupdate
(*args)
ext/spt/utils/widgets.py:41
Methodupdate_learning_rate
Learning rate scheduling per step
scene/gaussian_model.py:230
Methodv_edge_attr
Vertical edge features.
ext/spt/data/data.py:77
Methodv_edge_keys
All keys starting with `v_edge_`.
ext/spt/data/data.py:156
Methodvalue_keys
List of names for each value. These will be used as labels for serialization.
ext/spt/data/csr.py:396
Functionvisualize_poses
(poses, size=0.1,save_dir="logs/", file_name="test.glb")
gui/cam_utils.py:8
Functionvisualize_poses_stabledreamfusion
(poses, dirs, size=0.1)
gui/cam_utils.py:35
Methodvoid_edge_mask
Returns a mask on the edges indicating those connecting two void nodes.
ext/spt/utils/output_panoptic.py:158
Methodvoid_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
Methodvoxel_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
Functionworld2cam_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
Functionwrap
(cfg: DictConfig)
ext/spt/utils/utils.py:37
Functionwrapper
(*args, **kwargs)
utils/linetimer.py:147
Functionwrapper_torch_to_numba
(*args, **kwargs)
ext/spt/utils/tensor.py:219
Methody
(self)
ext/spt/data/instance.py:136
← previous801–879 of 879, ranked by callers