Add dataset QA and custom YOLO training flow

2026-06-30 14:04:11 +08:00
parent 43ed767b4f
commit 93af8bcd3a
14 changed files with 529 additions and 18 deletions
--- a/backend/tests/test_catalog.py
+++ b/backend/tests/test_catalog.py
@@ -7,9 +7,9 @@ def test_catalog_contains_required_capabilities():
    for task in [
        "dataset.video_frames",
        "segmodel.train",
+        "yolo.train_custom",
        "yolo.predict",
        "mmseg.flops_fps",
        "analysis.all",
    ]:
        assert task in TASK_TYPES
-