From 3a2d6843157cdda2d20c7e813d0851aaa9a5300c Mon Sep 17 00:00:00 2001
From: KlemenSkrlj <47853619+klemen1999@users.noreply.github.com>
Date: Fri, 20 Sep 2024 11:41:06 +0200
Subject: [PATCH] [Fix] Updated  `EfficientBBoxHead` parser to `YOLO` (#74)

Co-authored-by: Martin Kozlovsky <martin.kozlovsky@luxonis.com>
---
 luxonis_train/nodes/enums/head_categorization.py | 2 +-
 tests/integration/parking_lot.json               | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/luxonis_train/nodes/enums/head_categorization.py b/luxonis_train/nodes/enums/head_categorization.py
index d36c9647..90f75725 100644
--- 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
index 0059241e..a42b82b2 100644
--- 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": [