|
|
|
|
| import os
|
| from typing import Dict, Optional
|
|
|
| from detectron2.config import CfgNode
|
|
|
|
|
| def is_relative_local_path(path: str) -> bool:
|
| path_str = os.fsdecode(path)
|
| return ("://" not in path_str) and not os.path.isabs(path)
|
|
|
|
|
| def maybe_prepend_base_path(base_path: Optional[str], path: str):
|
| """
|
| Prepends the provided path with a base path prefix if:
|
| 1) base path is not None;
|
| 2) path is a local path
|
| """
|
| if base_path is None:
|
| return path
|
| if is_relative_local_path(path):
|
| return os.path.join(base_path, path)
|
| return path
|
|
|
|
|
| def get_class_to_mesh_name_mapping(cfg: CfgNode) -> Dict[int, str]:
|
| return {
|
| int(class_id): mesh_name
|
| for class_id, mesh_name in cfg.DATASETS.CLASS_TO_MESH_NAME_MAPPING.items()
|
| }
|
|
|
|
|
| def get_category_to_class_mapping(dataset_cfg: CfgNode) -> Dict[str, int]:
|
| return {
|
| category: int(class_id)
|
| for category, class_id in dataset_cfg.CATEGORY_TO_CLASS_MAPPING.items()
|
| }
|
|
|
