75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
|
"""ADAS class_id 映射（BK2/MOON 单源）。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
import yaml
|
|||
|
|
|
|||
|
|
from as_platform.config import WORKSPACE
|
|||
|
|
|
|||
|
|
_ADAS_REGISTRY = WORKSPACE / "datasets" / "adas" / "adas.registry.yaml"
|
|||
|
|
_LABELING_REGISTRY = WORKSPACE / "datasets" / "labeling.registry.yaml"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_adas_class_names() -> list[str]:
|
|||
|
|
if _ADAS_REGISTRY.is_file():
|
|||
|
|
reg = yaml.safe_load(_ADAS_REGISTRY.read_text(encoding="utf-8")) or {}
|
|||
|
|
names = (reg.get("classes") or {}).get("names")
|
|||
|
|
if names:
|
|||
|
|
return [str(n) for n in names]
|
|||
|
|
if _LABELING_REGISTRY.is_file():
|
|||
|
|
reg = yaml.safe_load(_LABELING_REGISTRY.read_text(encoding="utf-8")) or {}
|
|||
|
|
labels = (reg.get("profiles") or {}).get("cuboid_7cls", {}).get("cvat_labels")
|
|||
|
|
if labels:
|
|||
|
|
return [str(n) for n in labels]
|
|||
|
|
from as_platform.labeling.format_converter import CUBOID_7CLS_NAMES
|
|||
|
|
|
|||
|
|
return list(CUBOID_7CLS_NAMES)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def class_name_to_id(name: str, class_map: dict[str, int] | None = None) -> int | None:
|
|||
|
|
cmap = class_map or {n: i for i, n in enumerate(load_adas_class_names())}
|
|||
|
|
if name in cmap:
|
|||
|
|
return cmap[name]
|
|||
|
|
low = name.lower()
|
|||
|
|
for k, v in cmap.items():
|
|||
|
|
if k.lower() == low:
|
|||
|
|
return v
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_class_map(names: list[str] | None = None) -> dict[str, int]:
|
|||
|
|
return {str(n): idx for idx, n in enumerate(names or load_adas_class_names())}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def remap_class_id(old_names: list[str], new_names: list[str], class_id: int) -> int:
|
|||
|
|
if class_id < 0 or class_id >= len(old_names):
|
|||
|
|
return class_id
|
|||
|
|
label = old_names[class_id]
|
|||
|
|
new_id = build_class_map(new_names).get(label)
|
|||
|
|
if new_id is None:
|
|||
|
|
for k, v in build_class_map(new_names).items():
|
|||
|
|
if k.lower() == label.lower():
|
|||
|
|
return v
|
|||
|
|
return new_id if new_id is not None else class_id
|
|||
|
|
|
|||
|
|
|
|||
|
|
def normalize_detection_class(det: dict[str, Any], class_map: dict[str, int] | None = None) -> dict[str, Any]:
|
|||
|
|
cmap = class_map or build_class_map()
|
|||
|
|
name = str(det.get("class_name") or "")
|
|||
|
|
cid = det.get("class_id")
|
|||
|
|
if name:
|
|||
|
|
mapped = class_name_to_id(name, cmap)
|
|||
|
|
if mapped is not None:
|
|||
|
|
det = dict(det)
|
|||
|
|
det["class_id"] = mapped
|
|||
|
|
det["class_name"] = name
|
|||
|
|
elif cid is not None:
|
|||
|
|
names = list(cmap.keys())
|
|||
|
|
idx = int(cid)
|
|||
|
|
if 0 <= idx < len(names):
|
|||
|
|
det = dict(det)
|
|||
|
|
det["class_name"] = names[idx]
|
|||
|
|
return det
|