MCPcopy Create free account

hub / github.com/TencentARC/Pixal3D / functions

Functions865 in github.com/TencentARC/Pixal3D

Functiondict_all
Recursively apply a function to all non-dictionary leaf values in a dictionary.
pixal3d/utils/general_utils.py:89
Functiondict_merge
Merge two dictionaries.
pixal3d/utils/general_utils.py:30
Functiondownload
(metadata, root, **kwargs)
data_toolkit/datasets/ABO.py:20
Functiondownload
(metadata, output_dir, **kwargs)
data_toolkit/datasets/ObjaverseXL.py:25
Functiondownload
(metadata, root, **kwargs)
data_toolkit/datasets/TexVerse.py:16
Methoddtype
(self)
pixal3d/modules/sparse/basic.py:79
Methoddtype
(self)
pixal3d/modules/sparse/basic.py:535
Functionextract_glb_api
(state_path: str, decimation_target: int, texture_size: int, session_id: str = "")
app.py:508
Methodfeats
(self)
pixal3d/modules/sparse/basic.py:499
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/sparse_voxel_pbr.py:145
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/structured_latent_shape.py:288
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/structured_latent_svpbr.py:313
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/structured_latent_svpbr.py:498
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/sparse_structure_latent.py:260
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/sparse_structure_latent.py:336
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/structured_latent.py:146
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/flexi_dual_grid.py:92
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/components.py:156
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/components.py:213
Methodfilter_metadata
(self, metadata, dataset_name=None)
pixal3d/datasets/components.py:298
Methodfinetune_from
Override to tolerate DINOv3 backbone keys missing from image_cond checkpoint. For image_cond, the checkpoint only stores proj_linear
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:1170
Methodfloat
(self)
pixal3d/modules/sparse/basic.py:622
Functionforeach_instance
(metadata, output_dir, func, max_workers=None, desc='Processing objects')
data_toolkit/datasets/ABO.py:80
Functionforeach_instance
(metadata, output_dir, func, max_workers=None, desc='Processing objects', log_interval=500, timeout=None)
data_toolkit/datasets/ObjaverseXL.py:83
Functionforeach_instance
(metadata, output_dir, func, max_workers=None, desc='Processing objects', timeout=None)
data_toolkit/datasets/TexVerse.py:34
Methodforward
(self, x: torch.Tensor)
pixal3d/modules/norm.py:18
Methodforward
(self, x: torch.Tensor)
pixal3d/modules/norm.py:26
Methodforward
(self, x: torch.Tensor, context: Union[Dict[str, torch.Tensor], Tuple[torch.Tensor, torch.Tensor]])
pixal3d/modules/attention/proj_attention.py:38
Methodforward
(self, x: torch.Tensor, context: Union[Dict[str, torch.Tensor], Tuple])
pixal3d/modules/attention/proj_attention.py:91
Methodforward
Args: indices (torch.Tensor): [..., N, C] tensor of spatial positions
pixal3d/modules/attention/rope.py:35
Methodforward
(self, x: torch.Tensor)
pixal3d/modules/attention/modules.py:15
Methodforward
(self, x: torch.Tensor, context: Optional[torch.Tensor] = None, phases: Optional[torch.Tensor] = None)
pixal3d/modules/attention/modules.py:66
Methodforward
(self, input: VarLenTensor)
pixal3d/modules/sparse/linear.py:14
Methodforward
(self, input: VarLenTensor)
pixal3d/modules/sparse/norm.py:34
Methodforward
(self, x: VarLenTensor)
pixal3d/modules/sparse/norm.py:49
Methodforward
(self, x: VarLenTensor)
pixal3d/modules/sparse/norm.py:60
Methodforward
(self, input: VarLenTensor)
pixal3d/modules/sparse/nonlinearity.py:19
Methodforward
(self, input: VarLenTensor)
pixal3d/modules/sparse/nonlinearity.py:24
Methodforward
(self, input: VarLenTensor)
pixal3d/modules/sparse/nonlinearity.py:33
Methodforward
( self, x: SparseTensor, context: Union[Dict[str, Union[torch.Tensor, VarLenTensor,
pixal3d/modules/sparse/attention/proj_attention.py:26
Methodforward
( self, x: SparseTensor, context: Union[Dict[str, Union[torch.Tensor, VarLenTensor, Sp
pixal3d/modules/sparse/attention/proj_attention.py:79
Methodforward
Args: q (SparseTensor): [..., N, H, D] tensor of queries k (SparseTensor): [..., N, H, D] tensor of keys
pixal3d/modules/sparse/attention/rope.py:35
Methodforward
(self, x: Union[VarLenTensor, torch.Tensor])
pixal3d/modules/sparse/attention/modules.py:17
Methodforward
(self, x: SparseTensor, context: Optional[Union[VarLenTensor, torch.Tensor]] = None)
pixal3d/modules/sparse/attention/modules.py:99
Methodforward
(self, x: SparseTensor)
pixal3d/modules/sparse/spatial/spatial2channel.py:16
Methodforward
(self, x: SparseTensor, subdivision: Optional[SparseTensor] = None)
pixal3d/modules/sparse/spatial/spatial2channel.py:67
Methodforward
(self, x: SparseTensor)
pixal3d/modules/sparse/spatial/basic.py:23
Methodforward
(self, x: SparseTensor, subdivision: Optional[SparseTensor] = None)
pixal3d/modules/sparse/spatial/basic.py:82
Methodforward
(self, x: SparseTensor)
pixal3d/modules/sparse/conv/conv.py:18
Methodforward
(self, x: SparseTensor)
pixal3d/modules/sparse/conv/conv.py:29
Methodforward
(self, x: VarLenTensor)
pixal3d/modules/sparse/transformer/blocks.py:20
Methodforward
(self, x: SparseTensor)
pixal3d/modules/sparse/transformer/blocks.py:72
Methodforward
(self, x: SparseTensor, context: Union[torch.Tensor, VarLenTensor])
pixal3d/modules/sparse/transformer/blocks.py:141
Methodforward
(self, x: SparseTensor, mod: torch.Tensor)
pixal3d/modules/sparse/transformer/modulated.py:74
Methodforward
(self, x: SparseTensor, mod: torch.Tensor, context: Union[torch.Tensor, VarLenTensor])
pixal3d/modules/sparse/transformer/modulated.py:201
Methodforward
Args: x (torch.Tensor): (N, D) tensor of spatial positions
pixal3d/modules/transformer/blocks.py:35
Methodforward
(self, x: torch.Tensor)
pixal3d/modules/transformer/blocks.py:58
Methodforward
(self, x: torch.Tensor, phases: Optional[torch.Tensor] = None)
pixal3d/modules/transformer/blocks.py:110
Methodforward
(self, x: torch.Tensor, context: torch.Tensor, phases: Optional[torch.Tensor] = None)
pixal3d/modules/transformer/blocks.py:181
Methodforward
(self, x: torch.Tensor, mod: torch.Tensor, phases: Optional[torch.Tensor] = None)
pixal3d/modules/transformer/modulated.py:73
Methodforward
(self, x: torch.Tensor, mod: torch.Tensor, context: torch.Tensor, phases: Optional[torch.Tensor] = None)
pixal3d/modules/transformer/modulated.py:200
Methodforward
(self, *args, **kwargs)
pixal3d/utils/elastic_utils.py:219
Methodforward
Project 3D grid points to image and sample features. Args: features_map: Feature map, shape [B, H, W, C] if BHWC
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:180
Methodforward
Extract view-aligned features from the image. Args: image: Input image tensor [B, C, H, W] or list of PIL images
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:464
Methodforward
Extract gated features from the image. Returns: Tuple of (global_features, proj_semantic, proj_color):
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:734
Methodforward
(self, t)
pixal3d/models/sparse_structure_flow.py:50
Methodforward
Forward pass. Args: x: Input tensor [B, C, D, H, W] t: Timestep tensor [B] cond: Conditi
pixal3d/models/sparse_structure_flow.py:244
Methodforward
(self, x: torch.Tensor)
pixal3d/models/sparse_structure_vae.py:39
Methodforward
(self, x: torch.Tensor)
pixal3d/models/sparse_structure_vae.py:68
Methodforward
(self, x: torch.Tensor)
pixal3d/models/sparse_structure_vae.py:93
Methodforward
(self, x: torch.Tensor, sample_posterior: bool = False, return_raw: bool = False)
pixal3d/models/sparse_structure_vae.py:186
Methodforward
(self, x: torch.Tensor)
pixal3d/models/sparse_structure_vae.py:295
Methodforward
Forward pass. Args: x: SparseTensor input t: Timestep tensor [B] cond: Conditioning tens
pixal3d/models/structured_latent_flow.py:190
Methodforward
(self, x: sp.SparseTensor)
pixal3d/models/sc_vaes/sparse_unet_vae.py:86
Methodforward
(self, x: sp.SparseTensor)
pixal3d/models/sc_vaes/sparse_unet_vae.py:124
Methodforward
(self, x: sp.SparseTensor)
pixal3d/models/sc_vaes/sparse_unet_vae.py:172
Methodforward
(self, x: sp.SparseTensor)
pixal3d/models/sc_vaes/sparse_unet_vae.py:210
Methodforward
(self, x: sp.SparseTensor, subdiv: sp.SparseTensor = None)
pixal3d/models/sc_vaes/sparse_unet_vae.py:258
Methodforward
(self, x: sp.SparseTensor)
pixal3d/models/sc_vaes/sparse_unet_vae.py:290
Methodforward
(self, x: sp.SparseTensor, sample_posterior=False, return_raw=False)
pixal3d/models/sc_vaes/sparse_unet_vae.py:373
Methodforward
(self, x: sp.SparseTensor, guide_subs: Optional[List[sp.SparseTensor]] = None, return_subs: bool = False)
pixal3d/models/sc_vaes/sparse_unet_vae.py:478
Methodforward
(self, x: sp.SparseTensor, gt_intersected: sp.SparseTensor = None, **kwargs)
pixal3d/models/sc_vaes/fdg_vae.py:83
Functionfrom_pretrained
Load a pipeline from a model folder or a Hugging Face model hub. Args: path: The path to the model. Can be either local path or a Hu
pixal3d/pipelines/__init__.py:28
Functionfrom_pretrained
Load a model from a pretrained checkpoint. Args: path: The path to the checkpoint. Can be either local path or a Hugging Face model
pixal3d/models/__init__.py:38
Methodfrom_pretrained
Load a pretrained model. Args: path (str): The path to the model. Can be either local path or a Hugging Face repository.
pixal3d/pipelines/trellis2_image_to_3d.py:82
Methodfrom_pretrained
Load a pretrained model. Args: path (str): The path to the model. Can be either local path or a Hugging Face repository.
pixal3d/pipelines/pixal3d_image_to_3d.py:95
Methodfrom_pretrained
Load a pretrained model. Args: path (str): The path to the model. Can be either local path or a Hugging Face repository.
pixal3d/pipelines/trellis2_texturing.py:69
Methodfrom_tensor_list
Create a SparseTensor from a list of tensors.
pixal3d/modules/sparse/basic.py:435
Functiongamma_correction
Applies gamma correction to an HDR image tensor.
pixal3d/renderers/pbr_mesh_renderer.py:187
Functiongenerate_3d
( image: FileData, seed: int, resolution: int, ss_guidance_strength: float = 7.5, ss_gui
app.py:383
Methodget_batch_size
(cond)
pixal3d/trainers/flow_matching/mixins/classifier_free_guidance.py:20
Methodget_cond
Get the conditioning data.
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:1394
Methodget_cond
Get the conditioning data.
pixal3d/trainers/flow_matching/mixins/classifier_free_guidance.py:12
Methodget_cond
Get the conditioning data.
pixal3d/trainers/flow_matching/mixins/image_conditioned.py:211
Methodget_cond
Get the conditioning data.
pixal3d/trainers/flow_matching/mixins/text_conditioned.py:52
Functionget_config
Return server configuration for frontend (e.g. LOW_VRAM mode).
app.py:355
Methodget_inference_cond
Get the conditioning data for inference.
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:1447
Methodget_inference_cond
Get the conditioning data for inference.
pixal3d/trainers/flow_matching/mixins/classifier_free_guidance.py:49
Methodget_inference_cond
Get the conditioning data for inference.
pixal3d/trainers/flow_matching/mixins/image_conditioned.py:222
Methodget_inference_cond
Get the conditioning data for inference.
pixal3d/trainers/flow_matching/mixins/text_conditioned.py:61
← previousnext →601–700 of 865, ranked by callers