MCPcopy Create free account

hub / github.com/LuLing06/I-Scene-project / functions

Functions386 in github.com/LuLing06/I-Scene-project

⨍ Functions 386◇ Types & classes 70
↓ 111 callersMethodreshape
(self, *shape)
iscene/trellis/modules/sparse/basic.py:236
↓ 51 callersMethodreplace
(self, feats: torch.Tensor, coords: Optional[torch.Tensor] = None)
iscene/trellis/modules/sparse/basic.py:243
↓ 38 callersMethodcpu
(self)
iscene/trellis/pipelines/base.py:67
↓ 31 callersMethodcuda
(self)
iscene/trellis/pipelines/base.py:64
↓ 25 callersMethodtype
(self, dtype)
iscene/trellis/modules/sparse/basic.py:203
↓ 24 callersMethodunbind
(self, dim: int)
iscene/trellis/modules/sparse/basic.py:240
↓ 22 callersMethodfloat
(self)
iscene/trellis/modules/sparse/basic.py:221
↓ 20 callersMethoddetach
(self)
iscene/trellis/modules/sparse/basic.py:225
↓ 20 callersMethodto
(self, dtype: torch.dtype)
iscene/trellis/modules/sparse/basic.py:177
↓ 15 callersMethodto
(self, device: torch.device)
iscene/trellis/pipelines/base.py:60
↓ 13 callersMethodwrite
(self, x)
iscene/trellis/representations/gaussian/general_utils.py:118
↓ 8 callersMethod__elemwise__
(self, other: Union[torch.Tensor, 'SparseTensor'], op: callable)
iscene/trellis/modules/sparse/basic.py:305
↓ 8 callersFunctionscaled_dot_product_attention
Apply scaled dot product attention. Args: qkv (torch.Tensor): A [N, L, 3, H, C] tensor containing Qs, Ks, and Vs.
iscene/trellis/modules/attention/full_attn.py:39
↓ 7 callersMethodget_spatial_cache
Get a spatial cache.
iscene/trellis/modules/sparse/basic.py:381
↓ 7 callersMethodregister_spatial_cache
Register a spatial cache. The spatial cache can be any thing you want to cache. The registery and retrieval of the cache is b
iscene/trellis/modules/sparse/basic.py:370
↓ 5 callersMethoddim
(self)
iscene/trellis/modules/sparse/basic.py:133
↓ 4 callersMethod__init__
( self, channels: int, out_channels: Optional[int] = None, norm_type: Literal[
iscene/trellis/models/sparse_structure_vae.py:23
↓ 4 callersFunction_coerce_file_path
(value: Any)
interactive_demo.py:291
↓ 4 callersMethod_linear
(module: nn.Linear, x: Union[SparseTensor, torch.Tensor])
iscene/trellis/modules/sparse/attention/modules.py:78
↓ 4 callersFunction_resolve_package_file
(model_id_or_path: str | Path, filename: str, revision: str | None = None)
iscene/inference/inferencer.py:25
↓ 4 callersMethodfrom_pretrained
Load a pretrained model.
iscene/trellis/pipelines/base.py:22
↓ 4 callersFunctionnorm_layer
Return a normalization layer.
iscene/trellis/models/sparse_structure_vae.py:10
↓ 3 callersMethod__cal_layout
(self, coords, batch_size)
iscene/trellis/modules/sparse/basic.py:123
↓ 3 callersMethod__cal_shape
(self, feats, coords)
iscene/trellis/modules/sparse/basic.py:117
↓ 3 callersMethod__init__
(self, channels: int, mlp_ratio: float = 4.0)
iscene/trellis/modules/transformer/blocks.py:50
↓ 3 callersFunction_get_prompt_image
(image_prompts: Any)
interactive_demo.py:132
↓ 3 callersMethod_linear_interp
Computes the location of zero-crossings on 'edges_x' using linear interpolation with 'edges_weight'.
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:223
↓ 3 callersFunction_progress_bar
(percent: int)
interactive_demo.py:392
↓ 3 callersMethoddevice
(self)
iscene/trellis/pipelines/base.py:51
↓ 3 callersMethodforward
(self, input: SparseTensor)
iscene/trellis/modules/sparse/norm.py:18
↓ 3 callersMethodinfer_and_save_scene
( self, scene_rgb_path: str | Path, instance_seg_path: str | Path, output_dir:
iscene/inference/inferencer.py:109
↓ 3 callersFunctionzero_module
Zero out the parameters of a module and return it.
iscene/trellis/modules/utils.py:37
↓ 2 callersMethod__init__
(self, dim: int, heads: int)
iscene/trellis/modules/attention/modules.py:9
↓ 2 callersMethod__init__
(self, factor: Union[int, Tuple[int, int, int], List[int]])
iscene/trellis/modules/sparse/spatial.py:64
↓ 2 callersMethod__init__
(self, channels: int, mlp_ratio: float = 4.0)
iscene/trellis/modules/sparse/transformer/blocks.py:12
↓ 2 callersMethod_fused_pre
(self, x: Union[SparseTensor, torch.Tensor], num_fused: int)
iscene/trellis/modules/sparse/attention/modules.py:91
↓ 2 callersFunction_get_inferencer
()
interactive_demo.py:165
↓ 2 callersFunction_make_session_dir
(request: gr.Request | None, root: Path = UPLOAD_ROOT)
interactive_demo.py:119
↓ 2 callersMethod_reshape_chs
(x: Union[SparseTensor, torch.Tensor], shape: Tuple[int, ...])
iscene/trellis/modules/sparse/attention/modules.py:85
↓ 2 callersFunction_resolve_model_file
(model_id_or_path: str, filename: str, revision: str | None = None)
iscene/trellis/pipelines/trellis_image_to_3d_scene_context.py:16
↓ 2 callersMethod_rotary_embedding
(self, x: torch.Tensor, phases: torch.Tensor)
iscene/trellis/modules/attention/modules.py:34
↓ 2 callersMethodcuda
(self)
iscene/trellis/modules/sparse/basic.py:212
↓ 2 callersMethodencode_image
Encode the image. Args: image (Union[torch.Tensor, list[Image.Image]]): The image to encode Returns:
iscene/trellis/pipelines/trellis_image_to_3d_scene_context.py:198
↓ 2 callersMethodforward
(self, x: torch.Tensor)
iscene/trellis/modules/norm.py:6
↓ 2 callersMethodforward
(self, input: SparseTensor)
iscene/trellis/modules/sparse/nonlinearity.py:14
↓ 2 callersMethodfull
(aabb, dim, value, dtype=torch.float32, device=None)
iscene/trellis/modules/sparse/basic.py:277
↓ 2 callersFunctionget_deformed_verts
(v_pos : torch.Tensor, deform : torch.Tensor, res)
iscene/trellis/representations/mesh/utils_cube.py:59
↓ 2 callersFunctionget_dense_attrs
(coords : torch.Tensor, feats : torch.Tensor, res : int, sdf_init=True)
iscene/trellis/representations/mesh/utils_cube.py:50
↓ 2 callersFunctionhammersley_sequence
(dim, n, num_samples)
iscene/trellis/utils/random_utils.py:19
↓ 2 callersFunctionpatchify
Patchify a tensor. Args: x (torch.Tensor): (N, C, *spatial) tensor patch_size (int): Patch size
iscene/trellis/modules/spatial.py:16
↓ 2 callersMethodrender
Render the mesh. Args: mesh : meshmodel extrinsics (torch.Tensor): (4, 4) camera extrinsics intr
iscene/trellis/renderers/mesh_renderer.py:55
↓ 2 callersMethodsample
Sample from a model.
iscene/trellis/pipelines/samplers/base.py:11
↓ 2 callersMethodsample
Generate samples from the model using Euler method. Args: model: The model to sample from. noise: The initia
iscene/trellis/pipelines/samplers/flow_euler.py:80
↓ 2 callersFunctionsegmentation_to_id_map
Decode an instance segmentation image into one integer label per pixel.
iscene/inference/segmentation_utils.py:20
↓ 2 callersMethodself_attn_join_external
Self-attention where queries come from x, and keys/values are augmented with one or more external token sequences. All projections (Q
iscene/trellis/modules/attention/modules.py:267
↓ 2 callersMethodsetup_pipeline
(self)
iscene/inference/inferencer.py:83
↓ 2 callersFunctionsparse_batch_broadcast
Broadcast a 1D tensor to a sparse tensor along the batch dimension then perform an operation. Args: input (torch.Tensor): 1D tensor
iscene/trellis/modules/sparse/basic.py:392
↓ 2 callersFunctionsparse_scaled_dot_product_attention
Apply scaled dot product attention to a sparse tensor. Args: qkv (SparseTensor): A [N, *, 3, H, C] sparse tensor containing Qs, Ks,
iscene/trellis/modules/sparse/attention/full_attn.py:20
↓ 2 callersFunctionsphere_hammersley_sequence
(n, num_samples, offset=(0, 0), remap=False)
iscene/trellis/utils/random_utils.py:22
↓ 2 callersFunctionunpatchify
Unpatchify a tensor. Args: x (torch.Tensor): (N, C, *spatial) tensor patch_size (int): Patch size
iscene/trellis/modules/spatial.py:34
↓ 1 callersFunction__from_env
()
iscene/trellis/modules/attention/__init__.py:6
↓ 1 callersFunction__from_env
()
iscene/trellis/modules/sparse/__init__.py:7
↓ 1 callersFunction__from_env
()
iscene/trellis/modules/sparse/conv/__init__.py:6
↓ 1 callersFunction__getattr__
(name)
iscene/trellis/models/__init__.py:14
↓ 1 callersMethod__init__
(self, num_groups, num_channels, eps=1e-5, affine=True)
iscene/trellis/modules/sparse/norm.py:15
↓ 1 callersMethod__init__
(self, dim: int, heads: int)
iscene/trellis/modules/sparse/attention/modules.py:13
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, dilation=1, bias=True, indice_key=None)
iscene/trellis/modules/sparse/conv/conv_torchsparse.py:7
↓ 1 callersMethod__init__
(self, in_channels, out_channels, kernel_size, stride=1, dilation=1, padding=None, bias=True, indice_key=None)
iscene/trellis/modules/sparse/conv/conv_spconv.py:21
↓ 1 callersMethod__init__
( self, channels: int, num_heads: int, mlp_ratio: float = 4.0, attn_mo
iscene/trellis/modules/sparse/transformer/modulated.py:14
↓ 1 callersMethod__init__
( self, channels: int, num_heads: int, mlp_ratio: float = 4.0, attn_mo
iscene/trellis/modules/transformer/modulated.py:13
↓ 1 callersMethod__init__
(self, hidden_size, frequency_embedding_size=256)
iscene/trellis/models/sparse_structure_flow.py:16
↓ 1 callersMethod__init__
( self, resolution: int, in_channels: int, model_channels: int, cond_c
iscene/trellis/models/structured_latent_flow.py:69
↓ 1 callersMethod__init__
( self, resolution: int, model_channels: int, latent_channels: int, nu
iscene/trellis/models/structured_latent_vae/decoder_mesh.py:72
↓ 1 callersMethod__init__
(self, vertices, faces, vertex_attrs=None, res=64 )
iscene/trellis/representations/mesh/cube2mesh.py:8
↓ 1 callersMethod__merge_sparse_cache
(self, other: 'SparseTensor')
iscene/trellis/modules/sparse/basic.py:290
↓ 1 callersFunction_boxes_from_prompts
(image_prompts: Any)
interactive_demo.py:183
↓ 1 callersMethod_build_perturbation
(self)
iscene/trellis/models/structured_latent_vae/decoder_gs.py:59
↓ 1 callersMethod_calc_layout
(self)
iscene/trellis/models/structured_latent_vae/decoder_gs.py:66
↓ 1 callersMethod_calc_layout
(self)
iscene/trellis/representations/mesh/cube2mesh.py:73
↓ 1 callersMethod_compute_reg_loss
Regularizer L_dev as in Equation 8
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:110
↓ 1 callersMethod_compute_vd
Computes the location of dual vertices as described in Section 4.2
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:250
↓ 1 callersFunction_discover_examples
()
interactive_demo.py:79
↓ 1 callersFunction_fill_holes
Rasterize a mesh from multiple views and remove invisible faces. Also includes postprocessing to: 1. Remove connected components that
iscene/trellis/utils/postprocessing_utils.py:22
↓ 1 callersMethod_forward
(self, x: SparseTensor)
iscene/trellis/modules/sparse/transformer/blocks.py:65
↓ 1 callersMethod_forward
(self, x: SparseTensor, mod: torch.Tensor, context: torch.Tensor)
iscene/trellis/modules/sparse/transformer/blocks.py:135
↓ 1 callersMethod_forward
(self, x: SparseTensor, mod: torch.Tensor)
iscene/trellis/modules/sparse/transformer/modulated.py:57
↓ 1 callersMethod_forward
(self, x: SparseTensor, mod: torch.Tensor, context: torch.Tensor)
iscene/trellis/modules/sparse/transformer/modulated.py:142
↓ 1 callersMethod_forward
(self, x: torch.Tensor)
iscene/trellis/modules/transformer/blocks.py:99
↓ 1 callersMethod_forward
(self, x: torch.Tensor, context: torch.Tensor)
iscene/trellis/modules/transformer/blocks.py:165
↓ 1 callersMethod_forward
(self, x: torch.Tensor, mod: torch.Tensor)
iscene/trellis/modules/transformer/modulated.py:52
↓ 1 callersMethod_forward
(self, x: torch.Tensor, mod: torch.Tensor, context: torch.Tensor, external_tokens: Optional[Union[torch.Tensor
iscene/trellis/modules/transformer/modulated.py:134
↓ 1 callersMethod_forward_flash
(self, q: torch.Tensor, k: torch.Tensor, v: torch.Tensor)
iscene/trellis/modules/attention_resample.py:63
↓ 1 callersMethod_forward_torch
(self, q: torch.Tensor, k: torch.Tensor, v: torch.Tensor)
iscene/trellis/modules/attention_resample.py:52
↓ 1 callersMethod_get_case_id
Obtains the ID of topology cases based on cell corner occupancy. This function resolves the ambiguity in the Dual Marching Cubes (DMC
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:144
↓ 1 callersMethod_get_model_prediction
(self, model, x_t, t, cond=None, **kwargs)
iscene/trellis/pipelines/samplers/flow_euler.py:44
↓ 1 callersMethod_get_phases
(self, indices: torch.Tensor)
iscene/trellis/modules/attention/modules.py:28
↓ 1 callersFunction_get_sam_model
(model_choice: str)
interactive_demo.py:172
↓ 1 callersMethod_identify_surf_cubes
Identifies grid cubes that intersect with the underlying surface by checking if the signs at all corners are not identical.
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:212
↓ 1 callersMethod_identify_surf_edges
Identifies grid edges that intersect with the underlying surface by checking for opposite signs. As each edge can be shared by multip
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:187
↓ 1 callersMethod_inference_model
(self, model, x_t, t, cond=None, **kwargs)
iscene/trellis/pipelines/samplers/flow_euler.py:38
next →1–100 of 386, ranked by callers