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 0000000000000000000000000000000000000000..df4ac6bb453bc5c2da648e44b9fa4e7dc290ea55 --- /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 0000000000000000000000000000000000000000..224d3acabe7597385b3bdd86195e55aeddbf0610 --- /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 0000000000000000000000000000000000000000..eeb11f2c7d114b7a2c2020a0ab23df3efdd38a75 --- /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 0000000000000000000000000000000000000000..d65df7b5be88653ed816136d2370e4eb197fd8d0 --- /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 0000000000000000000000000000000000000000..6cb3cf8101aaabb31d09b47e34a29e4f830d79b8 --- /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 0000000000000000000000000000000000000000..548fdbf2668e1201ab2a16249ed94e29960a1fd6 --- /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 0000000000000000000000000000000000000000..474c3f15129ea0c7ce0d1f10b77d6d1a137917cb --- /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 0000000000000000000000000000000000000000..c303c093275f14818095c35eebb68eabd7c495d3 --- /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 0000000000000000000000000000000000000000..b2b4d9ce0fa9353c2c38be10592a653439eb8f9e --- /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 0000000000000000000000000000000000000000..4c7f0865190bdc8038bd3518acc02e82f8525889 --- /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 0000000000000000000000000000000000000000..bc39e7a36e06d1bda4b4d88f524923c04bcd3717 --- /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 0000000000000000000000000000000000000000..cf86e9416ae363554eb6753a2384691f812e0020 --- /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 0000000000000000000000000000000000000000..1ca82222c81d3fdfff1d1144f0530efab171213f --- /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 0000000000000000000000000000000000000000..d28210151b9f46905f68d55fbf6991cd79099eab --- /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 0000000000000000000000000000000000000000..d40c95f64dc0ca6b0522d849bff6dbf82eb561f7 --- /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 0000000000000000000000000000000000000000..42c0b193dbade22ad2fa104587a42c6ab2fb8ec2 --- /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 0000000000000000000000000000000000000000..2b3d9b44ef872ae031e34b35ebffedadfbedb736 --- /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 0000000000000000000000000000000000000000..c32c4b258ceffefa1841f09d5b7640e3a7f06eff --- /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 0000000000000000000000000000000000000000..2ccfbafe36a4e6c73b6a4639d1602b1663ef4e70 --- /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 0000000000000000000000000000000000000000..c1dfad049dd31cc2367ac0017ca465b14d7b3b77 --- /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 0000000000000000000000000000000000000000..07068cb8662f3bcb4098827e948d21b563cdc74e --- /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 0000000000000000000000000000000000000000..fba36c4b9efd3629f5f0ce4bc9e684324d549cd0 --- /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 0000000000000000000000000000000000000000..656a8c0ff0337da1a2037a6cb00db0c0f4fdbe64 --- /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 0000000000000000000000000000000000000000..3a0c447890cd30d1a7a5a3c0f5582366103a8a51 --- /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 0000000000000000000000000000000000000000..224d3acabe7597385b3bdd86195e55aeddbf0610 --- /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 0000000000000000000000000000000000000000..eeb11f2c7d114b7a2c2020a0ab23df3efdd38a75 --- /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 0000000000000000000000000000000000000000..d65df7b5be88653ed816136d2370e4eb197fd8d0 --- /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 0000000000000000000000000000000000000000..6cb3cf8101aaabb31d09b47e34a29e4f830d79b8 --- /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 0000000000000000000000000000000000000000..548fdbf2668e1201ab2a16249ed94e29960a1fd6 --- /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 0000000000000000000000000000000000000000..474c3f15129ea0c7ce0d1f10b77d6d1a137917cb --- /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 0000000000000000000000000000000000000000..c303c093275f14818095c35eebb68eabd7c495d3 --- /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 0000000000000000000000000000000000000000..b2b4d9ce0fa9353c2c38be10592a653439eb8f9e --- /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 0000000000000000000000000000000000000000..4c7f0865190bdc8038bd3518acc02e82f8525889 --- /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 0000000000000000000000000000000000000000..bc39e7a36e06d1bda4b4d88f524923c04bcd3717 --- /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 0000000000000000000000000000000000000000..cf86e9416ae363554eb6753a2384691f812e0020 --- /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 0000000000000000000000000000000000000000..1ca82222c81d3fdfff1d1144f0530efab171213f --- /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 0000000000000000000000000000000000000000..d28210151b9f46905f68d55fbf6991cd79099eab --- /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 0000000000000000000000000000000000000000..d40c95f64dc0ca6b0522d849bff6dbf82eb561f7 --- /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 0000000000000000000000000000000000000000..42c0b193dbade22ad2fa104587a42c6ab2fb8ec2 --- /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 0000000000000000000000000000000000000000..2b3d9b44ef872ae031e34b35ebffedadfbedb736 --- /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 0000000000000000000000000000000000000000..c32c4b258ceffefa1841f09d5b7640e3a7f06eff --- /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 0000000000000000000000000000000000000000..2ccfbafe36a4e6c73b6a4639d1602b1663ef4e70 --- /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 0000000000000000000000000000000000000000..c1dfad049dd31cc2367ac0017ca465b14d7b3b77 --- /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 0000000000000000000000000000000000000000..07068cb8662f3bcb4098827e948d21b563cdc74e --- /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 0000000000000000000000000000000000000000..fba36c4b9efd3629f5f0ce4bc9e684324d549cd0 --- /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 0000000000000000000000000000000000000000..656a8c0ff0337da1a2037a6cb00db0c0f4fdbe64 --- /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 0000000000000000000000000000000000000000..0789abe72d5bc5d4dd2773bdd4cf3e93ca6a9e01 --- /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 0000000000000000000000000000000000000000..224d3acabe7597385b3bdd86195e55aeddbf0610 --- /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 0000000000000000000000000000000000000000..eeb11f2c7d114b7a2c2020a0ab23df3efdd38a75 --- /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 0000000000000000000000000000000000000000..d65df7b5be88653ed816136d2370e4eb197fd8d0 --- /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 0000000000000000000000000000000000000000..6cb3cf8101aaabb31d09b47e34a29e4f830d79b8 --- /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 0000000000000000000000000000000000000000..548fdbf2668e1201ab2a16249ed94e29960a1fd6 --- /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 0000000000000000000000000000000000000000..474c3f15129ea0c7ce0d1f10b77d6d1a137917cb --- /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 0000000000000000000000000000000000000000..c303c093275f14818095c35eebb68eabd7c495d3 --- /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 0000000000000000000000000000000000000000..b2b4d9ce0fa9353c2c38be10592a653439eb8f9e --- /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 0000000000000000000000000000000000000000..4c7f0865190bdc8038bd3518acc02e82f8525889 --- /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 0000000000000000000000000000000000000000..bc39e7a36e06d1bda4b4d88f524923c04bcd3717 --- /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 0000000000000000000000000000000000000000..cf86e9416ae363554eb6753a2384691f812e0020 --- /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 0000000000000000000000000000000000000000..1ca82222c81d3fdfff1d1144f0530efab171213f --- /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 0000000000000000000000000000000000000000..d28210151b9f46905f68d55fbf6991cd79099eab --- /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 0000000000000000000000000000000000000000..d40c95f64dc0ca6b0522d849bff6dbf82eb561f7 --- /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 0000000000000000000000000000000000000000..42c0b193dbade22ad2fa104587a42c6ab2fb8ec2 --- /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 0000000000000000000000000000000000000000..2b3d9b44ef872ae031e34b35ebffedadfbedb736 --- /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 0000000000000000000000000000000000000000..c32c4b258ceffefa1841f09d5b7640e3a7f06eff --- /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 0000000000000000000000000000000000000000..2ccfbafe36a4e6c73b6a4639d1602b1663ef4e70 --- /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 0000000000000000000000000000000000000000..c1dfad049dd31cc2367ac0017ca465b14d7b3b77 --- /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 0000000000000000000000000000000000000000..07068cb8662f3bcb4098827e948d21b563cdc74e --- /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 0000000000000000000000000000000000000000..fba36c4b9efd3629f5f0ce4bc9e684324d549cd0 --- /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 0000000000000000000000000000000000000000..656a8c0ff0337da1a2037a6cb00db0c0f4fdbe64 --- /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 0000000000000000000000000000000000000000..06725740978efa4f64f583e755cf66e6f12924fd --- /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 0000000000000000000000000000000000000000..224d3acabe7597385b3bdd86195e55aeddbf0610 --- /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 0000000000000000000000000000000000000000..eeb11f2c7d114b7a2c2020a0ab23df3efdd38a75 --- /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 0000000000000000000000000000000000000000..d65df7b5be88653ed816136d2370e4eb197fd8d0 --- /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 0000000000000000000000000000000000000000..6cb3cf8101aaabb31d09b47e34a29e4f830d79b8 --- /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 0000000000000000000000000000000000000000..548fdbf2668e1201ab2a16249ed94e29960a1fd6 --- /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 0000000000000000000000000000000000000000..474c3f15129ea0c7ce0d1f10b77d6d1a137917cb --- /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 0000000000000000000000000000000000000000..c303c093275f14818095c35eebb68eabd7c495d3 --- /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 0000000000000000000000000000000000000000..b2b4d9ce0fa9353c2c38be10592a653439eb8f9e --- /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 0000000000000000000000000000000000000000..4c7f0865190bdc8038bd3518acc02e82f8525889 --- /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 0000000000000000000000000000000000000000..bc39e7a36e06d1bda4b4d88f524923c04bcd3717 --- /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 0000000000000000000000000000000000000000..cf86e9416ae363554eb6753a2384691f812e0020 --- /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 0000000000000000000000000000000000000000..1ca82222c81d3fdfff1d1144f0530efab171213f --- /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 0000000000000000000000000000000000000000..d28210151b9f46905f68d55fbf6991cd79099eab --- /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 0000000000000000000000000000000000000000..d40c95f64dc0ca6b0522d849bff6dbf82eb561f7 --- /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 0000000000000000000000000000000000000000..42c0b193dbade22ad2fa104587a42c6ab2fb8ec2 --- /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 0000000000000000000000000000000000000000..2b3d9b44ef872ae031e34b35ebffedadfbedb736 --- /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 0000000000000000000000000000000000000000..c32c4b258ceffefa1841f09d5b7640e3a7f06eff --- /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 0000000000000000000000000000000000000000..2ccfbafe36a4e6c73b6a4639d1602b1663ef4e70 --- /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 0000000000000000000000000000000000000000..c1dfad049dd31cc2367ac0017ca465b14d7b3b77 --- /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 0000000000000000000000000000000000000000..07068cb8662f3bcb4098827e948d21b563cdc74e --- /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 0000000000000000000000000000000000000000..fba36c4b9efd3629f5f0ce4bc9e684324d549cd0 --- /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 0000000000000000000000000000000000000000..656a8c0ff0337da1a2037a6cb00db0c0f4fdbe64 --- /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 c7cb9fb8c67ad9b3376d4c33d6dc8c77f09a1fdd..4fe785fd103413df959fdd4490387739b7030858 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 a21cf7f43ee8d328a2d1dd615a18ad6842c8ea23..176a28448003e2107060b089d1e9b0418c49296a 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 3a61ae3b2cffdb51cd9f9c52f17524eaff6a2a5d..23dd76b7e83a049040fae0fe111c3f187c2275dc 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 3a61ae3b2cffdb51cd9f9c52f17524eaff6a2a5d..23dd76b7e83a049040fae0fe111c3f187c2275dc 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"),