Skip to content
Snippets Groups Projects
Commit 22d59e8c authored by ogueler@idiap.ch's avatar ogueler@idiap.ch
Browse files

added alternative configurations for tbx11k

parent b68823ca
No related branches found
No related tags found
2 merge requests!5Tbx11k,!4Moved code to lightning
Showing
with 159 additions and 11 deletions
......@@ -2,7 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (default protocol) (extended with
DensenetRS predictions)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 0)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 1)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 2)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 3)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 4)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 5)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 6)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 7)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 8)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
......@@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol)
"""TBX11k simplified dataset for TB detection (cross validation fold 9)
* Split reference: first 62.5% of TB and healthy CXR for "train" 15.9% for
* "validation", 21.6% for "test"
......
# 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_v2 import dataset as raw
from .. import make_dataset as mk
return mk([raw.subsets(protocol)])
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (default protocol) (extended with
DensenetRS predictions)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("default")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 0)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_0")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 1)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_1")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 2)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_2")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 3)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_3")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 4)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_4")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 5)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_5")
# Copyright © 2022 Idiap Research Institute <contact@idiap.ch>
#
# SPDX-License-Identifier: GPL-3.0-or-later
"""TBX11k simplified dataset for TB detection (cross validation fold 6)
* Split reference: first 62.6% of CXR for "train", 16% for "validation",
* 21.4% for "test"
* This split consists of non-TB and active TB samples
* "healthy", "latent TB", and "sick & non-TB" samples are all merged under the label "non-TB"
* This configuration resolution: 512 x 512 (default)
* See :py:mod:`ptbench.data.tbx11k_v2` for dataset details
"""
from . import _maker
dataset = _maker("fold_6")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment