diff --git a/pyproject.toml b/pyproject.toml index 2219b5c2e35b15dd70278545aeb4864901b1adc3..609a0a08be71c96e5f338c59aeb4f8ad289ae8aa 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -209,6 +209,18 @@ tbx11k_simplified_f6_rgb = "ptbench.configs.datasets.tbx11k_simplified.fold_6_rg 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" +# extended TBX11K simplified dataset split 1 (with radiological signs) +tbx11k_simplified_rs = "ptbench.configs.datasets.tbx11k_simplified_RS.default" +tbx11k_simplified_rs_f0 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_0" +tbx11k_simplified_rs_f1 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_1" +tbx11k_simplified_rs_f2 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_2" +tbx11k_simplified_rs_f3 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_3" +tbx11k_simplified_rs_f4 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_4" +tbx11k_simplified_rs_f5 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_5" +tbx11k_simplified_rs_f6 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_6" +tbx11k_simplified_rs_f7 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_7" +tbx11k_simplified_rs_f8 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_8" +tbx11k_simplified_rs_f9 = "ptbench.configs.datasets.tbx11k_simplified_RS.fold_9" # 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" @@ -232,6 +244,18 @@ tbx11k_simplified_v2_f6_rgb = "ptbench.configs.datasets.tbx11k_simplified_v2.fol 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" +# extended TBX11K simplified dataset split 2 (with radiological signs) +tbx11k_simplified_v2_rs = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.default" +tbx11k_simplified_v2_rs_f0 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_0" +tbx11k_simplified_v2_rs_f1 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_1" +tbx11k_simplified_v2_rs_f2 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_2" +tbx11k_simplified_v2_rs_f3 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_3" +tbx11k_simplified_v2_rs_f4 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_4" +tbx11k_simplified_v2_rs_f5 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_5" +tbx11k_simplified_v2_rs_f6 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_6" +tbx11k_simplified_v2_rs_f7 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_7" +tbx11k_simplified_v2_rs_f8 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_8" +tbx11k_simplified_v2_rs_f9 = "ptbench.configs.datasets.tbx11k_simplified_v2_RS.fold_9" # 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" @@ -255,6 +279,18 @@ tbx11k_simplified_v3_f6_rgb = "ptbench.configs.datasets.tbx11k_simplified_v3.fol 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" +# extended TBX11K simplified dataset split 3 (with radiological signs) +tbx11k_simplified_v3_rs = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.default" +tbx11k_simplified_v3_rs_f0 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_0" +tbx11k_simplified_v3_rs_f1 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_1" +tbx11k_simplified_v3_rs_f2 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_2" +tbx11k_simplified_v3_rs_f3 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_3" +tbx11k_simplified_v3_rs_f4 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_4" +tbx11k_simplified_v3_rs_f5 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_5" +tbx11k_simplified_v3_rs_f6 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_6" +tbx11k_simplified_v3_rs_f7 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_7" +tbx11k_simplified_v3_rs_f8 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_8" +tbx11k_simplified_v3_rs_f9 = "ptbench.configs.datasets.tbx11k_simplified_v3_RS.fold_9" # montgomery-shenzhen aggregated dataset mc_ch = "ptbench.configs.datasets.mc_ch.default" mc_ch_rgb = "ptbench.configs.datasets.mc_ch.rgb" @@ -350,6 +386,19 @@ 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" +# extended montgomery-shenzhen-indian-tbx11k aggregated dataset +# (with radiological signs) +mc_ch_in_11k_rs = "ptbench.configs.datasets.mc_ch_in_11k_RS.default" +mc_ch_in_11k_rs_f0 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_0" +mc_ch_in_11k_rs_f1 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_1" +mc_ch_in_11k_rs_f2 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_2" +mc_ch_in_11k_rs_f3 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_3" +mc_ch_in_11k_rs_f4 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_4" +mc_ch_in_11k_rs_f5 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_5" +mc_ch_in_11k_rs_f6 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_6" +mc_ch_in_11k_rs_f7 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_7" +mc_ch_in_11k_rs_f8 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_8" +mc_ch_in_11k_rs_f9 = "ptbench.configs.datasets.mc_ch_in_11k_RS.fold_9" # 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" @@ -373,6 +422,19 @@ 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" +# extended montgomery-shenzhen-indian-tbx11kv2 aggregated dataset +# (with radiological signs) +mc_ch_in_11kv2_rs = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.default" +mc_ch_in_11kv2_rs_f0 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_0" +mc_ch_in_11kv2_rs_f1 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_1" +mc_ch_in_11kv2_rs_f2 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_2" +mc_ch_in_11kv2_rs_f3 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_3" +mc_ch_in_11kv2_rs_f4 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_4" +mc_ch_in_11kv2_rs_f5 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_5" +mc_ch_in_11kv2_rs_f6 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_6" +mc_ch_in_11kv2_rs_f7 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_7" +mc_ch_in_11kv2_rs_f8 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_8" +mc_ch_in_11kv2_rs_f9 = "ptbench.configs.datasets.mc_ch_in_11kv2_RS.fold_9" # tbpoc dataset (and cross-validation folds) tbpoc_f0 = "ptbench.configs.datasets.tbpoc.fold_0" tbpoc_f1 = "ptbench.configs.datasets.tbpoc.fold_1"