From c938869f16b092dfabc93ddd6666c7ae63803b3e Mon Sep 17 00:00:00 2001
From: "ogueler@idiap.ch" <ogueler@vws110.idiap.ch>
Date: Mon, 3 Apr 2023 01:15:03 +0200
Subject: [PATCH] added RS configs for tbx11k

---
 .../datasets/tbx11k_simplified_RS/__init__.py   | 10 ++++++++++
 .../datasets/tbx11k_simplified_RS/default.py    | 17 +++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 src/ptbench/configs/datasets/tbx11k_simplified_RS/__init__.py
 create mode 100644 src/ptbench/configs/datasets/tbx11k_simplified_RS/default.py

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 00000000..7e9e9b1f
--- /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 00000000..6aa1b81b
--- /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")
-- 
GitLab