MCPcopy Create free account

hub / github.com/VAST-AI-Research/TriplaneGaussian / functions

Functions315 in github.com/VAST-AI-Research/TriplaneGaussian

⨍ Functions 315◇ Types & classes 109↳ Endpoints 2
↓ 15 callersMethod__init__
(self, config)
tgs/models/tokenizers/dinov2.py:437
↓ 10 callersMethod__init__
(self, dim_in: int, dim_out: int, approximate: str = "none")
tgs/models/transformers.py:463
↓ 10 callersMethod__init__
(self, in_channel, layer_dims, bn=None)
tgs/models/snowflake/utils.py:49
↓ 7 callersMethodget_save_path
(self, filename)
tgs/utils/saving.py:45
↓ 7 callersFunctioninit_linear
(l, stddev)
tgs/models/snowflake/attention.py:64
↓ 6 callersFunctionparse_structured
(fields: Any, cfg: Optional[Union[dict, DictConfig]] = None)
tgs/utils/config.py:72
↓ 5 callersMethod__init__
(self, *, device: torch.device, dtype: torch.dtype, width: int, init_scale: float)
tgs/models/snowflake/attention.py:70
↓ 4 callersMethodapply
(module, name, dim, eps)
tgs/models/snowflake/utils.py:603
↓ 4 callersMethodgelu
(self, gate: torch.Tensor)
tgs/models/transformers.py:468
↓ 4 callersFunctionget_device
()
tgs/utils/misc.py:25
↓ 4 callersFunctionload_module_weights
( path, module_name=None, ignore_modules=None, map_location=None )
tgs/utils/misc.py:29
↓ 4 callersMethodset_use_memory_efficient_attention_xformers
( self, valid: bool, attention_op: Optional[Callable] = None )
tgs/models/transformers.py:73
↓ 3 callersMethod__init__
( self, mlp, npoint=None, radius=None, nsample=None, bn=True, use_xyz=True )
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py:135
↓ 3 callersMethodconvert_data
(self, data)
tgs/utils/saving.py:28
↓ 3 callersMethodcoordinate2index
(self, x)
tgs/models/pointclouds/pointnet.py:83
↓ 3 callersMethodgenerate_plane_features
(self, index, c)
tgs/models/pointclouds/pointnet.py:55
↓ 3 callersMethodget_save_dir
(self)
tgs/utils/saving.py:23
↓ 3 callersMethodmake_linear
(self, dim_in, dim_out, is_first, is_last, bias=True)
tgs/models/networks.py:95
↓ 3 callersFunctionquery_knn
Find k-NN of new_xyz in xyz
tgs/models/snowflake/utils.py:290
↓ 3 callersMethodtranspose_for_scores
(self, x: torch.Tensor)
tgs/models/tokenizers/dinov2.py:244
↓ 2 callersMethod__init__
( self, cfg: Optional[Union[dict, DictConfig]] = None, *args, **kwargs )
tgs/utils/base.py:76
↓ 2 callersMethod__init__
( self, dim_in: int, dim_out: int, n_neurons: int, n_hidden_layers: in
tgs/models/networks.py:58
↓ 2 callersFunctionball_query
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/ball_query.cpp:8
↓ 2 callersFunctionbuild_shared_mlp
(mlp_spec: List[int], bn: bool = True)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_modules.py:9
↓ 2 callersFunctioncheckpoint
Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas
tgs/models/snowflake/attention.py:10
↓ 2 callersMethodcompute_weight
(self, module, n_power_iterations)
tgs/models/snowflake/utils.py:550
↓ 2 callersMethoddetokenize
(self, *args, **kwargs)
tgs/models/tokenizers/image.py:134
↓ 2 callersFunctionget_activation
(name)
tgs/utils/ops.py:53
↓ 2 callersMethodget_grayscale_image_
(self, img, data_range, cmap)
tgs/utils/saving.py:114
↓ 2 callersFunctionget_intrinsic_from_fov
(fov, H, W, bs=-1)
tgs/utils/ops.py:219
↓ 2 callersMethodget_rgb_image_
(self, img, data_format, data_range, rgba=False)
tgs/utils/saving.py:59
↓ 2 callersFunctionimage_preprocess
(input_image, save_path, lower_contrast=True, rescale=True)
image_preprocess/utils.py:39
↓ 2 callersFunctioninfer
(image_path: str, cam_dist: float, only_3dgs: bool = False)
gradio_app.py:80
↓ 2 callersFunctionload_config
( *yamls: str, cli_args: list = [], from_string=False, makedirs=True, **kwargs )
tgs/utils/config.py:39
↓ 2 callersMethodmake_activation
(self, activation)
tgs/models/networks.py:99
↓ 2 callersMethodnon_module
(self, name: str)
tgs/utils/base.py:128
↓ 2 callersFunctionopt_n_threads
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/_ext-src/include/cuda_utils.h:15
↓ 2 callersFunctionpred_bbox
(image)
image_preprocess/utils.py:68
↓ 2 callersFunctionresize_image
(input_raw, size)
image_preprocess/utils.py:79
↓ 2 callersFunctionsam_init
(sam_checkpoint, device_id=0)
image_preprocess/utils.py:8
↓ 2 callersFunctionsam_out_nosave
(predictor, input_image, *bbox_sliders)
image_preprocess/utils.py:18
↓ 2 callersFunctionsample_and_group_all
Args: xyz: Tensor, (B, 3, nsample) points: Tensor, (B, f, nsample) use_xyz: boolean Returns: new_xyz: Tensor
tgs/models/snowflake/utils.py:137
↓ 2 callersMethodsave_img_sequences
( self, seq_dir, matcher, save_format="mp4", fps=30, delete=Tr
tgs/utils/saving.py:290
↓ 2 callersFunctionscale_tensor
( dat: Num[Tensor, "... D"], inp_scale: ValidScale, tgt_scale: ValidScale )
tgs/utils/ops.py:20
↓ 2 callersMethodset_save_dir
(self, save_dir: str)
tgs/utils/saving.py:20
↓ 2 callersFunctiontodevice
(vars, device="cuda")
tgs/utils/misc.py:76
↓ 1 callersMethod__init__
(self, use_xyz=True)
tgs/models/snowflake/pointnet2_ops_lib/pointnet2_ops/pointnet2_utils.py:344
↓ 1 callersMethod_forward
(self, batch: Dict[str, Any])
infer.py:96
↓ 1 callersFunction_parse_scene_list
(scene_list_path: Union[str, List[str]])
tgs/data.py:31
↓ 1 callersFunction_parse_scene_list_single
(scene_list_path: str)
tgs/data.py:18
↓ 1 callersMethod_save_grayscale_image
( self, filename, img, data_range, cmap, )
tgs/utils/saving.py:158
↓ 1 callersMethod_save_rgb_image
( self, filename, img, data_format, data_range )
tgs/utils/saving.py:93
↓ 1 callersMethod_set_gradient_checkpointing
( self, module: Dinov2Encoder, value: bool = False )
tgs/models/tokenizers/dinov2.py:657
↓ 1 callersMethodcalculate_image_token
Args:
tgs/models/snowflake/model_spdpp.py:106
↓ 1 callersMethodcalculate_pcl_token
(self, pcl_token, up_factor)
tgs/models/snowflake/model_spdpp.py:102
↓ 1 callersFunctioncompute_distance_transform
(mask: torch.Tensor)
tgs/utils/ops.py:271
↓ 1 callersMethodconfigure
(self, *args, **kwargs)
tgs/utils/base.py:84
↓ 1 callersMethodconfigure
(self, *args, **kwargs)
tgs/utils/base.py:119
↓ 1 callersMethodconfigure
(self)
tgs/models/networks.py:18
↓ 1 callersMethodconstruct_list_of_attributes
(self)
tgs/models/renderer.py:93
↓ 1 callersMethoddo_update_step
( self, epoch: int, global_step: int, on_load_weights: bool = False )
tgs/utils/base.py:21
↓ 1 callersMethoddo_update_step_end
(self, epoch: int, global_step: int)
tgs/utils/base.py:37
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). Comment by Ross Wightman: This is the same as the D
tgs/models/tokenizers/dinov2.py:395
↓ 1 callersFunctionfind_coupling_layer
(module)
tgs/models/snowflake/utils.py:695
↓ 1 callersMethodforward_gs
(self, x, p)
tgs/models/renderer.py:219
↓ 1 callersMethodforward_single_batch
( self, gs_hidden_features: Float[Tensor, "Np Cp"], query_points: Float[Tensor, "Np 3"
tgs/models/renderer.py:363
↓ 1 callersMethodforward_single_view
(self, gs: GaussianModel, viewpoint_camera: Camera, background_color: Optional[Float[T
tgs/models/renderer.py:224
↓ 1 callersMethodfrom_c2w
(c2w, intrinsic, height, width)
tgs/models/renderer.py:81
↓ 1 callersMethodgelu
(self, gate: torch.Tensor)
tgs/models/transformers.py:497
↓ 1 callersFunctiongetProjectionMatrix
(znear, zfar, fovX, fovY)
tgs/models/renderer.py:33
↓ 1 callersMethodget_image_grid_
(self, imgs, align)
tgs/utils/saving.py:179
↓ 1 callersFunctionget_rank
()
tgs/utils/misc.py:14
↓ 1 callersFunctionget_ray_directions
Get ray directions for all pixels in camera coordinate. Reference: https://www.scratchapixel.com/lessons/3d-basic-rendering/ r
tgs/utils/ops.py:87
↓ 1 callersFunctionget_rays
( directions: Float[Tensor, "... 3"], c2w: Float[Tensor, "... 4 4"], keepdim=False, noise_scal
tgs/utils/ops.py:127
↓ 1 callersMethodget_update_method
(self, module)
tgs/models/snowflake/utils.py:587
↓ 1 callersFunctioninit_trial_dir
()
gradio_app.py:74
↓ 1 callersMethodinitialize_weights
(self)
tgs/models/networks.py:23
↓ 1 callersFunctioninplace_spectral_norm
r"""Applies spectral normalization to a parameter in the given module. .. math:: \mathbf{W} = \dfrac{\mathbf{W}}{\sigma(\mathbf{W})} \\
tgs/models/snowflake/utils.py:628
↓ 1 callersMethodinterpolate_pos_encoding
This method allows to interpolate the pre-trained position encodings, to be able to use the model on higher resolution images.
tgs/models/tokenizers/dinov2.py:94
↓ 1 callersMethodinterpolate_pos_encoding
This method allows to interpolate the pre-trained position encodings, to be able to use the model on higher resolution images.
tgs/models/tokenizers/dinov2.py:1113
↓ 1 callersFunctionintrinsic_to_fov
(intrinsic, w, h)
tgs/models/renderer.py:55
↓ 1 callersFunctionlaunch
(port)
gradio_app.py:129
↓ 1 callersMethodload_weights
(self, weights: str, ignore_modules: Optional[List[str]] = None)
infer.py:49
↓ 1 callersFunctionpoints_projection
(points: Float[Tensor, "B Np 3"], c2ws: Float[Tensor, "B 4 4"], intrin
tgs/utils/ops.py:232
↓ 1 callersFunctionpoints_projection
points: (B, Np, 3)
tgs/models/snowflake/model_spdpp.py:34
↓ 1 callersMethodpool_local
(self, xy, index, c)
tgs/models/pointclouds/pointnet.py:68
↓ 1 callersFunctionpreprocess
(input_raw, sam_predictor=None)
gradio_app.py:65
↓ 1 callersMethodprune_heads
(self, heads: Set[int])
tgs/models/tokenizers/dinov2.py:342
↓ 1 callersMethodquery_triplane
( self, positions: Float[Tensor, "*B N 3"], triplanes: Float[Tensor, "*B 3 Cp Hp Wp"],
tgs/models/renderer.py:329
↓ 1 callersMethodregister_ada_norm_modulation
(self, norm1_mod: nn.Module, norm2_mod: nn.Module)
tgs/models/tokenizers/dinov2.py:545
↓ 1 callersMethodregister_non_module
(self, name: str, module: nn.Module)
tgs/utils/base.py:122
↓ 1 callersMethodremove
(self, module)
tgs/models/snowflake/utils.py:580
↓ 1 callersFunctionrun
(image_path: str, cam_dist: float, save_path: str)
gradio_app.py:108
↓ 1 callersFunctionrun_video
(image_path: str, cam_dist: float, save_path: str)
gradio_app.py:115
↓ 1 callersFunctionsample_and_group
Args: xyz: Tensor, (B, 3, N) points: Tensor, (B, f, N) npoint: int nsample: int radius: float use
tgs/models/snowflake/utils.py:101
↓ 1 callersFunctionsample_and_group_knn
Args: xyz: Tensor, (B, 3, N) points: Tensor, (B, f, N) npoint: int nsample: int radius: float use
tgs/models/snowflake/utils.py:298
↓ 1 callersMethodsave_image_grid
( self, filename, imgs, align=DEFAULT_GRID_KWARGS["align"], texts: Opt
tgs/utils/saving.py:225
↓ 1 callersMethodsave_img_sequence
( self, filename, img_dir, matcher, save_format="mp4", fps=30,
tgs/utils/saving.py:261
↓ 1 callersMethodsave_ply
(self, path)
tgs/models/renderer.py:108
↓ 1 callersMethodset_gradient_checkpointing
(self, value: bool = False)
tgs/models/tokenizers/dinov2.py:826
next →1–100 of 315, ranked by callers