NN Archive - v1.1 (#93)

luxonis · Oct 3, 2024 · 2aaeb52 · 2aaeb52
1 parent a283ea1
commit 2aaeb52
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/luxonis_train/core/core.py b/luxonis_train/core/core.py
@@ -704,7 +704,7 @@ def _mult(lst: list[float | int]) -> list[float]:
         }
 
         cfg_dict = {
-            "config_version": CONFIG_VERSION.__args__[0],  # type: ignore
+            "config_version": CONFIG_VERSION.__args__[-1],  # type: ignore
             "model": model,
         }
 

diff --git a/tests/integration/parking_lot.json b/tests/integration/parking_lot.json
@@ -1,5 +1,5 @@
 {
-    "config_version": "1.0",
+    "config_version": "1.1",
     "model": {
         "metadata": {
             "name": "parking_lot_model",
@@ -158,6 +158,7 @@
         ],
         "heads": [
             {
+                "name": "bbox-head",
                 "parser": "YOLO",
                 "metadata": {
                     "postprocessor_path": null,
@@ -179,6 +180,7 @@
                 ]
             },
             {
+                "name": "car-detection-head",
                 "parser": "YoloDetectionNetwork",
                 "metadata": {
                     "postprocessor_path": null,
@@ -197,6 +199,7 @@
                 ]
             },
             {
+                "name": "motorbike-detection-head",
                 "parser": "YoloDetectionNetwork",
                 "metadata": {
                     "postprocessor_path": null,
@@ -217,6 +220,7 @@
                 ]
             },
             {
+                "name": "color-segmentation-head",
                 "parser": "SegmentationParser",
                 "metadata": {
                     "postprocessor_path": null,
@@ -234,6 +238,7 @@
                 ]
             },
             {
+                "name": "any-vehicle-segmentation-head",
                 "parser": "SegmentationParser",
                 "metadata": {
                     "postprocessor_path": null,
@@ -248,6 +253,7 @@
                 ]
             },
             {
+                "name": "brand-segmentation-head",
                 "parser": "SegmentationParser",
                 "metadata": {
                     "postprocessor_path": null,
@@ -284,6 +290,7 @@
                 ]
             },
             {
+                "name": "vehicle-type-segmentation-head",
                 "parser": "SegmentationParser",
                 "metadata": {
                     "postprocessor_path": null,