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