Compute shortcut model loss with mixed flow matching and self-consistency objectives

Creates a video by updating the camera view location and saving snapshots. Args: scene: A Plotly Figure object, typically created wi

Applies data augmentation to the mask by randomly eroding or dilating the mask. Args: image: (C, H, W) float32 [0, 1] tensor.

Applies data augmentation to the mask by randomly translating the mask. Args: image: (C, H, W) float32 [0, 1] tensor. mask:

Complete pipeline for processing 3DB alignment and saving the result. Args: mesh_path: Path to input 3DB mesh (.ply) mask_pa

Resize RGB image, mask, and pointmap to the same size. This is crucial when pointmaps have different resolution than RGB images, whi

Scale the parameters of a module and return it.

Simplify 3D Gaussians NOTE: this function is not used in the current implementation for the unsatisfactory performance. Args: gs

Broadcast a 1D tensor to a sparse tensor along the batch dimension then perform an operation. Args: input (torch.Tensor): 1D tensor

Concatenate a list of sparse tensors. Args: inputs (List[SparseTensor]): List of sparse tensors to concatenate.

Control the structure of the octree. Args: mask (torch.Tensor): the mask to control the structure. 1 for subdivide, -1 f

Recursively moves all tensors in obj to the specified device. Args: obj: Object to move to device - can be a tensor, list, tuple, dict or

Convert a generated asset to a glb file. Args: app_rep (Union[Strivec, Gaussian]): Appearance representation. mesh (MeshExtr

Simple visualization of both meshes in a single 3D plot. DEPRECATED: Use visualize_meshes_interactive() for better interactive visualization

Interactive Gradio-based 3D visualization of aligned human and object meshes. Args: aligned_mesh_path: Path to aligned mesh PLY file