Skip to content
Snippets Groups Projects
Commit b2158758 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[data.padchest] Add more comments about labels in the various splits

parent 06cef603
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
Pipeline #76700 failed
......@@ -36,7 +36,7 @@ class RawDataLoader(_BaseRawDataLoader):
rc = load_rc()
self.datadir = rc.get("datadir.padchest", os.path.realpath(os.curdir))
def sample(self, sample: tuple[str, list[int]]) -> Sample:
def sample(self, sample: tuple[str, int | list[int]]) -> Sample:
"""Loads a single image sample from the disk.
Parameters
......@@ -68,7 +68,7 @@ class RawDataLoader(_BaseRawDataLoader):
return tensor, dict(label=sample[1], name=sample[0]) # type: ignore[arg-type]
def label(self, sample: tuple[str, list[int]]) -> list[int]:
def label(self, sample: tuple[str, int | list[int]]) -> int | list[int]:
"""Loads a single image sample label from the disk.
Parameters
......
......@@ -13,7 +13,7 @@ datamodule = DataModule("no-tb-idiap.json.bz2")
* Images path adapted to Idiap infrastructure
* Labels (in order):
* Labels (in order), match those of NIH CXR-14:
* cardiomegaly
* emphysema
......
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