|
|
|
|
| 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",
|
| )
|
|
|
