| |
|
|
| from typing import Optional |
|
|
| from detectron2.data import DatasetCatalog, MetadataCatalog |
|
|
| from ..utils import maybe_prepend_base_path |
| from .dataset_type import DatasetType |
|
|
| CHIMPNSEE_DATASET_NAME = "chimpnsee" |
|
|
|
|
| def register_dataset(datasets_root: Optional[str] = None) -> None: |
| def empty_load_callback(): |
| pass |
|
|
| video_list_fpath = maybe_prepend_base_path( |
| datasets_root, |
| "chimpnsee/cdna.eva.mpg.de/video_list.txt", |
| ) |
| video_base_path = maybe_prepend_base_path(datasets_root, "chimpnsee/cdna.eva.mpg.de") |
|
|
| DatasetCatalog.register(CHIMPNSEE_DATASET_NAME, empty_load_callback) |
| MetadataCatalog.get(CHIMPNSEE_DATASET_NAME).set( |
| dataset_type=DatasetType.VIDEO_LIST, |
| video_list_fpath=video_list_fpath, |
| video_base_path=video_base_path, |
| category="chimpanzee", |
| ) |
|
|
