[Fix] Updated parser to

luxonis · Sep 17, 2024 · 026ffc2 · 026ffc2
1 parent b3f9f5d
commit 026ffc2
Showing 1 changed file with 1 addition and 1 deletion.
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"