add new models

Smartappli · Jul 16, 2024 · aed092f · aed092f
1 parent 1f80505
commit aed092f
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 1 deletion.
diff --git a/fl_common/dlcl_getfamilly.py b/fl_common/dlcl_getfamilly.py
@@ -469,6 +469,7 @@ def get_family_model_e(model_type, num_classes):
         "semnasnet_140": get_efficientnet_model,
         "mnasnet_small": get_efficientnet_model,
         "mobilenet_100": get_efficientnet_model,
+        "mobilenet_100h": get_efficientnet_model,
         "mobilenet_125": get_efficientnet_model,
         "mobilenetv2_035": get_efficientnet_model,
         "mobilenetv2_050": get_efficientnet_model,
@@ -950,6 +951,8 @@ def get_family_model_m(model_type, num_classes):
         "mobilenetv4_hybrid_large": get_mobilenet_model,
         "mobilenetv4_conv_aa_medium": get_mobilenet_model,
         "mobilenetv4_conv_blur_medium": get_mobilenet_model,
+        "mobilenetv4_conv_aa_large": get_mobilenet_model,
+        "mobilenetv4_conv_blur_large": get_mobilenet_model,
         "mobilenetv4_hybrid_medium_075": get_mobilenet_model,
         "mobilenetv4_hybrid_large_075": get_mobilenet_model,
         "repghostnetv2_conv_small": get_mobilenet_model,

diff --git a/fl_common/models/efficientnet.py b/fl_common/models/efficientnet.py
@@ -77,7 +77,7 @@ def get_efficientnet_model(efficientnet_type, num_classes):
         "semnasnet_140",
         "mnasnet_small",
         "mobilenet_100",
-        # "mobilenet_100h",
+        "mobilenet_100h",
         "mobilenet_125",
         "mobilenetv2_035",
         "mobilenetv2_050",

diff --git a/fl_common/models/mobilenet.py b/fl_common/models/mobilenet.py
@@ -69,6 +69,8 @@ def get_mobilenet_model(mobilenet_type, num_classes):
         "mobilenetv4_hybrid_large",
         "mobilenetv4_conv_aa_medium",
         "mobilenetv4_conv_blur_medium",
+        "mobilenetv4_conv_aa_large",
+        "mobilenetv4_conv_blur_large",
         "mobilenetv4_hybrid_medium_075",
         "mobilenetv4_hybrid_large_075",
         "repghostnetv2_conv_small",

diff --git a/fl_common/tests/tests_models_efficientnet.py b/fl_common/tests/tests_models_efficientnet.py
@@ -42,6 +42,7 @@ def test_efficientnet_model(self):
             "semnasnet_140",
             "mnasnet_small",
             "mobilenet_100",
+            # "mobilenet_100h",
             "mobilenet_125",
             "mobilenetv2_035",
             "mobilenetv2_050",

diff --git a/fl_common/tests/tests_models_mobilenet.py b/fl_common/tests/tests_models_mobilenet.py
@@ -51,6 +51,8 @@ def test_get_mobilenet_model(self):
             "mobilenetv4_hybrid_large",
             "mobilenetv4_conv_aa_medium",
             "mobilenetv4_conv_blur_medium",
+            # "mobilenetv4_conv_aa_large",
+            # "mobilenetv4_conv_blur_large",
             "mobilenetv4_hybrid_medium_075",
             "mobilenetv4_hybrid_large_075",
             # "repghostnetv2_conv_small",