MCPcopy Create free account

hub / github.com/TencentARC/Pixal3D / functions

Functions865 in github.com/TencentARC/Pixal3D

⨍ Functions 865◇ Types & classes 147↳ Endpoints 3
Methodget_instance
(self, root, instance)
pixal3d/datasets/sparse_voxel_pbr.py:264
Methodget_instance
(self, root, instance)
pixal3d/datasets/structured_latent_shape.py:325
Methodget_instance
(self, root, instance)
pixal3d/datasets/structured_latent_svpbr.py:330
Methodget_instance
(self, root, instance)
pixal3d/datasets/structured_latent_svpbr.py:554
Methodget_instance
(self, root, instance)
pixal3d/datasets/sparse_structure_latent.py:276
Methodget_instance
(self, root, instance)
pixal3d/datasets/sparse_structure_latent.py:375
Methodget_instance
(self, root, instance)
pixal3d/datasets/structured_latent.py:164
Methodget_instance
(self, root, instance)
pixal3d/datasets/flexi_dual_grid.py:141
Methodget_instance
(self, root, instance)
pixal3d/datasets/components.py:162
Methodget_instance
Get instance with view-aligned image and camera info. Expects parent class to set: - pack['x_0']: the latent ten
pixal3d/datasets/components.py:219
Methodget_instance
(self, root, instance)
pixal3d/datasets/components.py:304
Methodget_mem_ratio
(self, input_size)
pixal3d/utils/elastic_utils.py:124
Functionget_metadata
(**kwargs)
data_toolkit/datasets/ABO.py:15
Functionget_metadata
(source, **kwargs)
data_toolkit/datasets/ObjaverseXL.py:15
Functionget_metadata
(**kwargs)
data_toolkit/datasets/TexVerse.py:12
Methodget_sampler
Get the sampler for the diffusion process.
pixal3d/trainers/flow_matching/mixins/classifier_free_guidance.py:56
Methodhalf
(self)
pixal3d/modules/sparse/basic.py:618
Functionhomepage
()
app.py:349
Methodinit_models_and_more
Override to handle image_cond_model initialization. Since proj_linear has been moved to per-block ProjectAttention in the de
pixal3d/trainers/flow_matching/mixins/image_conditioned_proj.py:849
Functionis_valid_sparse_tensor
(tensor)
data_toolkit/encode_ss_latent.py:18
Methodlayout
(self)
pixal3d/modules/sparse/basic.py:483
Methodlayout_from_seqlen
Create a layout from a tensor of sequence lengths.
pixal3d/modules/sparse/basic.py:31
Methodload
(self, path)
pixal3d/representations/voxel/voxel_model.py:42
Methodload_state_dict
(self, state_dict)
pixal3d/utils/grad_clip_utils.py:37
Methodload_state_dict
(self, state_dict)
pixal3d/utils/elastic_utils.py:135
Functionloader
(sha256)
data_toolkit/encode_shape_latent.py:121
Functionloader
(task)
data_toolkit/encode_shape_latent_view.py:159
Functionloader
(sha256)
data_toolkit/encode_ss_latent.py:115
Functionloader
(sha256)
data_toolkit/encode_pbr_latent.py:121
Functionloader
(task)
data_toolkit/encode_ss_latent_view.py:160
Functionloader
(task)
data_toolkit/encode_pbr_latent_view.py:161
Functionlocal_master_first
A context manager that ensures local master process executes first.
pixal3d/utils/dist_utils.py:81
Methodlog
(self)
pixal3d/utils/grad_clip_utils.py:43
Methodlog
(self)
pixal3d/utils/elastic_utils.py:138
Methodlr_lambda
(self, current_step)
pixal3d/trainers/utils.py:87
Functionmake_grid
(images, nrow=None, ncol=None, aspect_ratio=None)
data_toolkit/utils.py:225
Functionmake_grid
(images, nrow=None, ncol=None, aspect_ratio=None)
pixal3d/utils/general_utils.py:130
Functionmake_pbr_vis_frames
(result, resolution=1024)
pixal3d/utils/render_utils.py:211
Functionmaster_first
A context manager that ensures master process executes first.
pixal3d/utils/dist_utils.py:65
Functionmodulate
(x, shift, scale)
pixal3d/modules/utils.py:64
Methodndim
(self)
pixal3d/modules/sparse/basic.py:75
Functionnormal_angle
(pred, gt)
pixal3d/utils/loss_utils.py:83
Functionpatchify
Patchify a tensor. Args: x (torch.Tensor): (N, C, *spatial) tensor patch_size (int): Patch size
pixal3d/modules/spatial.py:16
Methodposition
(self)
pixal3d/representations/voxel/voxel_model.py:23
Methodprepare_dataloader
Prepare dataloader.
pixal3d/trainers/vae/shape_vae.py:92
Methodprepare_dataloader
Prepare dataloader.
pixal3d/trainers/vae/pbr_vae.py:82
Methodprepare_dataloader
Prepare dataloader.
pixal3d/trainers/flow_matching/sparse_flow_matching.py:57
Functionpreprocess
(image: FileData)
app.py:373
Functionprocess_one
Find the first existing GLB file for this uid and compute its sha256.
data_toolkit/build_texverse_metadata.py:32
Functionprogress_poll
Polling endpoint for real-time progress updates during generation.
app.py:360
Functionpsnr
(img1, img2, max_val=1.0)
pixal3d/utils/loss_utils.py:34
Methodquery_vertex_attrs
(self)
pixal3d/representations/mesh/base.py:233
Functionread_ply
Read a PLY file and return vertices, triangle faces, and quad faces. Args: filename (str): The file path to read from.
data_toolkit/utils.py:51
Functionread_ply
Read a PLY file and return vertices, triangle faces, and quad faces. Args: filename (str): The file path to read from.
pixal3d/utils/mesh_utils.py:9
Methodrecord
(self)
pixal3d/utils/elastic_utils.py:18
Methodrecord
(self)
pixal3d/utils/elastic_utils.py:87
Methodregister_memory_controller
(self, memory_controller: MemoryController)
pixal3d/utils/elastic_utils.py:216
Methodremove_faces
(self, face_mask: torch.Tensor)
pixal3d/representations/mesh/base.py:59
Functionrender_multiview
(sample, resolution=512, nviews=30)
pixal3d/utils/render_utils.py:106
Functionrender_proj_aligned_video
Render a turntable video starting from the proj camera viewpoint. The first frame matches the proj input image exactly. Subsequent frame
pixal3d/utils/render_utils.py:153
Functionrender_snapshot
(samples, resolution=512, bg_color=(0, 0, 0), offset=(-16 / 180 * np.pi, 20 / 180 * np.pi), r=10, fov=8, nview
pixal3d/utils/render_utils.py:117
Methodreshape
(self, *shape)
pixal3d/modules/sparse/basic.py:631
Methodreverse_diffuse
Get original image from noisy version under timestep t.
pixal3d/trainers/flow_matching/flow_matching.py:92
Functionrtol
Relative tolerance.
pixal3d/utils/general_utils.py:357
Methodrun
Run the pipeline. Args: image (Image.Image): The image prompt. num_samples (int): The number of samples to g
pixal3d/pipelines/trellis2_image_to_3d.py:504
Methodrun
Run the Pixal3D pipeline (proj mode, cascade). Args: image (Image.Image): The image prompt. camera_params (d
pixal3d/pipelines/pixal3d_image_to_3d.py:609
Methodrun
Run the pipeline. Args: mesh (trimesh.Trimesh): The mesh to texture. image (Image.Image): The image prompt.
pixal3d/pipelines/trellis2_texturing.py:375
Methodrun
Run training with projection visualization test before starting.
pixal3d/trainers/flow_matching/sparse_flow_matching.py:558
Methodrun
Run training with projection visualization test before starting.
pixal3d/trainers/flow_matching/flow_matching.py:598
Methodrun_snapshot
( self, num_samples: int, batch_size: int, verbose: bool = False, )
pixal3d/trainers/vae/shape_vae.py:219
Methodrun_snapshot
( self, num_samples: int, batch_size: int, verbose: bool = False, )
pixal3d/trainers/vae/pbr_vae.py:224
Methodrun_snapshot
( self, num_samples: int, batch_size: int, verbose: bool = False, )
pixal3d/trainers/vae/sparse_structure_vae.py:99
Methodrun_snapshot
( self, num_samples: int, batch_size: int, verbose: bool = False, )
pixal3d/trainers/flow_matching/sparse_flow_matching.py:126
Methodrun_snapshot
Run snapshot with coords passed to get_inference_cond for sparse feature extraction. For projection mode, we need to pass co
pixal3d/trainers/flow_matching/sparse_flow_matching.py:431
Methodrun_snapshot
( self, num_samples: int, batch_size: int, verbose: bool = False, )
pixal3d/trainers/flow_matching/flow_matching.py:188
Methodrun_snapshot
Run snapshot with camera parameters for GT view rendering. Overrides parent to include camera parameters in sample dicts for
pixal3d/trainers/flow_matching/flow_matching.py:446
Methodsample
Generate samples from the model using Euler method. Args: model: The model to sample from. noise: Th
pixal3d/pipelines/samplers/flow_euler.py:134
Methodsample
Generate samples from the model using Euler method. Args: model: The model to sample from. noise: Th
pixal3d/pipelines/samplers/flow_euler.py:174
Methodsample_shape_slat_cascade
Sample structured latent with cascade (LR → HR). Args: lr_cond (dict): The conditioning information for LR stage
pixal3d/pipelines/pixal3d_image_to_3d.py:390
Methodsave
(self, path)
pixal3d/representations/voxel/voxel_model.py:32
Functionsave_image_with_notes
Save an image with notes.
data_toolkit/utils.py:555
Functionsaver
(sha256, pack)
data_toolkit/encode_shape_latent.py:143
Functionsaver
(sha256, view_idx, pack)
data_toolkit/encode_shape_latent_view.py:204
Functionsaver
(sha256, pack)
data_toolkit/encode_ss_latent.py:129
Functionsaver
(sha256, pack)
data_toolkit/encode_pbr_latent.py:140
Functionsaver
(sha256, view_idx, pack)
data_toolkit/encode_ss_latent_view.py:206
Functionsaver
(sha256, view_idx, pack)
data_toolkit/encode_pbr_latent_view.py:213
Functionscale_module
Scale the parameters of a module and return it.
pixal3d/modules/utils.py:55
Methodselect
(cond, neg_cond, mask)
pixal3d/trainers/flow_matching/mixins/classifier_free_guidance.py:31
Methodseqlen
(self)
pixal3d/modules/sparse/basic.py:87
Methodseqlen
(self)
pixal3d/modules/sparse/basic.py:543
Functionset_attn_backend
(backend: Literal['xformers', 'flash_attn', 'flash_attn_3', 'flash_attn_4', 'sdpa'])
pixal3d/modules/sparse/config.py:41
Functionset_backend
(backend: Literal['xformers', 'flash_attn', 'flash_attn_3', 'flash_attn_4', 'sdpa', 'naive'])
pixal3d/modules/attention/config.py:26
Functionset_conv_backend
(backend: Literal['none', 'spconv', 'torchsparse', 'flex_gemm'])
pixal3d/modules/sparse/config.py:33
Functionset_debug
(debug: bool)
pixal3d/modules/attention/config.py:30
Functionset_debug
(debug: bool)
pixal3d/modules/sparse/config.py:37
Methodset_description
(self, desc=None, refresh=True)
app.py:324
Methodshape
(self)
pixal3d/modules/sparse/basic.py:68
Methodshape
(self)
pixal3d/modules/sparse/basic.py:477
Functionsmooth_l1_loss
(pred, target, beta=1.0)
pixal3d/utils/loss_utils.py:8
← previousnext →701–800 of 865, ranked by callers