File size: 387 Bytes
a6dd040 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from ...dataset import DatasetCfg
from .decoder import Decoder
from .decoder_splatting_cuda import DecoderSplattingCUDA, DecoderSplattingCUDACfg
DECODERS = {
"splatting_cuda": DecoderSplattingCUDA,
}
DecoderCfg = DecoderSplattingCUDACfg
def get_decoder(decoder_cfg: DecoderCfg, dataset_cfg: DatasetCfg) -> Decoder:
return DECODERS[decoder_cfg.name](decoder_cfg, dataset_cfg)
|