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/app/capabilities.py
+++ b/backend/app/capabilities.py
@@ -59,8 +59,10 @@ CAPABILITY_GROUPS = [
            "yolo.train",
            "yolo.train_custom",
            "yolo.predict",
+            "yolo.predict_custom",
            "yolo.batch_predict",
            "yolo.heatmap",
+            "yolo.heatmap_custom",
            "yolo.compare",
            "yolo.raw_mask_check",
            "yolo.video_visible",