Add custom YOLO prediction and heatmap workflow

2026-06-30 15:11:47 +08:00
parent 4d0c26be05
commit 777f168a75
12 changed files with 393 additions and 17 deletions
--- a/backend/tests/test_catalog.py
+++ b/backend/tests/test_catalog.py
@@ -8,6 +8,8 @@ def test_catalog_contains_required_capabilities():
        "dataset.video_frames",
        "segmodel.train",
        "yolo.train_custom",
+        "yolo.predict_custom",
+        "yolo.heatmap_custom",
        "yolo.predict",
        "mmseg.flops_fps",
        "analysis.all",