From 523b7751175593bd428c2e85874b3b51ba7b2e95 Mon Sep 17 00:00:00 2001
From: Jernej Sabadin <116955183+JSabadin@users.noreply.github.com>
Date: Tue, 19 Nov 2024 09:55:50 +0100
Subject: [PATCH] Keypoint M predefined model (#136)

---
 .../config/predefined_models/keypoint_detection_model.py     | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/luxonis_train/config/predefined_models/keypoint_detection_model.py b/luxonis_train/config/predefined_models/keypoint_detection_model.py
index cebae105..51d790a7 100644
--- a/luxonis_train/config/predefined_models/keypoint_detection_model.py
+++ b/luxonis_train/config/predefined_models/keypoint_detection_model.py
@@ -30,6 +30,11 @@ def get_variant(variant: VariantLiteral) -> KeypointDetectionVariant:
             backbone_params={"variant": "n"},
             neck_params={"variant": "n", "download_weights": True},
         ),
+        "medium": KeypointDetectionVariant(
+            backbone="EfficientRep",
+            backbone_params={"variant": "s"},
+            neck_params={"variant": "s", "download_weights": True},
+        ),
         "heavy": KeypointDetectionVariant(
             backbone="EfficientRep",
             backbone_params={"variant": "l"},