[Fix] Updated EfficientBBoxHead parser to YOLO (#74)

Co-authored-by: Martin Kozlovsky <[email protected]>
luxonis · Sep 20, 2024 · e7ab0c0 · e7ab0c0
1 parent 1297a04
commit e7ab0c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/luxonis_train/nodes/enums/head_categorization.py b/luxonis_train/nodes/enums/head_categorization.py
@@ -5,7 +5,7 @@ class ImplementedHeads(Enum):
     """Task categorization for the implemented heads."""
 
     ClassificationHead = "ClassificationParser"
-    EfficientBBoxHead = "YoloDetectionNetwork"
+    EfficientBBoxHead = "YOLO"
     ImplicitKeypointBBoxHead = "YoloDetectionNetwork"
     EfficientKeypointBBoxHead = "YoloDetectionNetwork"
     SegmentationHead = "SegmentationParser"

diff --git a/tests/integration/parking_lot.json b/tests/integration/parking_lot.json
@@ -158,7 +158,7 @@
         ],
         "heads": [
             {
-                "parser": "YoloDetectionNetwork",
+                "parser": "YOLO",
                 "metadata": {
                     "postprocessor_path": null,
                     "classes": [