Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
"""Adapter registry and auto detection for uploaded datasets."""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from as_platform.data.ingest.base import IngestAdapter, IngestContext, NormalizedDataset
|
|
from as_platform.data.ingest.dms_coco import DmsCocoAdapter
|
|
from as_platform.data.ingest.dms_yolo import DmsYoloAdapter
|
|
from as_platform.data.ingest.lane_lines import LaneLinesAdapter
|
|
from as_platform.data.ingest.lane_mask import LaneMaskAdapter
|
|
|
|
|
|
class UnknownFormatError(ValueError):
|
|
pass
|
|
|
|
|
|
ADAPTERS: tuple[IngestAdapter, ...] = (
|
|
DmsYoloAdapter(),
|
|
DmsCocoAdapter(),
|
|
LaneMaskAdapter(),
|
|
LaneLinesAdapter(),
|
|
)
|
|
|
|
|
|
def available_formats(project: str) -> list[str]:
|
|
return [a.format_id for a in ADAPTERS if project in a.projects]
|
|
|
|
|
|
def detect_adapter(ctx: IngestContext) -> IngestAdapter:
|
|
for adapter in ADAPTERS:
|
|
if ctx.project not in adapter.projects:
|
|
continue
|
|
if adapter.can_handle(ctx):
|
|
return adapter
|
|
raise UnknownFormatError(
|
|
f"unable to detect format for project={ctx.project}, task={ctx.task}, "
|
|
f"source={ctx.source_path}. supported={available_formats(ctx.project)}"
|
|
)
|
|
|
|
|
|
def inspect_uploaded_dataset(project: str, task: str | None, source_path: str | Path) -> NormalizedDataset:
|
|
ctx = IngestContext(project=project, task=task, source_path=Path(source_path).resolve())
|
|
if not ctx.source_path.exists():
|
|
raise FileNotFoundError(f"source path not found: {ctx.source_path}")
|
|
adapter = detect_adapter(ctx)
|
|
out = adapter.inspect(ctx)
|
|
# Ensure adapter id is always reflected in output.
|
|
out.format_id = adapter.format_id
|
|
return out
|