diff --git a/luxonis_train/core/core.py b/luxonis_train/core/core.py index 05fff23f..fea53acc 100644 --- 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 index a42b82b2..28ca0a61 100644 --- 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,