From 7a02ba55581aa2894a437207deba0e290c819eb1 Mon Sep 17 00:00:00 2001
From: "ogueler@idiap.ch" <ogueler@vws110.idiap.ch>
Date: Mon, 3 Apr 2023 01:11:01 +0200
Subject: [PATCH] added rgb file for tbx11k

---
 .../configs/datasets/tbx11k_simplified/rgb.py   | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 src/ptbench/configs/datasets/tbx11k_simplified/rgb.py

diff --git a/src/ptbench/configs/datasets/tbx11k_simplified/rgb.py b/src/ptbench/configs/datasets/tbx11k_simplified/rgb.py
new file mode 100644
index 00000000..c4ddb5a4
--- /dev/null
+++ b/src/ptbench/configs/datasets/tbx11k_simplified/rgb.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", RGB=True)
-- 
GitLab