MCPcopy Create free account
hub / github.com/JasonLSC/GSCodec_Studio / AttributeCodecs

Class AttributeCodecs

examples/simple_trainer_old.py:98–110  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

96
97@dataclass
98class AttributeCodecs:
99 means: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_png_16bit", "_decompress_png_16bit"))
100 scales: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_factorized_ans", "_decompress_factorized_ans"))
101 quats: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_factorized_ans", "_decompress_factorized_ans"))
102 opacities: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_png", "_decompress_png"))
103 sh0: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_png", "_decompress_png"))
104 shN: CodecConfig = field(default_factory=lambda: CodecConfig("_compress_masked_kmeans", "_decompress_masked_kmeans"))
105
106 def to_dict(self) -> Dict[str, Dict[str, str]]:
107 return {
108 attr: {"encode": getattr(self, attr).encode, "decode": getattr(self, attr).decode}
109 for attr in ["means", "scales", "quats", "opacities", "sh0", "shN"]
110 }
111
112@dataclass
113class CompressionConfig:

Callers 1

CompressionConfigClass · 0.70

Calls 1

CodecConfigClass · 0.70

Tested by

no test coverage detected