diff --git a/fl_common/dlcl_getfamilly.py b/fl_common/dlcl_getfamilly.py index 260ccd24..6e83404c 100644 --- a/fl_common/dlcl_getfamilly.py +++ b/fl_common/dlcl_getfamilly.py @@ -718,6 +718,8 @@ def get_family_model_h(model_type, num_classes): "hiera_base_plus_224": get_hiera_model, "hiera_large_224": get_hiera_model, "hiera_huge_224": get_hiera_model, + "hiera_small_abswin_256": get_hiera_model, + "hiera_base_abswin_256": get_hiera_model, "hrnet_w18_small": get_hrnet_model, "hrnet_w18_small_v2": get_hrnet_model, "hrnet_w18": get_hrnet_model, @@ -952,7 +954,6 @@ def get_family_model_m(model_type, num_classes): "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/hiera.py b/fl_common/models/hiera.py index b2c29883..8b524e7a 100644 --- a/fl_common/models/hiera.py +++ b/fl_common/models/hiera.py @@ -22,6 +22,8 @@ def get_hiera_model(hiera_type, num_classes): "hiera_base_plus_224", "hiera_large_224", "hiera_huge_224", + 'hiera_small_abswin_256', + 'hiera_base_abswin_256', } if hiera_type not in valid_hiera_types: diff --git a/fl_common/models/machine_learning/__init__.py b/fl_common/models/machine_learning/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/fl_common/models/mobilenet.py b/fl_common/models/mobilenet.py index 07d15d02..929576a2 100644 --- a/fl_common/models/mobilenet.py +++ b/fl_common/models/mobilenet.py @@ -70,7 +70,6 @@ def get_mobilenet_model(mobilenet_type, num_classes): "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/models/segmentation/__init__.py b/fl_common/models/segmentation/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/fl_common/models/segmentation/snp.py b/fl_common/models/segmentation/snp.py new file mode 100644 index 00000000..e5431775 --- /dev/null +++ b/fl_common/models/segmentation/snp.py @@ -0,0 +1,586 @@ +import segmentation_models_pytorch as smp +from segmentation_models_pytorch.encoders import get_preprocessing_fn + +encoders = [ + {"model": "tu-SelecSls42", "dilation": "False"}, + {"model": "tu-SelecSls42b", "dilation": "False"}, + {"model": "tu-SelecSls60", "dilation": "False"}, + {"model": "tu-SelecSls60b", "dilation": "False"}, + {"model": "tu-SelecSls84", "dilation": "False"}, + {"model": "tu-bat_resnext26ts", "dilation": "True"}, + {"model": "tu-botnet26t_256", "dilation": "True"}, + {"model": "tu-botnet50ts_256", "dilation": "True"}, + {"model": "tu-coatnet_0_224", "dilation": "False"}, + {"model": "tu-coatnet_0_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_1_224", "dilation": "False"}, + {"model": "tu-coatnet_1_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_2_224", "dilation": "False"}, + {"model": "tu-coatnet_2_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_3_224", "dilation": "False"}, + {"model": "tu-coatnet_3_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_4_224", "dilation": "False"}, + {"model": "tu-coatnet_5_224", "dilation": "False"}, + {"model": "tu-coatnet_bn_0_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_nano_cc_224", "dilation": "False"}, + {"model": "tu-coatnet_nano_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_pico_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_0_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_1_rw2_224", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_1_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_2_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_2_rw_384", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_3_rw_224", "dilation": "False"}, + {"model": "tu-coatnet_rmlp_nano_rw_224", "dilation": "False"}, + {"model": "tu-coatnext_nano_rw_224", "dilation": "False"}, + {"model": "tu-cs3darknet_focus_l", "dilation": "True"}, + {"model": "tu-cs3darknet_focus_m", "dilation": "True"}, + {"model": "tu-cs3darknet_focus_s", "dilation": "True"}, + {"model": "tu-cs3darknet_focus_x", "dilation": "True"}, + {"model": "tu-cs3darknet_l", "dilation": "True"}, + {"model": "tu-cs3darknet_m", "dilation": "True"}, + {"model": "tu-cs3darknet_s", "dilation": "True"}, + {"model": "tu-cs3darknet_x", "dilation": "True"}, + {"model": "tu-cs3edgenet_x", "dilation": "True"}, + {"model": "tu-cs3se_edgenet_x", "dilation": "True"}, + {"model": "tu-cs3sedarknet_l", "dilation": "True"}, + {"model": "tu-cs3sedarknet_x", "dilation": "True"}, + {"model": "tu-cs3sedarknet_xdw", "dilation": "True"}, + {"model": "tu-cspresnet50", "dilation": "True"}, + {"model": "tu-cspresnet50d", "dilation": "True"}, + {"model": "tu-cspresnet50w", "dilation": "True"}, + {"model": "tu-cspresnext50", "dilation": "True"}, + {"model": "tu-densenet121", "dilation": "False"}, + {"model": "tu-densenet161", "dilation": "False"}, + {"model": "tu-densenet169", "dilation": "False"}, + {"model": "tu-densenet201", "dilation": "False"}, + {"model": "tu-densenet264d", "dilation": "False"}, + {"model": "tu-densenetblur121d", "dilation": "False"}, + {"model": "tu-dla102", "dilation": "False"}, + {"model": "tu-dla102x", "dilation": "False"}, + {"model": "tu-dla102x2", "dilation": "False"}, + {"model": "tu-dla169", "dilation": "False"}, + {"model": "tu-dla34", "dilation": "False"}, + {"model": "tu-dla46_c", "dilation": "False"}, + {"model": "tu-dla46x_c", "dilation": "False"}, + {"model": "tu-dla60", "dilation": "False"}, + {"model": "tu-dla60_res2net", "dilation": "False"}, + {"model": "tu-dla60_res2next", "dilation": "False"}, + {"model": "tu-dla60x", "dilation": "False"}, + {"model": "tu-dla60x_c", "dilation": "False"}, + {"model": "tu-dm_nfnet_f0", "dilation": "True"}, + {"model": "tu-dm_nfnet_f1", "dilation": "True"}, + {"model": "tu-dm_nfnet_f2", "dilation": "True"}, + {"model": "tu-dm_nfnet_f3", "dilation": "True"}, + {"model": "tu-dm_nfnet_f4", "dilation": "True"}, + {"model": "tu-dm_nfnet_f5", "dilation": "True"}, + {"model": "tu-dm_nfnet_f6", "dilation": "True"}, + {"model": "tu-dpn107", "dilation": "False"}, + {"model": "tu-dpn131", "dilation": "False"}, + {"model": "tu-dpn48b", "dilation": "False"}, + {"model": "tu-dpn68", "dilation": "False"}, + {"model": "tu-dpn68b", "dilation": "False"}, + {"model": "tu-dpn92", "dilation": "False"}, + {"model": "tu-dpn98", "dilation": "False"}, + {"model": "tu-eca_botnext26ts_256", "dilation": "True"}, + {"model": "tu-eca_halonext26ts", "dilation": "True"}, + {"model": "tu-eca_nfnet_l0", "dilation": "True"}, + {"model": "tu-eca_nfnet_l1", "dilation": "True"}, + {"model": "tu-eca_nfnet_l2", "dilation": "True"}, + {"model": "tu-eca_nfnet_l3", "dilation": "True"}, + {"model": "tu-eca_resnet33ts", "dilation": "True"}, + {"model": "tu-eca_resnext26ts", "dilation": "True"}, + {"model": "tu-eca_vovnet39b", "dilation": "False"}, + {"model": "tu-ecaresnet101d", "dilation": "True"}, + {"model": "tu-ecaresnet101d_pruned", "dilation": "True"}, + {"model": "tu-ecaresnet200d", "dilation": "True"}, + {"model": "tu-ecaresnet269d", "dilation": "True"}, + {"model": "tu-ecaresnet26t", "dilation": "True"}, + {"model": "tu-ecaresnet50d", "dilation": "True"}, + {"model": "tu-ecaresnet50d_pruned", "dilation": "True"}, + {"model": "tu-ecaresnet50t", "dilation": "True"}, + {"model": "tu-ecaresnetlight", "dilation": "True"}, + {"model": "tu-ecaresnext26t_32x4d", "dilation": "True"}, + {"model": "tu-ecaresnext50t_32x4d", "dilation": "True"}, + {"model": "tu-efficientnet_b0", "dilation": "True"}, + {"model": "tu-efficientnet_b0_g16_evos", "dilation": "True"}, + {"model": "tu-efficientnet_b0_g8_gn", "dilation": "True"}, + {"model": "tu-efficientnet_b0_gn", "dilation": "True"}, + {"model": "tu-efficientnet_b1", "dilation": "True"}, + {"model": "tu-efficientnet_b1_pruned", "dilation": "True"}, + {"model": "tu-efficientnet_b2", "dilation": "True"}, + {"model": "tu-efficientnet_b2_pruned", "dilation": "True"}, + {"model": "tu-efficientnet_b2a", "dilation": "True"}, + {"model": "tu-efficientnet_b3", "dilation": "True"}, + {"model": "tu-efficientnet_b3_g8_gn", "dilation": "True"}, + {"model": "tu-efficientnet_b3_gn", "dilation": "True"}, + {"model": "tu-efficientnet_b3_pruned", "dilation": "True"}, + {"model": "tu-efficientnet_b3a", "dilation": "True"}, + {"model": "tu-efficientnet_b4", "dilation": "True"}, + {"model": "tu-efficientnet_b5", "dilation": "True"}, + {"model": "tu-efficientnet_b6", "dilation": "True"}, + {"model": "tu-efficientnet_b7", "dilation": "True"}, + {"model": "tu-efficientnet_b8", "dilation": "True"}, + {"model": "tu-efficientnet_cc_b0_4e", "dilation": "True"}, + {"model": "tu-efficientnet_cc_b0_8e", "dilation": "True"}, + {"model": "tu-efficientnet_cc_b1_8e", "dilation": "True"}, + {"model": "tu-efficientnet_el", "dilation": "True"}, + {"model": "tu-efficientnet_el_pruned", "dilation": "True"}, + {"model": "tu-efficientnet_em", "dilation": "True"}, + {"model": "tu-efficientnet_es", "dilation": "True"}, + {"model": "tu-efficientnet_es_pruned", "dilation": "True"}, + {"model": "tu-efficientnet_l2", "dilation": "True"}, + {"model": "tu-efficientnet_lite0", "dilation": "True"}, + {"model": "tu-efficientnet_lite1", "dilation": "True"}, + {"model": "tu-efficientnet_lite2", "dilation": "True"}, + {"model": "tu-efficientnet_lite3", "dilation": "True"}, + {"model": "tu-efficientnet_lite4", "dilation": "True"}, + {"model": "tu-efficientnetv2_l", "dilation": "True"}, + {"model": "tu-efficientnetv2_m", "dilation": "True"}, + {"model": "tu-efficientnetv2_rw_m", "dilation": "True"}, + {"model": "tu-efficientnetv2_rw_s", "dilation": "True"}, + {"model": "tu-efficientnetv2_rw_t", "dilation": "True"}, + {"model": "tu-efficientnetv2_s", "dilation": "True"}, + {"model": "tu-efficientnetv2_xl", "dilation": "True"}, + {"model": "tu-ese_vovnet19b_dw", "dilation": "False"}, + {"model": "tu-ese_vovnet19b_slim", "dilation": "False"}, + {"model": "tu-ese_vovnet19b_slim_dw", "dilation": "False"}, + {"model": "tu-ese_vovnet39b", "dilation": "False"}, + {"model": "tu-ese_vovnet39b_evos", "dilation": "False"}, + {"model": "tu-ese_vovnet57b", "dilation": "False"}, + {"model": "tu-ese_vovnet99b", "dilation": "False"}, + {"model": "tu-fbnetc_100", "dilation": "True"}, + {"model": "tu-fbnetv3_b", "dilation": "True"}, + {"model": "tu-fbnetv3_d", "dilation": "True"}, + {"model": "tu-fbnetv3_g", "dilation": "True"}, + {"model": "tu-gc_efficientnetv2_rw_t", "dilation": "True"}, + {"model": "tu-gcresnet33ts", "dilation": "True"}, + {"model": "tu-gcresnet50t", "dilation": "True"}, + {"model": "tu-gcresnext26ts", "dilation": "True"}, + {"model": "tu-gcresnext50ts", "dilation": "True"}, + {"model": "tu-gernet_l", "dilation": "True"}, + {"model": "tu-gernet_m", "dilation": "True"}, + {"model": "tu-gernet_s", "dilation": "True"}, + {"model": "tu-ghostnet_050", "dilation": "False"}, + {"model": "tu-ghostnet_100", "dilation": "False"}, + {"model": "tu-ghostnet_130", "dilation": "False"}, + {"model": "tu-halo2botnet50ts_256", "dilation": "True"}, + {"model": "tu-halonet26t", "dilation": "True"}, + {"model": "tu-halonet50ts", "dilation": "True"}, + {"model": "tu-halonet_h1", "dilation": "True"}, + {"model": "tu-haloregnetz_b", "dilation": "True"}, + {"model": "tu-hardcorenas_a", "dilation": "True"}, + {"model": "tu-hardcorenas_b", "dilation": "True"}, + {"model": "tu-hardcorenas_c", "dilation": "True"}, + {"model": "tu-hardcorenas_d", "dilation": "True"}, + {"model": "tu-hardcorenas_e", "dilation": "True"}, + {"model": "tu-hardcorenas_f", "dilation": "True"}, + {"model": "tu-hrnet_w18", "dilation": "False"}, + {"model": "tu-hrnet_w18_slim", "dilation": "False"}, + {"model": "tu-hrnet_w18_small_v2", "dilation": "False"}, + {"model": "tu-hrnet_w18_ssld", "dilation": "False"}, + {"model": "tu-hrnet_w30", "dilation": "False"}, + {"model": "tu-hrnet_w32", "dilation": "False"}, + {"model": "tu-hrnet_w40", "dilation": "False"}, + {"model": "tu-hrnet_w44", "dilation": "False"}, + {"model": "tu-hrnet_w48", "dilation": "False"}, + {"model": "tu-hrnet_w48_ssld", "dilation": "False"}, + {"model": "tu-hrnet_w64", "dilation": "False"}, + {"model": "tu-inception_resnet_v2", "dilation": "False"}, + {"model": "tu-inception_v3", "dilation": "False"}, + {"model": "tu-inception_v4", "dilation": "False"}, + {"model": "tu-lambda_resnet26rpt_256", "dilation": "True"}, + {"model": "tu-lambda_resnet26t", "dilation": "True"}, + {"model": "tu-lambda_resnet50ts", "dilation": "True"}, + {"model": "tu-lamhalobotnet50ts_256", "dilation": "True"}, + {"model": "tu-lcnet_035", "dilation": "True"}, + {"model": "tu-lcnet_050", "dilation": "True"}, + {"model": "tu-lcnet_075", "dilation": "True"}, + {"model": "tu-lcnet_100", "dilation": "True"}, + {"model": "tu-lcnet_150", "dilation": "True"}, + {"model": "tu-legacy_senet154", "dilation": "False"}, + {"model": "tu-legacy_seresnet101", "dilation": "False"}, + {"model": "tu-legacy_seresnet152", "dilation": "False"}, + {"model": "tu-legacy_seresnet18", "dilation": "False"}, + {"model": "tu-legacy_seresnet34", "dilation": "False"}, + {"model": "tu-legacy_seresnet50", "dilation": "False"}, + {"model": "tu-legacy_seresnext101_32x4d", "dilation": "False"}, + {"model": "tu-legacy_seresnext26_32x4d", "dilation": "False"}, + {"model": "tu-legacy_seresnext50_32x4d", "dilation": "False"}, + {"model": "tu-legacy_xception", "dilation": "False"}, + {"model": "tu-maxvit_base_tf_224", "dilation": "False"}, + {"model": "tu-maxvit_base_tf_384", "dilation": "False"}, + {"model": "tu-maxvit_base_tf_512", "dilation": "False"}, + {"model": "tu-maxvit_large_tf_224", "dilation": "False"}, + {"model": "tu-maxvit_large_tf_384", "dilation": "False"}, + {"model": "tu-maxvit_large_tf_512", "dilation": "False"}, + {"model": "tu-maxvit_nano_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_pico_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_base_rw_224", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_base_rw_384", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_nano_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_pico_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_small_rw_224", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_small_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_rmlp_tiny_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_small_tf_224", "dilation": "False"}, + {"model": "tu-maxvit_small_tf_384", "dilation": "False"}, + {"model": "tu-maxvit_small_tf_512", "dilation": "False"}, + {"model": "tu-maxvit_tiny_pm_256", "dilation": "False"}, + {"model": "tu-maxvit_tiny_rw_224", "dilation": "False"}, + {"model": "tu-maxvit_tiny_rw_256", "dilation": "False"}, + {"model": "tu-maxvit_tiny_tf_224", "dilation": "False"}, + {"model": "tu-maxvit_tiny_tf_384", "dilation": "False"}, + {"model": "tu-maxvit_tiny_tf_512", "dilation": "False"}, + {"model": "tu-maxvit_xlarge_tf_224", "dilation": "False"}, + {"model": "tu-maxvit_xlarge_tf_384", "dilation": "False"}, + {"model": "tu-maxvit_xlarge_tf_512", "dilation": "False"}, + {"model": "tu-maxxvit_rmlp_nano_rw_256", "dilation": "False"}, + {"model": "tu-maxxvit_rmlp_small_rw_256", "dilation": "False"}, + {"model": "tu-maxxvit_rmlp_tiny_rw_256", "dilation": "False"}, + {"model": "tu-maxxvitv2_nano_rw_256", "dilation": "False"}, + {"model": "tu-maxxvitv2_rmlp_base_rw_224", "dilation": "False"}, + {"model": "tu-maxxvitv2_rmlp_base_rw_384", "dilation": "False"}, + {"model": "tu-maxxvitv2_rmlp_large_rw_224", "dilation": "False"}, + {"model": "tu-mixnet_l", "dilation": "True"}, + {"model": "tu-mixnet_m", "dilation": "True"}, + {"model": "tu-mixnet_s", "dilation": "True"}, + {"model": "tu-mixnet_xl", "dilation": "True"}, + {"model": "tu-mixnet_xxl", "dilation": "True"}, + {"model": "tu-mnasnet_050", "dilation": "True"}, + {"model": "tu-mnasnet_075", "dilation": "True"}, + {"model": "tu-mnasnet_100", "dilation": "True"}, + {"model": "tu-mnasnet_140", "dilation": "True"}, + {"model": "tu-mnasnet_a1", "dilation": "True"}, + {"model": "tu-mnasnet_b1", "dilation": "True"}, + {"model": "tu-mnasnet_small", "dilation": "True"}, + {"model": "tu-mobilenetv2_035", "dilation": "True"}, + {"model": "tu-mobilenetv2_050", "dilation": "True"}, + {"model": "tu-mobilenetv2_075", "dilation": "True"}, + {"model": "tu-mobilenetv2_100", "dilation": "True"}, + {"model": "tu-mobilenetv2_110d", "dilation": "True"}, + {"model": "tu-mobilenetv2_120d", "dilation": "True"}, + {"model": "tu-mobilenetv2_140", "dilation": "True"}, + {"model": "tu-mobilenetv3_large_075", "dilation": "True"}, + {"model": "tu-mobilenetv3_large_100", "dilation": "True"}, + {"model": "tu-mobilenetv3_rw", "dilation": "True"}, + {"model": "tu-mobilenetv3_small_050", "dilation": "True"}, + {"model": "tu-mobilenetv3_small_075", "dilation": "True"}, + {"model": "tu-mobilenetv3_small_100", "dilation": "True"}, + {"model": "tu-mobilevit_s", "dilation": "True"}, + {"model": "tu-mobilevit_xs", "dilation": "True"}, + {"model": "tu-mobilevit_xxs", "dilation": "True"}, + {"model": "tu-mobilevitv2_050", "dilation": "True"}, + {"model": "tu-mobilevitv2_075", "dilation": "True"}, + {"model": "tu-mobilevitv2_100", "dilation": "True"}, + {"model": "tu-mobilevitv2_125", "dilation": "True"}, + {"model": "tu-mobilevitv2_150", "dilation": "True"}, + {"model": "tu-mobilevitv2_175", "dilation": "True"}, + {"model": "tu-mobilevitv2_200", "dilation": "True"}, + {"model": "tu-nasnetalarge", "dilation": "False"}, + {"model": "tu-nf_ecaresnet101", "dilation": "True"}, + {"model": "tu-nf_ecaresnet26", "dilation": "True"}, + {"model": "tu-nf_ecaresnet50", "dilation": "True"}, + {"model": "tu-nf_regnet_b0", "dilation": "True"}, + {"model": "tu-nf_regnet_b1", "dilation": "True"}, + {"model": "tu-nf_regnet_b2", "dilation": "True"}, + {"model": "tu-nf_regnet_b3", "dilation": "True"}, + {"model": "tu-nf_regnet_b4", "dilation": "True"}, + {"model": "tu-nf_regnet_b5", "dilation": "True"}, + {"model": "tu-nf_resnet101", "dilation": "True"}, + {"model": "tu-nf_resnet26", "dilation": "True"}, + {"model": "tu-nf_resnet50", "dilation": "True"}, + {"model": "tu-nf_seresnet101", "dilation": "True"}, + {"model": "tu-nf_seresnet26", "dilation": "True"}, + {"model": "tu-nf_seresnet50", "dilation": "True"}, + {"model": "tu-nfnet_f0", "dilation": "True"}, + {"model": "tu-nfnet_f1", "dilation": "True"}, + {"model": "tu-nfnet_f2", "dilation": "True"}, + {"model": "tu-nfnet_f3", "dilation": "True"}, + {"model": "tu-nfnet_f4", "dilation": "True"}, + {"model": "tu-nfnet_f5", "dilation": "True"}, + {"model": "tu-nfnet_f6", "dilation": "True"}, + {"model": "tu-nfnet_f7", "dilation": "True"}, + {"model": "tu-nfnet_l0", "dilation": "True"}, + {"model": "tu-pnasnet5large", "dilation": "False"}, + {"model": "tu-regnetv_040", "dilation": "True"}, + {"model": "tu-regnetv_064", "dilation": "True"}, + {"model": "tu-regnetx_002", "dilation": "True"}, + {"model": "tu-regnetx_004", "dilation": "True"}, + {"model": "tu-regnetx_004_tv", "dilation": "True"}, + {"model": "tu-regnetx_006", "dilation": "True"}, + {"model": "tu-regnetx_008", "dilation": "True"}, + {"model": "tu-regnetx_016", "dilation": "True"}, + {"model": "tu-regnetx_032", "dilation": "True"}, + {"model": "tu-regnetx_040", "dilation": "True"}, + {"model": "tu-regnetx_064", "dilation": "True"}, + {"model": "tu-regnetx_080", "dilation": "True"}, + {"model": "tu-regnetx_120", "dilation": "True"}, + {"model": "tu-regnetx_160", "dilation": "True"}, + {"model": "tu-regnetx_320", "dilation": "True"}, + {"model": "tu-regnety_002", "dilation": "True"}, + {"model": "tu-regnety_004", "dilation": "True"}, + {"model": "tu-regnety_006", "dilation": "True"}, + {"model": "tu-regnety_008", "dilation": "True"}, + {"model": "tu-regnety_008_tv", "dilation": "True"}, + {"model": "tu-regnety_016", "dilation": "True"}, + {"model": "tu-regnety_032", "dilation": "True"}, + {"model": "tu-regnety_040", "dilation": "True"}, + {"model": "tu-regnety_040_sgn", "dilation": "True"}, + {"model": "tu-regnety_064", "dilation": "True"}, + {"model": "tu-regnety_080", "dilation": "True"}, + {"model": "tu-regnety_080_tv", "dilation": "True"}, + {"model": "tu-regnety_120", "dilation": "True"}, + {"model": "tu-regnety_1280", "dilation": "True"}, + {"model": "tu-regnety_160", "dilation": "True"}, + {"model": "tu-regnety_2560", "dilation": "True"}, + {"model": "tu-regnety_320", "dilation": "True"}, + {"model": "tu-regnety_640", "dilation": "True"}, + {"model": "tu-regnetz_005", "dilation": "True"}, + {"model": "tu-regnetz_040", "dilation": "True"}, + {"model": "tu-regnetz_040_h", "dilation": "True"}, + {"model": "tu-regnetz_b16", "dilation": "True"}, + {"model": "tu-regnetz_b16_evos", "dilation": "True"}, + {"model": "tu-regnetz_c16", "dilation": "True"}, + {"model": "tu-regnetz_c16_evos", "dilation": "True"}, + {"model": "tu-regnetz_d32", "dilation": "True"}, + {"model": "tu-regnetz_d8", "dilation": "True"}, + {"model": "tu-regnetz_d8_evos", "dilation": "True"}, + {"model": "tu-regnetz_e8", "dilation": "True"}, + {"model": "tu-repvgg_a2", "dilation": "True"}, + {"model": "tu-repvgg_b0", "dilation": "True"}, + {"model": "tu-repvgg_b1", "dilation": "True"}, + {"model": "tu-repvgg_b1g4", "dilation": "True"}, + {"model": "tu-repvgg_b2", "dilation": "True"}, + {"model": "tu-repvgg_b2g4", "dilation": "True"}, + {"model": "tu-repvgg_b3", "dilation": "True"}, + {"model": "tu-repvgg_b3g4", "dilation": "True"}, + {"model": "tu-res2net101_26w_4s", "dilation": "True"}, + {"model": "tu-res2net101d", "dilation": "True"}, + {"model": "tu-res2net50_14w_8s", "dilation": "True"}, + {"model": "tu-res2net50_26w_4s", "dilation": "True"}, + {"model": "tu-res2net50_26w_6s", "dilation": "True"}, + {"model": "tu-res2net50_26w_8s", "dilation": "True"}, + {"model": "tu-res2net50_48w_2s", "dilation": "True"}, + {"model": "tu-res2net50d", "dilation": "True"}, + {"model": "tu-res2next50", "dilation": "True"}, + {"model": "tu-resnest101e", "dilation": "True"}, + {"model": "tu-resnest14d", "dilation": "True"}, + {"model": "tu-resnest200e", "dilation": "True"}, + {"model": "tu-resnest269e", "dilation": "True"}, + {"model": "tu-resnest26d", "dilation": "True"}, + {"model": "tu-resnest50d", "dilation": "True"}, + {"model": "tu-resnest50d_1s4x24d", "dilation": "True"}, + {"model": "tu-resnest50d_4s2x40d", "dilation": "True"}, + {"model": "tu-resnet101", "dilation": "True"}, + {"model": "tu-resnet101c", "dilation": "True"}, + {"model": "tu-resnet101d", "dilation": "True"}, + {"model": "tu-resnet101s", "dilation": "True"}, + {"model": "tu-resnet10t", "dilation": "True"}, + {"model": "tu-resnet14t", "dilation": "True"}, + {"model": "tu-resnet152", "dilation": "True"}, + {"model": "tu-resnet152c", "dilation": "True"}, + {"model": "tu-resnet152d", "dilation": "True"}, + {"model": "tu-resnet152s", "dilation": "True"}, + {"model": "tu-resnet18", "dilation": "True"}, + {"model": "tu-resnet18d", "dilation": "True"}, + {"model": "tu-resnet200", "dilation": "True"}, + {"model": "tu-resnet200d", "dilation": "True"}, + {"model": "tu-resnet26", "dilation": "True"}, + {"model": "tu-resnet26d", "dilation": "True"}, + {"model": "tu-resnet26t", "dilation": "True"}, + {"model": "tu-resnet32ts", "dilation": "True"}, + {"model": "tu-resnet33ts", "dilation": "True"}, + {"model": "tu-resnet34", "dilation": "True"}, + {"model": "tu-resnet34d", "dilation": "True"}, + {"model": "tu-resnet50", "dilation": "True"}, + {"model": "tu-resnet50_gn", "dilation": "True"}, + {"model": "tu-resnet50c", "dilation": "True"}, + {"model": "tu-resnet50d", "dilation": "True"}, + {"model": "tu-resnet50s", "dilation": "True"}, + {"model": "tu-resnet50t", "dilation": "True"}, + {"model": "tu-resnet51q", "dilation": "True"}, + {"model": "tu-resnet61q", "dilation": "True"}, + {"model": "tu-resnetaa101d", "dilation": "True"}, + {"model": "tu-resnetaa34d", "dilation": "True"}, + {"model": "tu-resnetaa50", "dilation": "True"}, + {"model": "tu-resnetaa50d", "dilation": "True"}, + {"model": "tu-resnetblur101d", "dilation": "True"}, + {"model": "tu-resnetblur18", "dilation": "True"}, + {"model": "tu-resnetblur50", "dilation": "True"}, + {"model": "tu-resnetblur50d", "dilation": "True"}, + {"model": "tu-resnetrs101", "dilation": "True"}, + {"model": "tu-resnetrs152", "dilation": "True"}, + {"model": "tu-resnetrs200", "dilation": "True"}, + {"model": "tu-resnetrs270", "dilation": "True"}, + {"model": "tu-resnetrs350", "dilation": "True"}, + {"model": "tu-resnetrs420", "dilation": "True"}, + {"model": "tu-resnetrs50", "dilation": "True"}, + {"model": "tu-resnetv2_101", "dilation": "True"}, + {"model": "tu-resnetv2_101d", "dilation": "True"}, + {"model": "tu-resnetv2_101x1_bit", "dilation": "True"}, + {"model": "tu-resnetv2_101x3_bit", "dilation": "True"}, + {"model": "tu-resnetv2_152", "dilation": "True"}, + {"model": "tu-resnetv2_152d", "dilation": "True"}, + {"model": "tu-resnetv2_152x2_bit", "dilation": "True"}, + {"model": "tu-resnetv2_152x4_bit", "dilation": "True"}, + {"model": "tu-resnetv2_50", "dilation": "True"}, + {"model": "tu-resnetv2_50d", "dilation": "True"}, + {"model": "tu-resnetv2_50d_evos", "dilation": "True"}, + {"model": "tu-resnetv2_50d_frn", "dilation": "True"}, + {"model": "tu-resnetv2_50d_gn", "dilation": "True"}, + {"model": "tu-resnetv2_50t", "dilation": "True"}, + {"model": "tu-resnetv2_50x1_bit", "dilation": "True"}, + {"model": "tu-resnetv2_50x3_bit", "dilation": "True"}, + {"model": "tu-resnext101_32x16d", "dilation": "True"}, + {"model": "tu-resnext101_32x32d", "dilation": "True"}, + {"model": "tu-resnext101_32x4d", "dilation": "True"}, + {"model": "tu-resnext101_32x8d", "dilation": "True"}, + {"model": "tu-resnext101_64x4d", "dilation": "True"}, + {"model": "tu-resnext26ts", "dilation": "True"}, + {"model": "tu-resnext50_32x4d", "dilation": "True"}, + {"model": "tu-resnext50d_32x4d", "dilation": "True"}, + {"model": "tu-rexnet_100", "dilation": "True"}, + {"model": "tu-rexnet_130", "dilation": "True"}, + {"model": "tu-rexnet_150", "dilation": "True"}, + {"model": "tu-rexnet_200", "dilation": "True"}, + {"model": "tu-rexnet_300", "dilation": "True"}, + {"model": "tu-rexnetr_100", "dilation": "True"}, + {"model": "tu-rexnetr_130", "dilation": "True"}, + {"model": "tu-rexnetr_150", "dilation": "True"}, + {"model": "tu-rexnetr_200", "dilation": "True"}, + {"model": "tu-rexnetr_300", "dilation": "True"}, + {"model": "tu-sebotnet33ts_256", "dilation": "True"}, + {"model": "tu-sehalonet33ts", "dilation": "True"}, + {"model": "tu-semnasnet_050", "dilation": "True"}, + {"model": "tu-semnasnet_075", "dilation": "True"}, + {"model": "tu-semnasnet_100", "dilation": "True"}, + {"model": "tu-semnasnet_140", "dilation": "True"}, + {"model": "tu-senet154", "dilation": "True"}, + {"model": "tu-seresnet101", "dilation": "True"}, + {"model": "tu-seresnet152", "dilation": "True"}, + {"model": "tu-seresnet152d", "dilation": "True"}, + {"model": "tu-seresnet18", "dilation": "True"}, + {"model": "tu-seresnet200d", "dilation": "True"}, + {"model": "tu-seresnet269d", "dilation": "True"}, + {"model": "tu-seresnet33ts", "dilation": "True"}, + {"model": "tu-seresnet34", "dilation": "True"}, + {"model": "tu-seresnet50", "dilation": "True"}, + {"model": "tu-seresnet50t", "dilation": "True"}, + {"model": "tu-seresnext101_32x4d", "dilation": "True"}, + {"model": "tu-seresnext101_32x8d", "dilation": "True"}, + {"model": "tu-seresnext101_64x4d", "dilation": "True"}, + {"model": "tu-seresnext101d_32x8d", "dilation": "True"}, + {"model": "tu-seresnext26d_32x4d", "dilation": "True"}, + {"model": "tu-seresnext26t_32x4d", "dilation": "True"}, + {"model": "tu-sseresnext26tn_32x4d", "dilation": "True"}, + {"model": "tu-seresnext26ts", "dilation": "True"}, + {"model": "tu-seresnext50_32x4d", "dilation": "True"}, + {"model": "tu-seresnextaa101d_32x8d", "dilation": "True"}, + {"model": "tu-skresnet18", "dilation": "True"}, + {"model": "tu-skresnet34", "dilation": "True"}, + {"model": "tu-skresnet50", "dilation": "True"}, + {"model": "tu-skresnet50d", "dilation": "True"}, + {"model": "tu-skresnext50_32x4d", "dilation": "True"}, + {"model": "tu-spnasnet_100", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b0", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b1", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b2", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b3", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b4", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b5", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b6", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b7", "dilation": "True"}, + {"model": "tu-tf_efficientnet_b8", "dilation": "True"}, + {"model": "tu-tf_efficientnet_cc_b0_4e", "dilation": "True"}, + {"model": "tu-tf_efficientnet_cc_b0_8e", "dilation": "True"}, + {"model": "tu-tf_efficientnet_cc_b1_8e", "dilation": "True"}, + {"model": "tu-tf_efficientnet_el", "dilation": "True"}, + {"model": "tu-tf_efficientnet_em", "dilation": "True"}, + {"model": "tu-tf_efficientnet_es", "dilation": "True"}, + {"model": "tu-tf_efficientnet_l2", "dilation": "True"}, + {"model": "tu-tf_efficientnet_lite0", "dilation": "True"}, + {"model": "tu-tf_efficientnet_lite1", "dilation": "True"}, + {"model": "tu-tf_efficientnet_lite2", "dilation": "True"}, + {"model": "tu-tf_efficientnet_lite3", "dilation": "True"}, + {"model": "tu-tf_efficientnet_lite4", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_b0", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_b1", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_b2", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_b3", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_l", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_m", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_s", "dilation": "True"}, + {"model": "tu-tf_efficientnetv2_xl", "dilation": "True"}, + {"model": "tu-tf_mixnet_l", "dilation": "True"}, + {"model": "tu-tf_mixnet_m", "dilation": "True"}, + {"model": "tu-tf_mixnet_s", "dilation": "True"}, + {"model": "tu-tf_mobilenetv3_large_075", "dilation": "True"}, + {"model": "tu-tf_mobilenetv3_large_100", "dilation": "True"}, + {"model": "tu-tf_mobilenetv3_large_minimal_100", "dilation": "True"}, + {"model": "tu-tf_mobilenetv3_small_075", "dilation": "True"}, + {"model": "tu-tf_mobilenetv3_small_100", "dilation": "True"}, + {"model": "tu-tf_mobilenetv3_small_minimal_100", "dilation": "True"}, + {"model": "tu-tinynet_a", "dilation": "True"}, + {"model": "tu-tinynet_b", "dilation": "True"}, + {"model": "tu-tinynet_c", "dilation": "True"}, + {"model": "tu-tinynet_d", "dilation": "True"}, + {"model": "tu-tinynet_e", "dilation": "True"}, + {"model": "tu-vovnet39a", "dilation": "False"}, + {"model": "tu-vovnet57a", "dilation": "False"}, + {"model": "tu-wide_resnet101_2", "dilation": "True"}, + {"model": "tu-wide_resnet50_2", "dilation": "True"}, + {"model": "tu-xception41", "dilation": "True"}, + {"model": "tu-xception41p", "dilation": "True"}, + {"model": "tu-xception65", "dilation": "True"}, + {"model": "tu-xception65p", "dilation": "True"}, + {"model": "tu-xception71", "dilation": "True"}, +] + +architectures = [ + {"model": "unet", "dilation": "False"}, + {"model": "unetplusplus", "dilation": "False"}, + {"model": "fpn", "dilation": "False"}, + {"model": "pspnet", "dilation": "False"}, + {"model": "deeplabv3", "dilation": "False"}, + {"model": "deeplabv3plus", "dilation": "False"}, + {"model": "linknet", "dilation": "False"}, + {"model": "manet", "dilation": "False"}, + {"model": "pan", "dilation": "True"}, +] + + +def test_model_combinations(encoders, architectures): + """ + Test all valid combinations of segmentation model architectures and encoders. + + This function iterates through the provided lists of architectures and encoders, + creating a model for each valid combination where the dilation condition is met. + It prints a success message for each valid combination and captures any exceptions + that occur during model creation, printing an error message instead. + + Parameters: + encoders (list): A list of dictionaries where each dictionary contains: + - "model": The name of the encoder model. + - "dilation": A string indicating whether dilation is "True" or "False". + architectures (list): A list of dictionaries where each dictionary contains: + - "model": The name of the architecture model. + - "dilation": A string indicating whether dilation is "True" or "False". + """ + for arch in architectures: + for encoder in encoders: + if arch["dilation"] == encoder["dilation"]: + try: + model = smp.create_model( + arch=arch["model"], + encoder_name=encoder["model"], + encoder_weights="imagenet", + in_channels=3, + classes=2, + ) + preprocess_input = get_preprocessing_fn(encoder["model"], pretrained='imagenet') + print(f'Success: Created {arch["model"]} with {encoder["model"]}') + except Exception as e: + print(f'Failed: {arch["model"]} with {encoder["model"]} - {e}') + + +test_model_combinations(encoders, architectures) diff --git a/fl_common/tests/tests_models_hiera.py b/fl_common/tests/tests_models_hiera.py index fc165126..a6ee7af2 100644 --- a/fl_common/tests/tests_models_hiera.py +++ b/fl_common/tests/tests_models_hiera.py @@ -29,6 +29,8 @@ def test_known_hiera_types(self): "hiera_base_plus_224", "hiera_large_224", "hiera_huge_224", + "hiera_small_abswin_256", + "hiera_base_abswin_256", ] num_classes = 1000 # Assuming 1000 classes for the test diff --git a/fl_common/tests/tests_models_mobilenet.py b/fl_common/tests/tests_models_mobilenet.py index b65f79ae..9652b26b 100644 --- a/fl_common/tests/tests_models_mobilenet.py +++ b/fl_common/tests/tests_models_mobilenet.py @@ -52,7 +52,6 @@ def test_get_mobilenet_model(self): "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_server/server.py b/fl_server/server.py index 55dce9f9..a226936a 100644 --- a/fl_server/server.py +++ b/fl_server/server.py @@ -44,7 +44,7 @@ def register_user(node, email, password, name, institution, website): Returns: client: The registered client. """ - client = node.login(email=email, password=password) + client = login(login_node=node, login_email=email, login_password=password) client.register( name=name, email=email, @@ -93,19 +93,19 @@ def launch_and_register( return node, client -def login(node, login_email, login_password): +def login(login_node, login_email, login_password): """ Log in to the given node with the provided email and password. Args: - node: The node to log in to. + login_node: The node to log in to. login_email (str): The email to use for logging in. login_password (str): The password to use for logging in. Returns: client: The client returned after successful login. """ - return node.login( + return login_node.login( email=login_email, password=login_password, ) diff --git a/fl_server/tests/tests_server.py b/fl_server/tests/tests_server.py index ad033eeb..b07d3c74 100644 --- a/fl_server/tests/tests_server.py +++ b/fl_server/tests/tests_server.py @@ -49,12 +49,6 @@ def test_launch_node_main(self): """ Test the main node launch functionality. """ - ds_client = login( - "node_humani", - "janedoe@caltech.edu", - "abc123", - ) - data_subjects = self.client_humani.data_subject_registry.get_all() self.assertIsNotNone(data_subjects) @@ -82,6 +76,11 @@ def test_launch_node_main(self): ) self.client_humani.upload_dataset(dataset) + ds_client = login('node_humani', + "janedoe@caltech.edu", + "abc123", + ) + asset = ds_client.datasets[-1].assets["ages"] mock = asset.mock