Converts a depth image to color for easier analysis. Args: depth (torch.Tensor): (..., 1) float32 depth. acc (torch.Tensor | None

Computes total variations of bilateral grids.

Calculate memory per frame in bytes from bitrate in kbps.

Calculate memory per frame in bytes from bitrate in kbps.

Sanity check for the parameters and optimizers. Check if: * `params` and `optimizers` have the same keys. * Each opti

Sanity check for the parameters and optimizers. Check if: * `params` and `optimizers` have the same keys. * Each opti

Sanity check for the parameters and optimizers. Check if: * `params` and `optimizers` have the same keys. * Each opti

Run compression Args: compress_dir (str): directory to save compressed files splats (Dict[str, Tensor]): Gaussian spl

Run compression Args: compress_dir (str): directory to save compressed files splats (Dict[str, Tensor]): Gaussian spl

Run compression Args: compress_dir (str): directory to save compressed files splats (Dict[str, Tensor]): Gaussian spl

Run compression Args: compress_dir (str): directory to save compressed files splats (Dict[str, Tensor]): Gaussian spl

Run compression Args: splats_videos (Dict[str, Tensor]): dictionary of splats videos in Tensor format compress_dir (s

Run compression Args: compress_dir (str): directory to save compressed files

Compresses a list of splat frames. 1. Reorganizes data into attribute-based videos. 2. Saves each video as a raw YUV

Convert MP4 video to PNG sequence using ffmpeg Args: input_mp4 (str): Path to input MP4 file output_path (str): Director

Convert MP4 video to PNG sequence using ffmpeg with optional resolution specification Args: input_mp4 (str): Path to input MP4 f

Get the cuda toolkit version.

Decode artifacts back into tensors (usually integer labels/values).

Run decompression Args: compress_dir (str): directory that contains compressed files Returns: Dict[str, Tens

Run decompression Args: compress_dir (str): directory that contains compressed files Returns: Dict[str, Tens

Run decompression Args: compress_dir (str): directory that contains compressed files Returns: Dict[str, Tens

Run decompression Args: compress_dir (str): directory that contains compressed files Returns: Dict[str, Tens

Run decompression Args: compress_dir (str): directory that contains compressed files Returns: Dict[str, Tens

Run decompression Args: compress_dir (str): directory that contains compressed files Returns: Dict[str, Tens

default attribute configs

default qp values

default qp values

Deorganize the dictionary of splats videos into a list of splats. Args: splats_videos_c (Dict[str, Tensor]): dictionary o

Reconstruct floating tensors from quantized tensors and context.

Encode quantized splats to files and return artifact metadata.

Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d

Convert byte size to readable format (KB, MB, GB, etc.)

Convert byte size to readable format (KB, MB, GB, etc.)