nerfbaselines.registry¶
- class nerfbaselines.registry.DatasetSpec[source]¶
Bases:
TypedDict
- download_dataset_function: str¶
- evaluation_protocol: str | EvaluationProtocolWithNameSpec¶
- load_dataset_function: Required[str]¶
- metadata: DatasetSpecMetadata¶
- priority: Required[int]¶
- class nerfbaselines.registry.EvaluationProtocolSpec[source]¶
Bases:
TypedDict
- evaluation_protocol: Required[str]¶
- class nerfbaselines.registry.EvaluationProtocolWithNameSpec[source]¶
Bases:
EvaluationProtocolSpec
- evaluation_protocol: Required[str]¶
- name: Required[str]¶
- class nerfbaselines.registry.MethodSpec[source]¶
Bases:
TypedDict
- apptainer: NotRequired[dict]¶
- backends_order: List[str]¶
- conda: NotRequired[dict]¶
- dataset_overrides: Dict[str, Any]¶
- docker: NotRequired[dict]¶
- kwargs: Dict[str, Any]¶
- metadata: Dict[str, Any]¶
- method: Required[str]¶
- nerfbaselines.registry.build_evaluation_protocol(name: str) EvaluationProtocol [source]¶
- nerfbaselines.registry.get_dataset_downloaders() Dict[str, DownloadDatasetFunction] [source]¶
- nerfbaselines.registry.get_dataset_loaders() Sequence[Tuple[str, LoadDatasetFunction]] [source]¶
- nerfbaselines.registry.get_dataset_overrides(method_name: str, dataset_metadata: Dict[str, Any]) Dict[str, Any] | None [source]¶
- nerfbaselines.registry.get_dataset_spec(name: str) DatasetSpec [source]¶
- nerfbaselines.registry.get_method_spec(name: str) MethodSpec [source]¶
Get a method by name
- nerfbaselines.registry.get_spec_type(spec: MethodSpec | DatasetSpec | EvaluationProtocolSpec) Literal['method', 'dataset', 'evaluation_protocol'] [source]¶
- nerfbaselines.registry.get_supported_methods(backend_name: str | None = None) FrozenSet[str] [source]¶
- nerfbaselines.registry.register(spec: MethodSpec | DatasetSpec | EvaluationProtocolSpec, *, name: str, metadata=None, kwargs=None, dataset_overrides=None) None [source]¶