diff --git a/src/ptbench/configs/datasets/tbx11k_simplified_RS/__init__.py b/src/ptbench/configs/datasets/tbx11k_simplified_RS/__init__.py new file mode 100644 index 0000000000000000000000000000000000000000..7e9e9b1f0b62dccd9a2e66fefda238176b868178 --- /dev/null +++ b/src/ptbench/configs/datasets/tbx11k_simplified_RS/__init__.py @@ -0,0 +1,10 @@ +# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + + +def _maker(protocol): + from ....data.tbx11k_simplified_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/default.py b/src/ptbench/configs/datasets/tbx11k_simplified_RS/default.py new file mode 100644 index 0000000000000000000000000000000000000000..6aa1b81b16f4049b3998f434ee084eef2d11704b --- /dev/null +++ b/src/ptbench/configs/datasets/tbx11k_simplified_RS/default.py @@ -0,0 +1,17 @@ +# Copyright © 2022 Idiap Research Institute <contact@idiap.ch> +# +# SPDX-License-Identifier: GPL-3.0-or-later + +"""TBX11k simplified dataset for TB detection (default protocol) + +* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for +* "validation", 21.6% for "test" +* This split only consists of healthy and active TB samples +* "Latent TB" or "sick & non-TB" samples are not included in this configuration +* This configuration resolution: 512 x 512 (default) +* See :py:mod:`ptbench.data.tbx11k` for dataset details +""" + +from . import _maker + +dataset = _maker("default")