Spaces:
Running on Zero
Running on Zero
| from src.model.encoder.encoder import Encoder | |
| from src.model.encoder.encoder_infinidepth_query import ( | |
| EncoderInfiniDepthQuery, | |
| EncoderInfiniDepthQueryCfg, | |
| ) | |
| from src.model.encoder.encoder_infinisplat import EncoderInfiniSplat, EncoderInfiniSplatCfg | |
| ENCODERS = { | |
| "infinisplat": EncoderInfiniSplat, | |
| "infinisplat_infinidepth": EncoderInfiniDepthQuery, | |
| } | |
| EncoderCfg = EncoderInfiniSplatCfg | EncoderInfiniDepthQueryCfg | |
| def get_encoder(cfg: EncoderCfg) -> Encoder: | |
| encoder = ENCODERS[cfg.name] | |
| encoder = encoder(cfg) | |
| return encoder | |