From 0d1f8d36bad71e558f766789eb80b875b169b1e7 Mon Sep 17 00:00:00 2001 From: "ogueler@idiap.ch" <ogueler@vws110.idiap.ch> Date: Sat, 15 Apr 2023 02:33:52 +0200 Subject: [PATCH] added new combined datasets --- .../configs/datasets/mc_ch_in_11k/__init__.py | 140 ++++++++++++++++++ .../configs/datasets/mc_ch_in_11k/default.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_0.py | 10 ++ .../datasets/mc_ch_in_11k/fold_0_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_1.py | 10 ++ .../datasets/mc_ch_in_11k/fold_1_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_2.py | 10 ++ .../datasets/mc_ch_in_11k/fold_2_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_3.py | 10 ++ .../datasets/mc_ch_in_11k/fold_3_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_4.py | 10 ++ .../datasets/mc_ch_in_11k/fold_4_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_5.py | 10 ++ .../datasets/mc_ch_in_11k/fold_5_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_6.py | 10 ++ .../datasets/mc_ch_in_11k/fold_6_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_7.py | 10 ++ .../datasets/mc_ch_in_11k/fold_7_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_8.py | 10 ++ .../datasets/mc_ch_in_11k/fold_8_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/fold_9.py | 10 ++ .../datasets/mc_ch_in_11k/fold_9_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k/rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/__init__.py | 140 ++++++++++++++++++ .../datasets/mc_ch_in_11k_RS/default.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_0.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_0_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_1.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_1_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_2.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_2_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_3.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_3_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_4.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_4_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_5.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_5_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_6.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_6_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_7.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_7_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_8.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_8_rgb.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_9.py | 10 ++ .../datasets/mc_ch_in_11k_RS/fold_9_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11k_RS/rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2/__init__.py | 140 ++++++++++++++++++ .../datasets/mc_ch_in_11kv2/default.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_0.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_0_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_1.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_1_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_2.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_2_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_3.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_3_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_4.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_4_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_5.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_5_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_6.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_6_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_7.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_7_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_8.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_8_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/fold_9.py | 10 ++ .../datasets/mc_ch_in_11kv2/fold_9_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2/rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/__init__.py | 140 ++++++++++++++++++ .../datasets/mc_ch_in_11kv2_RS/default.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_0.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_1.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_2.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_3.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_4.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_5.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_6.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_7.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_8.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_9.py | 10 ++ .../datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py | 10 ++ .../configs/datasets/mc_ch_in_11kv2_RS/rgb.py | 10 ++ .../__init__.py | 2 +- .../default.py | 0 .../fold_0.py | 0 .../fold_1.py | 0 .../fold_2.py | 0 .../fold_3.py | 0 .../fold_4.py | 0 .../fold_5.py | 0 .../fold_6.py | 0 .../fold_7.py | 0 .../fold_8.py | 0 .../fold_9.py | 0 .../__init__.py | 2 +- .../default.py | 0 .../fold_0.py | 0 .../fold_1.py | 0 .../fold_2.py | 0 .../fold_3.py | 0 .../fold_4.py | 0 .../fold_5.py | 0 .../fold_6.py | 0 .../fold_7.py | 0 .../fold_8.py | 0 .../fold_9.py | 0 .../__init__.py | 2 +- .../__init__.py | 2 +- 118 files changed, 1444 insertions(+), 4 deletions(-) create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/default.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py create mode 100644 src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/__init__.py (78%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/default.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_0.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_1.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_2.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_3.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_4.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_5.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_6.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_7.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_8.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/fold_9.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/__init__.py (78%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/default.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_0.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_1.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_2.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_3.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_4.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_5.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_6.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_7.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_8.py (100%) rename src/ptbench/configs/datasets/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/fold_9.py (100%) rename src/ptbench/data/{tbx11k_simplified_RS_v2 => tbx11k_simplified_v2_RS}/__init__.py (98%) rename src/ptbench/data/{tbx11k_simplified_RS_v3 => tbx11k_simplified_v3_RS}/__init__.py (98%) diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py new file mode 100644 index 00000000..df4ac6bb --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/__init__.py @@ -0,0 +1,140 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +from torch.utils.data.dataset import ConcatDataset + + +def _maker(protocol): + if protocol == "default": + from ..indian import default as indian + from ..montgomery import default as mc + from ..shenzhen import default as ch + from ..tbx11k_simplified import default as tbx11k + elif protocol == "rgb": + from ..indian import rgb as indian + from ..montgomery import rgb as mc + from ..shenzhen import rgb as ch + from ..tbx11k_simplified import rgb as tbx11k + elif protocol == "fold_0": + from ..indian import fold_0 as indian + from ..montgomery import fold_0 as mc + from ..shenzhen import fold_0 as ch + from ..tbx11k_simplified import fold_0 as tbx11k + elif protocol == "fold_1": + from ..indian import fold_1 as indian + from ..montgomery import fold_1 as mc + from ..shenzhen import fold_1 as ch + from ..tbx11k_simplified import fold_1 as tbx11k + elif protocol == "fold_2": + from ..indian import fold_2 as indian + from ..montgomery import fold_2 as mc + from ..shenzhen import fold_2 as ch + from ..tbx11k_simplified import fold_2 as tbx11k + elif protocol == "fold_3": + from ..indian import fold_3 as indian + from ..montgomery import fold_3 as mc + from ..shenzhen import fold_3 as ch + from ..tbx11k_simplified import fold_3 as tbx11k + elif protocol == "fold_4": + from ..indian import fold_4 as indian + from ..montgomery import fold_4 as mc + from ..shenzhen import fold_4 as ch + from ..tbx11k_simplified import fold_4 as tbx11k + elif protocol == "fold_5": + from ..indian import fold_5 as indian + from ..montgomery import fold_5 as mc + from ..shenzhen import fold_5 as ch + from ..tbx11k_simplified import fold_5 as tbx11k + elif protocol == "fold_6": + from ..indian import fold_6 as indian + from ..montgomery import fold_6 as mc + from ..shenzhen import fold_6 as ch + from ..tbx11k_simplified import fold_6 as tbx11k + elif protocol == "fold_7": + from ..indian import fold_7 as indian + from ..montgomery import fold_7 as mc + from ..shenzhen import fold_7 as ch + from ..tbx11k_simplified import fold_7 as tbx11k + elif protocol == "fold_8": + from ..indian import fold_8 as indian + from ..montgomery import fold_8 as mc + from ..shenzhen import fold_8 as ch + from ..tbx11k_simplified import fold_8 as tbx11k + elif protocol == "fold_9": + from ..indian import fold_9 as indian + from ..montgomery import fold_9 as mc + from ..shenzhen import fold_9 as ch + from ..tbx11k_simplified import fold_9 as tbx11k + elif protocol == "fold_0_rgb": + from ..indian import fold_0_rgb as indian + from ..montgomery import fold_0_rgb as mc + from ..shenzhen import fold_0_rgb as ch + from ..tbx11k_simplified import fold_0_rgb as tbx11k + elif protocol == "fold_1_rgb": + from ..indian import fold_1_rgb as indian + from ..montgomery import fold_1_rgb as mc + from ..shenzhen import fold_1_rgb as ch + from ..tbx11k_simplified import fold_1_rgb as tbx11k + elif protocol == "fold_2_rgb": + from ..indian import fold_2_rgb as indian + from ..montgomery import fold_2_rgb as mc + from ..shenzhen import fold_2_rgb as ch + from ..tbx11k_simplified import fold_2_rgb as tbx11k + elif protocol == "fold_3_rgb": + from ..indian import fold_3_rgb as indian + from ..montgomery import fold_3_rgb as mc + from ..shenzhen import fold_3_rgb as ch + from ..tbx11k_simplified import fold_3_rgb as tbx11k + elif protocol == "fold_4_rgb": + from ..indian import fold_4_rgb as indian + from ..montgomery import fold_4_rgb as mc + from ..shenzhen import fold_4_rgb as ch + from ..tbx11k_simplified import fold_4_rgb as tbx11k + elif protocol == "fold_5_rgb": + from ..indian import fold_5_rgb as indian + from ..montgomery import fold_5_rgb as mc + from ..shenzhen import fold_5_rgb as ch + from ..tbx11k_simplified import fold_5_rgb as tbx11k + elif protocol == "fold_6_rgb": + from ..indian import fold_6_rgb as indian + from ..montgomery import fold_6_rgb as mc + from ..shenzhen import fold_6_rgb as ch + from ..tbx11k_simplified import fold_6_rgb as tbx11k + elif protocol == "fold_7_rgb": + from ..indian import fold_7_rgb as indian + from ..montgomery import fold_7_rgb as mc + from ..shenzhen import fold_7_rgb as ch + from ..tbx11k_simplified import fold_7_rgb as tbx11k + elif protocol == "fold_8_rgb": + from ..indian import fold_8_rgb as indian + from ..montgomery import fold_8_rgb as mc + from ..shenzhen import fold_8_rgb as ch + from ..tbx11k_simplified import fold_8_rgb as tbx11k + elif protocol == "fold_9_rgb": + from ..indian import fold_9_rgb as indian + from ..montgomery import fold_9_rgb as mc + from ..shenzhen import fold_9_rgb as ch + from ..tbx11k_simplified import fold_9_rgb as tbx11k + + mc = mc.dataset + ch = ch.dataset + indian = indian.dataset + tbx11k = tbx11k.dataset + + dataset = {} + dataset["__train__"] = ConcatDataset( + [mc["__train__"], ch["__train__"], indian["__train__"], tbx11k["__train__"]] + ) + dataset["train"] = ConcatDataset( + [mc["train"], ch["train"], indian["train"], tbx11k["train"]] + ) + dataset["__valid__"] = ConcatDataset( + [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11k["__valid__"]] + ) + dataset["validation"] = ConcatDataset( + [mc["validation"], ch["validation"], indian["validation"], tbx11k["validation"]] + ) + dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11k["test"]]) + + return dataset diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/default.py b/src/ptbench/configs/datasets/mc_ch_in_11k/default.py new file mode 100644 index 00000000..224d3aca --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/default.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets.""" + +from . import _maker + +dataset = _maker("default") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py new file mode 100644 index 00000000..eeb11f2c --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0)""" + +from . import _maker + +dataset = _maker("fold_0") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py new file mode 100644 index 00000000..d65df7b5 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_0_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0, RGB)""" + +from . import _maker + +dataset = _maker("fold_0_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py new file mode 100644 index 00000000..6cb3cf81 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1)""" + +from . import _maker + +dataset = _maker("fold_1") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py new file mode 100644 index 00000000..548fdbf2 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_1_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1, RGB)""" + +from . import _maker + +dataset = _maker("fold_1_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py new file mode 100644 index 00000000..474c3f15 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2)""" + +from . import _maker + +dataset = _maker("fold_2") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py new file mode 100644 index 00000000..c303c093 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_2_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2, RGB)""" + +from . import _maker + +dataset = _maker("fold_2_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py new file mode 100644 index 00000000..b2b4d9ce --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3)""" + +from . import _maker + +dataset = _maker("fold_3") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py new file mode 100644 index 00000000..4c7f0865 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_3_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3, RGB)""" + +from . import _maker + +dataset = _maker("fold_3_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py new file mode 100644 index 00000000..bc39e7a3 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4)""" + +from . import _maker + +dataset = _maker("fold_4") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py new file mode 100644 index 00000000..cf86e941 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_4_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4, RGB)""" + +from . import _maker + +dataset = _maker("fold_4_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py new file mode 100644 index 00000000..1ca82222 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5)""" + +from . import _maker + +dataset = _maker("fold_5") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py new file mode 100644 index 00000000..d2821015 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_5_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5, RGB)""" + +from . import _maker + +dataset = _maker("fold_5_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py new file mode 100644 index 00000000..d40c95f6 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6)""" + +from . import _maker + +dataset = _maker("fold_6") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py new file mode 100644 index 00000000..42c0b193 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_6_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6, RGB)""" + +from . import _maker + +dataset = _maker("fold_6_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py new file mode 100644 index 00000000..2b3d9b44 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7)""" + +from . import _maker + +dataset = _maker("fold_7") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py new file mode 100644 index 00000000..c32c4b25 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_7_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7, RGB)""" + +from . import _maker + +dataset = _maker("fold_7_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py new file mode 100644 index 00000000..2ccfbafe --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8)""" + +from . import _maker + +dataset = _maker("fold_8") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py new file mode 100644 index 00000000..c1dfad04 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_8_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8, RGB)""" + +from . import _maker + +dataset = _maker("fold_8_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py new file mode 100644 index 00000000..07068cb8 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9)""" + +from . import _maker + +dataset = _maker("fold_9") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py new file mode 100644 index 00000000..fba36c4b --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/fold_9_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9, RGB)""" + +from . import _maker + +dataset = _maker("fold_9_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py new file mode 100644 index 00000000..656a8c0f --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k/rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (RGB)""" + +from . import _maker + +dataset = _maker("rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py new file mode 100644 index 00000000..3a0c4478 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/__init__.py @@ -0,0 +1,140 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +from torch.utils.data.dataset import ConcatDataset + + +def _maker(protocol): + if protocol == "default": + from ..indian_RS import default as indian + from ..montgomery_RS import default as mc + from ..shenzhen_RS import default as ch + from ..tbx11k_simplified_RS import default as tbx11k + elif protocol == "rgb": + from ..indian_RS import rgb as indian + from ..montgomery_RS import rgb as mc + from ..shenzhen_RS import rgb as ch + from ..tbx11k_simplified_RS import rgb as tbx11k + elif protocol == "fold_0": + from ..indian_RS import fold_0 as indian + from ..montgomery_RS import fold_0 as mc + from ..shenzhen_RS import fold_0 as ch + from ..tbx11k_simplified_RS import fold_0 as tbx11k + elif protocol == "fold_1": + from ..indian_RS import fold_1 as indian + from ..montgomery_RS import fold_1 as mc + from ..shenzhen_RS import fold_1 as ch + from ..tbx11k_simplified_RS import fold_1 as tbx11k + elif protocol == "fold_2": + from ..indian_RS import fold_2 as indian + from ..montgomery_RS import fold_2 as mc + from ..shenzhen_RS import fold_2 as ch + from ..tbx11k_simplified_RS import fold_2 as tbx11k + elif protocol == "fold_3": + from ..indian_RS import fold_3 as indian + from ..montgomery_RS import fold_3 as mc + from ..shenzhen_RS import fold_3 as ch + from ..tbx11k_simplified_RS import fold_3 as tbx11k + elif protocol == "fold_4": + from ..indian_RS import fold_4 as indian + from ..montgomery_RS import fold_4 as mc + from ..shenzhen_RS import fold_4 as ch + from ..tbx11k_simplified_RS import fold_4 as tbx11k + elif protocol == "fold_5": + from ..indian_RS import fold_5 as indian + from ..montgomery_RS import fold_5 as mc + from ..shenzhen_RS import fold_5 as ch + from ..tbx11k_simplified_RS import fold_5 as tbx11k + elif protocol == "fold_6": + from ..indian_RS import fold_6 as indian + from ..montgomery_RS import fold_6 as mc + from ..shenzhen_RS import fold_6 as ch + from ..tbx11k_simplified_RS import fold_6 as tbx11k + elif protocol == "fold_7": + from ..indian_RS import fold_7 as indian + from ..montgomery_RS import fold_7 as mc + from ..shenzhen_RS import fold_7 as ch + from ..tbx11k_simplified_RS import fold_7 as tbx11k + elif protocol == "fold_8": + from ..indian_RS import fold_8 as indian + from ..montgomery_RS import fold_8 as mc + from ..shenzhen_RS import fold_8 as ch + from ..tbx11k_simplified_RS import fold_8 as tbx11k + elif protocol == "fold_9": + from ..indian_RS import fold_9 as indian + from ..montgomery_RS import fold_9 as mc + from ..shenzhen_RS import fold_9 as ch + from ..tbx11k_simplified_RS import fold_9 as tbx11k + elif protocol == "fold_0_rgb": + from ..indian_RS import fold_0_rgb as indian + from ..montgomery_RS import fold_0_rgb as mc + from ..shenzhen_RS import fold_0_rgb as ch + from ..tbx11k_simplified_RS import fold_0_rgb as tbx11k + elif protocol == "fold_1_rgb": + from ..indian_RS import fold_1_rgb as indian + from ..montgomery_RS import fold_1_rgb as mc + from ..shenzhen_RS import fold_1_rgb as ch + from ..tbx11k_simplified_RS import fold_1_rgb as tbx11k + elif protocol == "fold_2_rgb": + from ..indian_RS import fold_2_rgb as indian + from ..montgomery_RS import fold_2_rgb as mc + from ..shenzhen_RS import fold_2_rgb as ch + from ..tbx11k_simplified_RS import fold_2_rgb as tbx11k + elif protocol == "fold_3_rgb": + from ..indian_RS import fold_3_rgb as indian + from ..montgomery_RS import fold_3_rgb as mc + from ..shenzhen_RS import fold_3_rgb as ch + from ..tbx11k_simplified_RS import fold_3_rgb as tbx11k + elif protocol == "fold_4_rgb": + from ..indian_RS import fold_4_rgb as indian + from ..montgomery_RS import fold_4_rgb as mc + from ..shenzhen_RS import fold_4_rgb as ch + from ..tbx11k_simplified_RS import fold_4_rgb as tbx11k + elif protocol == "fold_5_rgb": + from ..indian_RS import fold_5_rgb as indian + from ..montgomery_RS import fold_5_rgb as mc + from ..shenzhen_RS import fold_5_rgb as ch + from ..tbx11k_simplified_RS import fold_5_rgb as tbx11k + elif protocol == "fold_6_rgb": + from ..indian_RS import fold_6_rgb as indian + from ..montgomery_RS import fold_6_rgb as mc + from ..shenzhen_RS import fold_6_rgb as ch + from ..tbx11k_simplified_RS import fold_6_rgb as tbx11k + elif protocol == "fold_7_rgb": + from ..indian_RS import fold_7_rgb as indian + from ..montgomery_RS import fold_7_rgb as mc + from ..shenzhen_RS import fold_7_rgb as ch + from ..tbx11k_simplified_RS import fold_7_rgb as tbx11k + elif protocol == "fold_8_rgb": + from ..indian_RS import fold_8_rgb as indian + from ..montgomery_RS import fold_8_rgb as mc + from ..shenzhen_RS import fold_8_rgb as ch + from ..tbx11k_simplified_RS import fold_8_rgb as tbx11k + elif protocol == "fold_9_rgb": + from ..indian_RS import fold_9_rgb as indian + from ..montgomery_RS import fold_9_rgb as mc + from ..shenzhen_RS import fold_9_rgb as ch + from ..tbx11k_simplified_RS import fold_9_rgb as tbx11k + + mc = mc.dataset + ch = ch.dataset + indian = indian.dataset + tbx11k = tbx11k.dataset + + dataset = {} + dataset["__train__"] = ConcatDataset( + [mc["__train__"], ch["__train__"], indian["__train__"], tbx11k["__train__"]] + ) + dataset["train"] = ConcatDataset( + [mc["train"], ch["train"], indian["train"], tbx11k["train"]] + ) + dataset["__valid__"] = ConcatDataset( + [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11k["__valid__"]] + ) + dataset["validation"] = ConcatDataset( + [mc["validation"], ch["validation"], indian["validation"], tbx11k["validation"]] + ) + dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11k["test"]]) + + return dataset diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py new file mode 100644 index 00000000..224d3aca --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/default.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets.""" + +from . import _maker + +dataset = _maker("default") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py new file mode 100644 index 00000000..eeb11f2c --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0)""" + +from . import _maker + +dataset = _maker("fold_0") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py new file mode 100644 index 00000000..d65df7b5 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_0_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0, RGB)""" + +from . import _maker + +dataset = _maker("fold_0_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py new file mode 100644 index 00000000..6cb3cf81 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1)""" + +from . import _maker + +dataset = _maker("fold_1") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py new file mode 100644 index 00000000..548fdbf2 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_1_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1, RGB)""" + +from . import _maker + +dataset = _maker("fold_1_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py new file mode 100644 index 00000000..474c3f15 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2)""" + +from . import _maker + +dataset = _maker("fold_2") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py new file mode 100644 index 00000000..c303c093 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_2_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2, RGB)""" + +from . import _maker + +dataset = _maker("fold_2_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py new file mode 100644 index 00000000..b2b4d9ce --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3)""" + +from . import _maker + +dataset = _maker("fold_3") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py new file mode 100644 index 00000000..4c7f0865 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_3_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3, RGB)""" + +from . import _maker + +dataset = _maker("fold_3_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py new file mode 100644 index 00000000..bc39e7a3 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4)""" + +from . import _maker + +dataset = _maker("fold_4") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py new file mode 100644 index 00000000..cf86e941 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_4_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4, RGB)""" + +from . import _maker + +dataset = _maker("fold_4_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py new file mode 100644 index 00000000..1ca82222 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5)""" + +from . import _maker + +dataset = _maker("fold_5") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py new file mode 100644 index 00000000..d2821015 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_5_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5, RGB)""" + +from . import _maker + +dataset = _maker("fold_5_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py new file mode 100644 index 00000000..d40c95f6 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6)""" + +from . import _maker + +dataset = _maker("fold_6") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py new file mode 100644 index 00000000..42c0b193 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_6_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6, RGB)""" + +from . import _maker + +dataset = _maker("fold_6_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py new file mode 100644 index 00000000..2b3d9b44 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7)""" + +from . import _maker + +dataset = _maker("fold_7") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py new file mode 100644 index 00000000..c32c4b25 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_7_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7, RGB)""" + +from . import _maker + +dataset = _maker("fold_7_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py new file mode 100644 index 00000000..2ccfbafe --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8)""" + +from . import _maker + +dataset = _maker("fold_8") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py new file mode 100644 index 00000000..c1dfad04 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_8_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8, RGB)""" + +from . import _maker + +dataset = _maker("fold_8_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py new file mode 100644 index 00000000..07068cb8 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9)""" + +from . import _maker + +dataset = _maker("fold_9") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py new file mode 100644 index 00000000..fba36c4b --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/fold_9_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9, RGB)""" + +from . import _maker + +dataset = _maker("fold_9_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py new file mode 100644 index 00000000..656a8c0f --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11k_RS/rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (RGB)""" + +from . import _maker + +dataset = _maker("rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py new file mode 100644 index 00000000..0789abe7 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/__init__.py @@ -0,0 +1,140 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +from torch.utils.data.dataset import ConcatDataset + + +def _maker(protocol): + if protocol == "default": + from ..indian import default as indian + from ..montgomery import default as mc + from ..shenzhen import default as ch + from ..tbx11k_simplified_v2 import default as tbx11kv2 + elif protocol == "rgb": + from ..indian import rgb as indian + from ..montgomery import rgb as mc + from ..shenzhen import rgb as ch + from ..tbx11k_simplified_v2 import rgb as tbx11kv2 + elif protocol == "fold_0": + from ..indian import fold_0 as indian + from ..montgomery import fold_0 as mc + from ..shenzhen import fold_0 as ch + from ..tbx11k_simplified_v2 import fold_0 as tbx11kv2 + elif protocol == "fold_1": + from ..indian import fold_1 as indian + from ..montgomery import fold_1 as mc + from ..shenzhen import fold_1 as ch + from ..tbx11k_simplified_v2 import fold_1 as tbx11kv2 + elif protocol == "fold_2": + from ..indian import fold_2 as indian + from ..montgomery import fold_2 as mc + from ..shenzhen import fold_2 as ch + from ..tbx11k_simplified_v2 import fold_2 as tbx11kv2 + elif protocol == "fold_3": + from ..indian import fold_3 as indian + from ..montgomery import fold_3 as mc + from ..shenzhen import fold_3 as ch + from ..tbx11k_simplified_v2 import fold_3 as tbx11kv2 + elif protocol == "fold_4": + from ..indian import fold_4 as indian + from ..montgomery import fold_4 as mc + from ..shenzhen import fold_4 as ch + from ..tbx11k_simplified_v2 import fold_4 as tbx11kv2 + elif protocol == "fold_5": + from ..indian import fold_5 as indian + from ..montgomery import fold_5 as mc + from ..shenzhen import fold_5 as ch + from ..tbx11k_simplified_v2 import fold_5 as tbx11kv2 + elif protocol == "fold_6": + from ..indian import fold_6 as indian + from ..montgomery import fold_6 as mc + from ..shenzhen import fold_6 as ch + from ..tbx11k_simplified_v2 import fold_6 as tbx11kv2 + elif protocol == "fold_7": + from ..indian import fold_7 as indian + from ..montgomery import fold_7 as mc + from ..shenzhen import fold_7 as ch + from ..tbx11k_simplified_v2 import fold_7 as tbx11kv2 + elif protocol == "fold_8": + from ..indian import fold_8 as indian + from ..montgomery import fold_8 as mc + from ..shenzhen import fold_8 as ch + from ..tbx11k_simplified_v2 import fold_8 as tbx11kv2 + elif protocol == "fold_9": + from ..indian import fold_9 as indian + from ..montgomery import fold_9 as mc + from ..shenzhen import fold_9 as ch + from ..tbx11k_simplified_v2 import fold_9 as tbx11kv2 + elif protocol == "fold_0_rgb": + from ..indian import fold_0_rgb as indian + from ..montgomery import fold_0_rgb as mc + from ..shenzhen import fold_0_rgb as ch + from ..tbx11k_simplified_v2 import fold_0_rgb as tbx11kv2 + elif protocol == "fold_1_rgb": + from ..indian import fold_1_rgb as indian + from ..montgomery import fold_1_rgb as mc + from ..shenzhen import fold_1_rgb as ch + from ..tbx11k_simplified_v2 import fold_1_rgb as tbx11kv2 + elif protocol == "fold_2_rgb": + from ..indian import fold_2_rgb as indian + from ..montgomery import fold_2_rgb as mc + from ..shenzhen import fold_2_rgb as ch + from ..tbx11k_simplified_v2 import fold_2_rgb as tbx11kv2 + elif protocol == "fold_3_rgb": + from ..indian import fold_3_rgb as indian + from ..montgomery import fold_3_rgb as mc + from ..shenzhen import fold_3_rgb as ch + from ..tbx11k_simplified_v2 import fold_3_rgb as tbx11kv2 + elif protocol == "fold_4_rgb": + from ..indian import fold_4_rgb as indian + from ..montgomery import fold_4_rgb as mc + from ..shenzhen import fold_4_rgb as ch + from ..tbx11k_simplified_v2 import fold_4_rgb as tbx11kv2 + elif protocol == "fold_5_rgb": + from ..indian import fold_5_rgb as indian + from ..montgomery import fold_5_rgb as mc + from ..shenzhen import fold_5_rgb as ch + from ..tbx11k_simplified_v2 import fold_5_rgb as tbx11kv2 + elif protocol == "fold_6_rgb": + from ..indian import fold_6_rgb as indian + from ..montgomery import fold_6_rgb as mc + from ..shenzhen import fold_6_rgb as ch + from ..tbx11k_simplified_v2 import fold_6_rgb as tbx11kv2 + elif protocol == "fold_7_rgb": + from ..indian import fold_7_rgb as indian + from ..montgomery import fold_7_rgb as mc + from ..shenzhen import fold_7_rgb as ch + from ..tbx11k_simplified_v2 import fold_7_rgb as tbx11kv2 + elif protocol == "fold_8_rgb": + from ..indian import fold_8_rgb as indian + from ..montgomery import fold_8_rgb as mc + from ..shenzhen import fold_8_rgb as ch + from ..tbx11k_simplified_v2 import fold_8_rgb as tbx11kv2 + elif protocol == "fold_9_rgb": + from ..indian import fold_9_rgb as indian + from ..montgomery import fold_9_rgb as mc + from ..shenzhen import fold_9_rgb as ch + from ..tbx11k_simplified_v2 import fold_9_rgb as tbx11kv2 + + mc = mc.dataset + ch = ch.dataset + indian = indian.dataset + tbx11kv2 = tbx11kv2.dataset + + dataset = {} + dataset["__train__"] = ConcatDataset( + [mc["__train__"], ch["__train__"], indian["__train__"], tbx11kv2["__train__"]] + ) + dataset["train"] = ConcatDataset( + [mc["train"], ch["train"], indian["train"], tbx11kv2["train"]] + ) + dataset["__valid__"] = ConcatDataset( + [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11kv2["__valid__"]] + ) + dataset["validation"] = ConcatDataset( + [mc["validation"], ch["validation"], indian["validation"], tbx11kv2["validation"]] + ) + dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11kv2["test"]]) + + return dataset diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py new file mode 100644 index 00000000..224d3aca --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/default.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets.""" + +from . import _maker + +dataset = _maker("default") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py new file mode 100644 index 00000000..eeb11f2c --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0)""" + +from . import _maker + +dataset = _maker("fold_0") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py new file mode 100644 index 00000000..d65df7b5 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_0_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0, RGB)""" + +from . import _maker + +dataset = _maker("fold_0_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py new file mode 100644 index 00000000..6cb3cf81 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1)""" + +from . import _maker + +dataset = _maker("fold_1") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py new file mode 100644 index 00000000..548fdbf2 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_1_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1, RGB)""" + +from . import _maker + +dataset = _maker("fold_1_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py new file mode 100644 index 00000000..474c3f15 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2)""" + +from . import _maker + +dataset = _maker("fold_2") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py new file mode 100644 index 00000000..c303c093 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_2_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2, RGB)""" + +from . import _maker + +dataset = _maker("fold_2_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py new file mode 100644 index 00000000..b2b4d9ce --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3)""" + +from . import _maker + +dataset = _maker("fold_3") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py new file mode 100644 index 00000000..4c7f0865 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_3_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3, RGB)""" + +from . import _maker + +dataset = _maker("fold_3_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py new file mode 100644 index 00000000..bc39e7a3 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4)""" + +from . import _maker + +dataset = _maker("fold_4") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py new file mode 100644 index 00000000..cf86e941 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_4_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4, RGB)""" + +from . import _maker + +dataset = _maker("fold_4_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py new file mode 100644 index 00000000..1ca82222 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5)""" + +from . import _maker + +dataset = _maker("fold_5") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py new file mode 100644 index 00000000..d2821015 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_5_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5, RGB)""" + +from . import _maker + +dataset = _maker("fold_5_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py new file mode 100644 index 00000000..d40c95f6 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6)""" + +from . import _maker + +dataset = _maker("fold_6") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py new file mode 100644 index 00000000..42c0b193 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_6_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6, RGB)""" + +from . import _maker + +dataset = _maker("fold_6_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py new file mode 100644 index 00000000..2b3d9b44 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7)""" + +from . import _maker + +dataset = _maker("fold_7") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py new file mode 100644 index 00000000..c32c4b25 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_7_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7, RGB)""" + +from . import _maker + +dataset = _maker("fold_7_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py new file mode 100644 index 00000000..2ccfbafe --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8)""" + +from . import _maker + +dataset = _maker("fold_8") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py new file mode 100644 index 00000000..c1dfad04 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_8_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8, RGB)""" + +from . import _maker + +dataset = _maker("fold_8_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py new file mode 100644 index 00000000..07068cb8 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9)""" + +from . import _maker + +dataset = _maker("fold_9") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py new file mode 100644 index 00000000..fba36c4b --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/fold_9_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9, RGB)""" + +from . import _maker + +dataset = _maker("fold_9_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py new file mode 100644 index 00000000..656a8c0f --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2/rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (RGB)""" + +from . import _maker + +dataset = _maker("rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py new file mode 100644 index 00000000..06725740 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/__init__.py @@ -0,0 +1,140 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +from torch.utils.data.dataset import ConcatDataset + + +def _maker(protocol): + if protocol == "default": + from ..indian_RS import default as indian + from ..montgomery_RS import default as mc + from ..shenzhen_RS import default as ch + from ..tbx11k_simplified_v2_RS import default as tbx11kv2 + elif protocol == "rgb": + from ..indian_RS import rgb as indian + from ..montgomery_RS import rgb as mc + from ..shenzhen_RS import rgb as ch + from ..tbx11k_simplified_v2_RS import rgb as tbx11kv2 + elif protocol == "fold_0": + from ..indian_RS import fold_0 as indian + from ..montgomery_RS import fold_0 as mc + from ..shenzhen_RS import fold_0 as ch + from ..tbx11k_simplified_v2_RS import fold_0 as tbx11kv2 + elif protocol == "fold_1": + from ..indian_RS import fold_1 as indian + from ..montgomery_RS import fold_1 as mc + from ..shenzhen_RS import fold_1 as ch + from ..tbx11k_simplified_v2_RS import fold_1 as tbx11kv2 + elif protocol == "fold_2": + from ..indian_RS import fold_2 as indian + from ..montgomery_RS import fold_2 as mc + from ..shenzhen_RS import fold_2 as ch + from ..tbx11k_simplified_v2_RS import fold_2 as tbx11kv2 + elif protocol == "fold_3": + from ..indian_RS import fold_3 as indian + from ..montgomery_RS import fold_3 as mc + from ..shenzhen_RS import fold_3 as ch + from ..tbx11k_simplified_v2_RS import fold_3 as tbx11kv2 + elif protocol == "fold_4": + from ..indian_RS import fold_4 as indian + from ..montgomery_RS import fold_4 as mc + from ..shenzhen_RS import fold_4 as ch + from ..tbx11k_simplified_v2_RS import fold_4 as tbx11kv2 + elif protocol == "fold_5": + from ..indian_RS import fold_5 as indian + from ..montgomery_RS import fold_5 as mc + from ..shenzhen_RS import fold_5 as ch + from ..tbx11k_simplified_v2_RS import fold_5 as tbx11kv2 + elif protocol == "fold_6": + from ..indian_RS import fold_6 as indian + from ..montgomery_RS import fold_6 as mc + from ..shenzhen_RS import fold_6 as ch + from ..tbx11k_simplified_v2_RS import fold_6 as tbx11kv2 + elif protocol == "fold_7": + from ..indian_RS import fold_7 as indian + from ..montgomery_RS import fold_7 as mc + from ..shenzhen_RS import fold_7 as ch + from ..tbx11k_simplified_v2_RS import fold_7 as tbx11kv2 + elif protocol == "fold_8": + from ..indian_RS import fold_8 as indian + from ..montgomery_RS import fold_8 as mc + from ..shenzhen_RS import fold_8 as ch + from ..tbx11k_simplified_v2_RS import fold_8 as tbx11kv2 + elif protocol == "fold_9": + from ..indian_RS import fold_9 as indian + from ..montgomery_RS import fold_9 as mc + from ..shenzhen_RS import fold_9 as ch + from ..tbx11k_simplified_v2_RS import fold_9 as tbx11kv2 + elif protocol == "fold_0_rgb": + from ..indian_RS import fold_0_rgb as indian + from ..montgomery_RS import fold_0_rgb as mc + from ..shenzhen_RS import fold_0_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_0_rgb as tbx11kv2 + elif protocol == "fold_1_rgb": + from ..indian_RS import fold_1_rgb as indian + from ..montgomery_RS import fold_1_rgb as mc + from ..shenzhen_RS import fold_1_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_1_rgb as tbx11kv2 + elif protocol == "fold_2_rgb": + from ..indian_RS import fold_2_rgb as indian + from ..montgomery_RS import fold_2_rgb as mc + from ..shenzhen_RS import fold_2_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_2_rgb as tbx11kv2 + elif protocol == "fold_3_rgb": + from ..indian_RS import fold_3_rgb as indian + from ..montgomery_RS import fold_3_rgb as mc + from ..shenzhen_RS import fold_3_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_3_rgb as tbx11kv2 + elif protocol == "fold_4_rgb": + from ..indian_RS import fold_4_rgb as indian + from ..montgomery_RS import fold_4_rgb as mc + from ..shenzhen_RS import fold_4_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_4_rgb as tbx11kv2 + elif protocol == "fold_5_rgb": + from ..indian_RS import fold_5_rgb as indian + from ..montgomery_RS import fold_5_rgb as mc + from ..shenzhen_RS import fold_5_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_5_rgb as tbx11kv2 + elif protocol == "fold_6_rgb": + from ..indian_RS import fold_6_rgb as indian + from ..montgomery_RS import fold_6_rgb as mc + from ..shenzhen_RS import fold_6_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_6_rgb as tbx11kv2 + elif protocol == "fold_7_rgb": + from ..indian_RS import fold_7_rgb as indian + from ..montgomery_RS import fold_7_rgb as mc + from ..shenzhen_RS import fold_7_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_7_rgb as tbx11kv2 + elif protocol == "fold_8_rgb": + from ..indian_RS import fold_8_rgb as indian + from ..montgomery_RS import fold_8_rgb as mc + from ..shenzhen_RS import fold_8_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_8_rgb as tbx11kv2 + elif protocol == "fold_9_rgb": + from ..indian_RS import fold_9_rgb as indian + from ..montgomery_RS import fold_9_rgb as mc + from ..shenzhen_RS import fold_9_rgb as ch + from ..tbx11k_simplified_v2_RS import fold_9_rgb as tbx11kv2 + + mc = mc.dataset + ch = ch.dataset + indian = indian.dataset + tbx11kv2 = tbx11kv2.dataset + + dataset = {} + dataset["__train__"] = ConcatDataset( + [mc["__train__"], ch["__train__"], indian["__train__"], tbx11kv2["__train__"]] + ) + dataset["train"] = ConcatDataset( + [mc["train"], ch["train"], indian["train"], tbx11kv2["train"]] + ) + dataset["__valid__"] = ConcatDataset( + [mc["__valid__"], ch["__valid__"], indian["__valid__"], tbx11kv2["__valid__"]] + ) + dataset["validation"] = ConcatDataset( + [mc["validation"], ch["validation"], indian["validation"], tbx11kv2["validation"]] + ) + dataset["test"] = ConcatDataset([mc["test"], ch["test"], indian["test"], tbx11kv2["test"]]) + + return dataset diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py new file mode 100644 index 00000000..224d3aca --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/default.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets.""" + +from . import _maker + +dataset = _maker("default") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py new file mode 100644 index 00000000..eeb11f2c --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0)""" + +from . import _maker + +dataset = _maker("fold_0") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py new file mode 100644 index 00000000..d65df7b5 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_0_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 0, RGB)""" + +from . import _maker + +dataset = _maker("fold_0_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py new file mode 100644 index 00000000..6cb3cf81 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1)""" + +from . import _maker + +dataset = _maker("fold_1") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py new file mode 100644 index 00000000..548fdbf2 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_1_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 1, RGB)""" + +from . import _maker + +dataset = _maker("fold_1_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py new file mode 100644 index 00000000..474c3f15 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2)""" + +from . import _maker + +dataset = _maker("fold_2") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py new file mode 100644 index 00000000..c303c093 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_2_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 2, RGB)""" + +from . import _maker + +dataset = _maker("fold_2_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py new file mode 100644 index 00000000..b2b4d9ce --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3)""" + +from . import _maker + +dataset = _maker("fold_3") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py new file mode 100644 index 00000000..4c7f0865 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_3_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 3, RGB)""" + +from . import _maker + +dataset = _maker("fold_3_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py new file mode 100644 index 00000000..bc39e7a3 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4)""" + +from . import _maker + +dataset = _maker("fold_4") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py new file mode 100644 index 00000000..cf86e941 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_4_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 4, RGB)""" + +from . import _maker + +dataset = _maker("fold_4_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py new file mode 100644 index 00000000..1ca82222 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5)""" + +from . import _maker + +dataset = _maker("fold_5") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py new file mode 100644 index 00000000..d2821015 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_5_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 5, RGB)""" + +from . import _maker + +dataset = _maker("fold_5_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py new file mode 100644 index 00000000..d40c95f6 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6)""" + +from . import _maker + +dataset = _maker("fold_6") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py new file mode 100644 index 00000000..42c0b193 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_6_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 6, RGB)""" + +from . import _maker + +dataset = _maker("fold_6_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py new file mode 100644 index 00000000..2b3d9b44 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7)""" + +from . import _maker + +dataset = _maker("fold_7") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py new file mode 100644 index 00000000..c32c4b25 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_7_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 7, RGB)""" + +from . import _maker + +dataset = _maker("fold_7_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py new file mode 100644 index 00000000..2ccfbafe --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8)""" + +from . import _maker + +dataset = _maker("fold_8") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py new file mode 100644 index 00000000..c1dfad04 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_8_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 8, RGB)""" + +from . import _maker + +dataset = _maker("fold_8_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py new file mode 100644 index 00000000..07068cb8 --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9)""" + +from . import _maker + +dataset = _maker("fold_9") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py new file mode 100644 index 00000000..fba36c4b --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/fold_9_rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (cross validation fold 9, RGB)""" + +from . import _maker + +dataset = _maker("fold_9_rgb") diff --git a/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py new file mode 100644 index 00000000..656a8c0f --- /dev/null +++ b/src/ptbench/configs/datasets/mc_ch_in_11kv2_RS/rgb.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""Aggregated dataset composed of Montgomery, Shenzhen, Indian and +the default TBX11K-simplified datasets (RGB)""" + +from . import _maker + +dataset = _maker("rgb") diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/__init__.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/__init__.py similarity index 78% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/__init__.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/__init__.py index c7cb9fb8..4fe785fd 100644 --- a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/__init__.py +++ b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/__init__.py @@ -4,7 +4,7 @@ def _maker(protocol): - from ....data.tbx11k_simplified_RS_v2 import dataset as raw + from ....data.tbx11k_simplified_v2_RS import dataset as raw from .. import make_dataset as mk return mk([raw.subsets(protocol)]) diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/default.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/default.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/default.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/default.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_0.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_0.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_0.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_0.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_1.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_1.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_1.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_1.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_2.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_2.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_2.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_2.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_3.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_3.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_3.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_3.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_4.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_4.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_4.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_4.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_5.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_5.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_5.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_5.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_6.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_6.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_6.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_6.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_7.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_7.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_7.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_7.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_8.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_8.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_8.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_8.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_9.py b/src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_9.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v2/fold_9.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v2_RS/fold_9.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/__init__.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/__init__.py similarity index 78% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/__init__.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/__init__.py index a21cf7f4..176a2844 100644 --- a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/__init__.py +++ b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/__init__.py @@ -4,7 +4,7 @@ def _maker(protocol): - from ....data.tbx11k_simplified_RS_v3 import dataset as raw + from ....data.tbx11k_simplified_v3_RS import dataset as raw from .. import make_dataset as mk return mk([raw.subsets(protocol)]) diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/default.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/default.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/default.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/default.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_0.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_0.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_0.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_0.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_1.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_1.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_1.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_1.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_2.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_2.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_2.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_2.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_3.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_3.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_3.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_3.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_4.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_4.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_4.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_4.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_5.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_5.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_5.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_5.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_6.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_6.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_6.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_6.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_7.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_7.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_7.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_7.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_8.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_8.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_8.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_8.py diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_9.py b/src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_9.py similarity index 100% rename from src/ptbench/configs/datasets/tbx11k_simplified_RS_v3/fold_9.py rename to src/ptbench/configs/datasets/tbx11k_simplified_v3_RS/fold_9.py diff --git a/src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py b/src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py similarity index 98% rename from src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py rename to src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py index 3a61ae3b..23dd76b7 100644 --- a/src/ptbench/data/tbx11k_simplified_RS_v2/__init__.py +++ b/src/ptbench/data/tbx11k_simplified_v2_RS/__init__.py @@ -27,7 +27,7 @@ The version of the dataset used in this benchmark is a simplified. import importlib.resources from ..dataset import JSONDataset -from ..loader import make_delayed +from ..loader import make_delayed, make_delayed_bbox _protocols = [ importlib.resources.files(__name__).joinpath("default.json.bz2"), diff --git a/src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py b/src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py similarity index 98% rename from src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py rename to src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py index 3a61ae3b..23dd76b7 100644 --- a/src/ptbench/data/tbx11k_simplified_RS_v3/__init__.py +++ b/src/ptbench/data/tbx11k_simplified_v3_RS/__init__.py @@ -27,7 +27,7 @@ The version of the dataset used in this benchmark is a simplified. import importlib.resources from ..dataset import JSONDataset -from ..loader import make_delayed +from ..loader import make_delayed, make_delayed_bbox _protocols = [ importlib.resources.files(__name__).joinpath("default.json.bz2"), -- GitLab