MCPcopy Create free account

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

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

↓ 1 callersFunctionvalidate_args
(args: argparse.Namespace)
run_inference.py:67
↓ 1 callersFunctionyaw_pitch_r_fov_to_extrinsics_intrinsics
(yaws, pitchs, rs, fovs)
iscene/trellis/utils/render_utils.py:12
FunctionPILtoTorch
(pil_image, resolution)
iscene/trellis/representations/gaussian/general_utils.py:21
FunctionRGB2SH
(rgb)
iscene/trellis/renderers/sh_utils.py:114
FunctionSH2RGB
(sh)
iscene/trellis/renderers/sh_utils.py:117
Method__add__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:320
Method__call__
Generates a mesh based on the specified sparse voxel structures. Args: cube_attrs [Nx21] : Sparse Tensor attrs about cube
iscene/trellis/representations/mesh/cube2mesh.py:96
Method__call__
(self, voxelgrid_vertices, scalar_field, cube_idx, resolution, qef_reg_scale=1e-3, weight_sca
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:56
Function__getattr__
(name)
iscene/trellis/modules/sparse/__init__.py:79
Method__getitem__
(self, idx)
iscene/trellis/modules/sparse/basic.py:344
Method__init__
( self, model_id_or_path: str | Path, *, base_model_id: str | Path | None = No
iscene/inference/inferencer.py:37
Method__init__
( self, d_model: int = 1024, n_target: int = 4096, *, n_heads: int = 1
iscene/trellis/modules/attention_resample.py:19
Method__init__
(self, hidden_size: int, in_channels: int = 3)
iscene/trellis/modules/attention/modules.py:19
Method__init__
( self, channels: int, num_heads: int, ctx_channels: Optional[int]=None,
iscene/trellis/modules/attention/modules.py:64
Method__init__
(self, in_features, out_features, bias=True)
iscene/trellis/modules/sparse/linear.py:11
Method__init__
(self, factor: Union[int, Tuple[int, ...], List[int]])
iscene/trellis/modules/sparse/spatial.py:18
Method__init__
(self)
iscene/trellis/modules/sparse/spatial.py:89
Method__init__
(self, normalized_shape, eps=1e-5, elementwise_affine=True)
iscene/trellis/modules/sparse/norm.py:32
Method__init__
(self, feats: torch.Tensor, coords: torch.Tensor, shape: Optional[torch.Size] = None, layout: Optional[List[sl
iscene/trellis/modules/sparse/basic.py:33
Method__init__
(self, activation: nn.Module)
iscene/trellis/modules/sparse/nonlinearity.py:29
Method__init__
( self, channels: int, num_heads: int, ctx_channels: Optional[int] = None,
iscene/trellis/modules/sparse/attention/modules.py:29
Method__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:23
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, dilation=1, bias=True, indice_key=None)
iscene/trellis/modules/sparse/conv/conv_spconv.py:67
Method__init__
( self, channels: int, num_heads: int, mlp_ratio: float = 4.0, attn_mo
iscene/trellis/modules/sparse/transformer/blocks.py:28
Method__init__
( self, channels: int, ctx_channels: int, num_heads: int, mlp_ratio: f
iscene/trellis/modules/sparse/transformer/blocks.py:85
Method__init__
( self, channels: int, ctx_channels: int, num_heads: int, mlp_ratio: f
iscene/trellis/modules/sparse/transformer/modulated.py:85
Method__init__
(self, channels: int, in_channels: int = 3)
iscene/trellis/modules/transformer/blocks.py:12
Method__init__
( self, channels: int, num_heads: int, mlp_ratio: float = 4.0, attn_mo
iscene/trellis/modules/transformer/blocks.py:66
Method__init__
( self, channels: int, ctx_channels: int, num_heads: int, mlp_ratio: f
iscene/trellis/modules/transformer/blocks.py:119
Method__init__
( self, channels: int, ctx_channels: int, num_heads: int, mlp_ratio: f
iscene/trellis/modules/transformer/modulated.py:80
Method__init__
( self, models: dict[str, nn.Module] = None, )
iscene/trellis/pipelines/base.py:11
Method__init__
( self, models: dict[str, nn.Module] = None, sparse_structure_sampler: samplers.Sample
iscene/trellis/pipelines/trellis_image_to_3d_scene_context.py:37
Method__init__
( self, sigma_min: float, )
iscene/trellis/pipelines/samplers/flow_euler.py:18
Method__init__
(self, rendering_options={}, device='cuda')
iscene/trellis/renderers/mesh_renderer.py:44
Method__init__
(self, rendering_options={})
iscene/trellis/renderers/gaussian_render.py:151
Method__init__
(self, image_cond_model: str = 'dinov2_vitl14_reg', cond_in_channels: int = 10, use_fp16: bool = True)
iscene/trellis/models/image_conditioner.py:11
Method__init__
( self, resolution: int, in_channels: int, model_channels: int, cond_c
iscene/trellis/models/sparse_structure_flow.py:57
Method__init__
( self, resolution: int, in_channels: int, model_channels: int, cond_c
iscene/trellis/models/sparse_structure_sc_flow.py:10
Method__init__
( self, in_channels: int, out_channels: int, mode: Literal["conv", "avgpool"]
iscene/trellis/models/sparse_structure_vae.py:51
Method__init__
( self, in_channels: int, out_channels: int, mode: Literal["conv", "nearest"]
iscene/trellis/models/sparse_structure_vae.py:76
Method__init__
( self, in_channels: int, latent_channels: int, num_res_blocks: int, c
iscene/trellis/models/sparse_structure_vae.py:114
Method__init__
( self, out_channels: int, latent_channels: int, num_res_blocks: int,
iscene/trellis/models/sparse_structure_vae.py:223
Method__init__
( self, channels: int, emb_channels: int, out_channels: Optional[int] = None,
iscene/trellis/models/structured_latent_flow.py:15
Method__init__
( self, resolution: int, model_channels: int, latent_channels: int, nu
iscene/trellis/models/structured_latent_vae/decoder_gs.py:12
Method__init__
( self, in_channels: int, model_channels: int, num_blocks: int, num_he
iscene/trellis/models/structured_latent_vae/base.py:32
Method__init__
( self, channels: int, resolution: int, out_channels: Optional[int] = None,
iscene/trellis/models/structured_latent_vae/decoder_mesh.py:22
Method__init__
a model to generate a mesh from sparse features structures using flexicube
iscene/trellis/representations/mesh/cube2mesh.py:56
Method__init__
(self, device="cuda")
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:25
Method__init__
( self, aabb : list, sh_degree : int = 0, mininum_kernel_size
iscene/trellis/representations/gaussian/gaussian_model.py:9
Method__init__
(self, silent)
iscene/trellis/representations/gaussian/general_utils.py:115
Method__mul__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:332
Method__neg__
(self)
iscene/trellis/modules/sparse/basic.py:302
Method__radd__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:323
Method__rmul__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:335
Method__rsub__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:329
Method__rtruediv__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:341
Method__sub__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:326
Method__truediv__
(self, other: Union[torch.Tensor, 'SparseTensor', float])
iscene/trellis/modules/sparse/basic.py:338
Method_basic_init
(module)
iscene/trellis/models/sparse_structure_flow.py:153
Method_basic_init
(module)
iscene/trellis/models/structured_latent_flow.py:215
Method_basic_init
(module)
iscene/trellis/models/structured_latent_vae/base.py:103
Method_eps_to_xstart
(self, x_t, t, eps)
iscene/trellis/pipelines/samplers/flow_euler.py:24
Method_inference_model
(self, model, x_t, t, cond, neg_cond, cfg_strength, **kwargs)
iscene/trellis/pipelines/samplers/classifier_free_guidance_mixin.py:9
Method_inference_model
(self, model, x_t, t, cond, neg_cond, cfg_strength, cfg_interval, **kwargs)
iscene/trellis/pipelines/samplers/guidance_interval_mixin.py:9
Method_solve_vd_QEF
(self, p_bxnx3, norm_bxnx3, c_bx3, qef_reg_scale)
iscene/trellis/representations/mesh/flexicubes/flexicubes.py:236
Method_xstart_to_eps
(self, x_t, t, x_0)
iscene/trellis/pipelines/samplers/flow_euler.py:28
Methodbuild_covariance_from_scaling_rotation
(scaling, scaling_modifier, rotation)
iscene/trellis/representations/gaussian/gaussian_model.py:46
Functionclear_generation_outputs
()
interactive_demo.py:437
Functionclear_glb_outputs
()
interactive_demo.py:433
Methodcompute_vertex_normals
(self, verts, faces)
iscene/trellis/representations/mesh/cube2mesh.py:37
Functionconvert_module_to_f16
Convert primitive modules to float16.
iscene/trellis/modules/utils.py:19
Functionconvert_module_to_f32
Convert primitive modules to float32, undoing convert_module_to_f16().
iscene/trellis/modules/utils.py:28
Methodconvert_to_fp32
(self)
iscene/trellis/models/image_conditioner.py:27
Methodconvert_to_fp32
Convert the torso of the model to float32.
iscene/trellis/models/sparse_structure_flow.py:145
Methodconvert_to_fp32
Convert the torso of the model to float32.
iscene/trellis/models/sparse_structure_vae.py:177
Methodconvert_to_fp32
Convert the torso of the model to float32.
iscene/trellis/models/sparse_structure_vae.py:286
Methodconvert_to_fp32
Convert the torso of the model to float32.
iscene/trellis/models/structured_latent_flow.py:205
Methodconvert_to_fp32
Convert the torso of the model to float32.
iscene/trellis/models/structured_latent_vae/base.py:95
Methodconvert_to_fp32
Convert the torso of the model to float32.
iscene/trellis/models/structured_latent_vae/decoder_mesh.py:138
Methodcoords
(self)
iscene/trellis/modules/sparse/basic.py:155
Methodcpu
(self)
iscene/trellis/modules/sparse/basic.py:207
Methodcrop_bbox
(alpha_np)
iscene/trellis/pipelines/trellis_image_to_3d_scene_context.py:125
Methoddense
(self)
iscene/trellis/modules/sparse/basic.py:230
Methoddevice
(self)
iscene/trellis/modules/sparse/basic.py:173
Methoddevice
Return the device of the model.
iscene/trellis/models/sparse_structure_flow.py:133
Methoddevice
Return the device of the model.
iscene/trellis/models/sparse_structure_vae.py:162
Methoddevice
Return the device of the model.
iscene/trellis/models/sparse_structure_vae.py:271
Methoddevice
Return the device of the model.
iscene/trellis/models/structured_latent_flow.py:191
Methoddevice
Return the device of the model.
iscene/trellis/models/structured_latent_vae/base.py:83
Methoddtype
(self)
iscene/trellis/modules/sparse/basic.py:169
Functionexp_anealing
(optimizer, step, total_steps, start_lr, end_lr)
iscene/trellis/utils/postprocessing_utils.py:362
Methodfeats
(self)
iscene/trellis/modules/sparse/basic.py:141
Methodflush
(self)
iscene/trellis/representations/gaussian/general_utils.py:125
Methodforward
(self, x: torch.Tensor)
iscene/trellis/modules/norm.py:14
Methodforward
(self, x: torch.Tensor)
iscene/trellis/modules/norm.py:19
Methodforward
Return a tensor with shape (B, n_target, d_model).
iscene/trellis/modules/attention_resample.py:40
Methodforward
(self, x: torch.Tensor)
iscene/trellis/modules/attention/modules.py:14
Methodforward
Args: q (sp.SparseTensor): [..., N, D] tensor of queries k (sp.SparseTensor): [..., N, D] tensor of keys
iscene/trellis/modules/attention/modules.py:40
Methodforward
(self, x: torch.Tensor, context: Optional[torch.Tensor] = None, indices: Optional[torch.Tensor] = None)
iscene/trellis/modules/attention/modules.py:129
Methodforward
(self, input: SparseTensor)
iscene/trellis/modules/sparse/linear.py:14
← previousnext →201–300 of 386, ranked by callers