diff --git a/pyproject.toml b/pyproject.toml index 763df56df5e3de39d8a788b46035dec8385f2640..2219b5c2e35b15dd70278545aeb4864901b1adc3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -186,9 +186,75 @@ indian_rs_f6 = "ptbench.configs.datasets.indian_RS.fold_6" indian_rs_f7 = "ptbench.configs.datasets.indian_RS.fold_7" indian_rs_f8 = "ptbench.configs.datasets.indian_RS.fold_8" indian_rs_f9 = "ptbench.configs.datasets.indian_RS.fold_9" -# TBX11K simplified dataset +# TBX11K simplified dataset split 1 (and cross-validation folds) tbx11k_simplified = "ptbench.configs.datasets.tbx11k_simplified.default" tbx11k_simplified_rgb = "ptbench.configs.datasets.tbx11k_simplified.rgb" +tbx11k_simplified_f0 = "ptbench.configs.datasets.tbx11k_simplified.fold_0" +tbx11k_simplified_f1 = "ptbench.configs.datasets.tbx11k_simplified.fold_1" +tbx11k_simplified_f2 = "ptbench.configs.datasets.tbx11k_simplified.fold_2" +tbx11k_simplified_f3 = "ptbench.configs.datasets.tbx11k_simplified.fold_3" +tbx11k_simplified_f4 = "ptbench.configs.datasets.tbx11k_simplified.fold_4" +tbx11k_simplified_f5 = "ptbench.configs.datasets.tbx11k_simplified.fold_5" +tbx11k_simplified_f6 = "ptbench.configs.datasets.tbx11k_simplified.fold_6" +tbx11k_simplified_f7 = "ptbench.configs.datasets.tbx11k_simplified.fold_7" +tbx11k_simplified_f8 = "ptbench.configs.datasets.tbx11k_simplified.fold_8" +tbx11k_simplified_f9 = "ptbench.configs.datasets.tbx11k_simplified.fold_9" +tbx11k_simplified_f0_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_0_rgb" +tbx11k_simplified_f1_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_1_rgb" +tbx11k_simplified_f2_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_2_rgb" +tbx11k_simplified_f3_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_3_rgb" +tbx11k_simplified_f4_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_4_rgb" +tbx11k_simplified_f5_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_5_rgb" +tbx11k_simplified_f6_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_6_rgb" +tbx11k_simplified_f7_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_7_rgb" +tbx11k_simplified_f8_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_8_rgb" +tbx11k_simplified_f9_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_9_rgb" +# TBX11K simplified dataset split 2 (and cross-validation folds) +tbx11k_simplified_v2 = "ptbench.configs.datasets.tbx11k_simplified_v2.default" +tbx11k_simplified_v2_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.rgb" +tbx11k_simplified_v2_f0 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_0" +tbx11k_simplified_v2_f1 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_1" +tbx11k_simplified_v2_f2 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_2" +tbx11k_simplified_v2_f3 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_3" +tbx11k_simplified_v2_f4 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_4" +tbx11k_simplified_v2_f5 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_5" +tbx11k_simplified_v2_f6 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_6" +tbx11k_simplified_v2_f7 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_7" +tbx11k_simplified_v2_f8 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_8" +tbx11k_simplified_v2_f9 = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_9" +tbx11k_simplified_v2_f0_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_0_rgb" +tbx11k_simplified_v2_f1_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_1_rgb" +tbx11k_simplified_v2_f2_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_2_rgb" +tbx11k_simplified_v2_f3_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_3_rgb" +tbx11k_simplified_v2_f4_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_4_rgb" +tbx11k_simplified_v2_f5_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_5_rgb" +tbx11k_simplified_v2_f6_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_6_rgb" +tbx11k_simplified_v2_f7_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_7_rgb" +tbx11k_simplified_v2_f8_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_8_rgb" +tbx11k_simplified_v2_f9_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fold_9_rgb" +# TBX11K simplified dataset split 3 (and cross-validation folds) +tbx11k_simplified_v3 = "ptbench.configs.datasets.tbx11k_simplified_v3.default" +tbx11k_simplified_v3_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.rgb" +tbx11k_simplified_v3_f0 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_0" +tbx11k_simplified_v3_f1 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_1" +tbx11k_simplified_v3_f2 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_2" +tbx11k_simplified_v3_f3 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_3" +tbx11k_simplified_v3_f4 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_4" +tbx11k_simplified_v3_f5 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_5" +tbx11k_simplified_v3_f6 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_6" +tbx11k_simplified_v3_f7 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_7" +tbx11k_simplified_v3_f8 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_8" +tbx11k_simplified_v3_f9 = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_9" +tbx11k_simplified_v3_f0_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_0_rgb" +tbx11k_simplified_v3_f1_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_1_rgb" +tbx11k_simplified_v3_f2_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_2_rgb" +tbx11k_simplified_v3_f3_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_3_rgb" +tbx11k_simplified_v3_f4_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_4_rgb" +tbx11k_simplified_v3_f5_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_5_rgb" +tbx11k_simplified_v3_f6_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_6_rgb" +tbx11k_simplified_v3_f7_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_7_rgb" +tbx11k_simplified_v3_f8_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_8_rgb" +tbx11k_simplified_v3_f9_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fold_9_rgb" # montgomery-shenzhen aggregated dataset mc_ch = "ptbench.configs.datasets.mc_ch.default" mc_ch_rgb = "ptbench.configs.datasets.mc_ch.rgb" @@ -261,6 +327,52 @@ mc_ch_in_rs_f6 = "ptbench.configs.datasets.mc_ch_in_RS.fold_6" mc_ch_in_rs_f7 = "ptbench.configs.datasets.mc_ch_in_RS.fold_7" mc_ch_in_rs_f8 = "ptbench.configs.datasets.mc_ch_in_RS.fold_8" mc_ch_in_rs_f9 = "ptbench.configs.datasets.mc_ch_in_RS.fold_9" +# montgomery-shenzhen-indian-tbx11k aggregated dataset +mc_ch_in_11k = "ptbench.configs.datasets.mc_ch_in_11k.default" +mc_ch_in_11k_rgb = "ptbench.configs.datasets.mc_ch_in_11k.rgb" +mc_ch_in_11k_f0 = "ptbench.configs.datasets.mc_ch_in_11k.fold_0" +mc_ch_in_11k_f1 = "ptbench.configs.datasets.mc_ch_in_11k.fold_1" +mc_ch_in_11k_f2 = "ptbench.configs.datasets.mc_ch_in_11k.fold_2" +mc_ch_in_11k_f3 = "ptbench.configs.datasets.mc_ch_in_11k.fold_3" +mc_ch_in_11k_f4 = "ptbench.configs.datasets.mc_ch_in_11k.fold_4" +mc_ch_in_11k_f5 = "ptbench.configs.datasets.mc_ch_in_11k.fold_5" +mc_ch_in_11k_f6 = "ptbench.configs.datasets.mc_ch_in_11k.fold_6" +mc_ch_in_11k_f7 = "ptbench.configs.datasets.mc_ch_in_11k.fold_7" +mc_ch_in_11k_f8 = "ptbench.configs.datasets.mc_ch_in_11k.fold_8" +mc_ch_in_11k_f9 = "ptbench.configs.datasets.mc_ch_in_11k.fold_9" +mc_ch_in_11k_f0_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_0_rgb" +mc_ch_in_11k_f1_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_1_rgb" +mc_ch_in_11k_f2_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_2_rgb" +mc_ch_in_11k_f3_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_3_rgb" +mc_ch_in_11k_f4_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_4_rgb" +mc_ch_in_11k_f5_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_5_rgb" +mc_ch_in_11k_f6_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_6_rgb" +mc_ch_in_11k_f7_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_7_rgb" +mc_ch_in_11k_f8_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_8_rgb" +mc_ch_in_11k_f9_rgb = "ptbench.configs.datasets.mc_ch_in_11k.fold_9_rgb" +# montgomery-shenzhen-indian-tbx11kv2 aggregated dataset +mc_ch_in_11kv2 = "ptbench.configs.datasets.mc_ch_in_11kv2.default" +mc_ch_in_11kv2_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.rgb" +mc_ch_in_11kv2_f0 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_0" +mc_ch_in_11kv2_f1 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_1" +mc_ch_in_11kv2_f2 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_2" +mc_ch_in_11kv2_f3 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_3" +mc_ch_in_11kv2_f4 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_4" +mc_ch_in_11kv2_f5 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_5" +mc_ch_in_11kv2_f6 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_6" +mc_ch_in_11kv2_f7 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_7" +mc_ch_in_11kv2_f8 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_8" +mc_ch_in_11kv2_f9 = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_9" +mc_ch_in_11kv2_f0_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_0_rgb" +mc_ch_in_11kv2_f1_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_1_rgb" +mc_ch_in_11kv2_f2_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_2_rgb" +mc_ch_in_11kv2_f3_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_3_rgb" +mc_ch_in_11kv2_f4_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_4_rgb" +mc_ch_in_11kv2_f5_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_5_rgb" +mc_ch_in_11kv2_f6_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_6_rgb" +mc_ch_in_11kv2_f7_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_7_rgb" +mc_ch_in_11kv2_f8_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_8_rgb" +mc_ch_in_11kv2_f9_rgb = "ptbench.configs.datasets.mc_ch_in_11kv2.fold_9_rgb" # tbpoc dataset (and cross-validation folds) tbpoc_f0 = "ptbench.configs.datasets.tbpoc.fold_0" tbpoc_f1 = "ptbench.configs.datasets.tbpoc.fold_1"