Spaces:

hugging-apps
/

codecsplat

Running on Zero

App Files Files Community

codecsplat / src /model /types.py

multimodalart HF Staff

CodecSplat demo

7405e85 verified 7 days ago

Raw History Blame Contribute Delete

516 Bytes

from dataclasses import dataclass

from jaxtyping import Float
from torch import Tensor


@dataclass
class Gaussians:
    means: Float[Tensor, "batch gaussian dim"]
    covariances: Float[Tensor, "batch gaussian dim dim"]
    harmonics: Float[Tensor, "batch gaussian 3 d_sh"]
    opacities: Float[Tensor, "batch gaussian"]
    scales: Float[Tensor, "batch gaussian 3"] | None = None
    rotations: Float[Tensor, "batch gaussian 4"] | None = None
    rotations_unnorm: Float[Tensor, "batch gaussian 4"] | None = None

	from dataclasses import dataclass

	from jaxtyping import Float
	from torch import Tensor


	@dataclass
	class Gaussians:
	means: Float[Tensor, "batch gaussian dim"]
	covariances: Float[Tensor, "batch gaussian dim dim"]
	harmonics: Float[Tensor, "batch gaussian 3 d_sh"]
	opacities: Float[Tensor, "batch gaussian"]
	scales: Float[Tensor, "batch gaussian 3"] \| None = None
	rotations: Float[Tensor, "batch gaussian 4"] \| None = None
	rotations_unnorm: Float[Tensor, "batch gaussian 4"] \| None = None