Spaces:
Running on Zero
Running on Zero
File size: 1,236 Bytes
7405e85 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | from typing import Any
from ...misc.step_tracker import StepTracker
from ..types import Stage
from .view_sampler import ViewSampler
from .view_sampler_all import ViewSamplerAll, ViewSamplerAllCfg
from .view_sampler_arbitrary import ViewSamplerArbitrary, ViewSamplerArbitraryCfg
from .view_sampler_bounded import ViewSamplerBounded, ViewSamplerBoundedCfg
from .view_sampler_evaluation import ViewSamplerEvaluation, ViewSamplerEvaluationCfg
from .view_sampler_bounded_v2 import ViewSamplerBoundedV2, ViewSamplerBoundedV2Cfg
VIEW_SAMPLERS: dict[str, ViewSampler[Any]] = {
"all": ViewSamplerAll,
"arbitrary": ViewSamplerArbitrary,
"bounded": ViewSamplerBounded,
"evaluation": ViewSamplerEvaluation,
"boundedv2": ViewSamplerBoundedV2,
}
ViewSamplerCfg = (
ViewSamplerArbitraryCfg
| ViewSamplerBoundedCfg
| ViewSamplerEvaluationCfg
| ViewSamplerAllCfg
| ViewSamplerBoundedV2Cfg
)
def get_view_sampler(
cfg: ViewSamplerCfg,
stage: Stage,
overfit: bool,
cameras_are_circular: bool,
step_tracker: StepTracker | None,
) -> ViewSampler[Any]:
return VIEW_SAMPLERS[cfg.name](
cfg,
stage,
overfit,
cameras_are_circular,
step_tracker,
)
|