in covidprognosis/data/nih_chest_xrays.py [0:0]
def __getitem__(self, idx: int) -> Dict:
assert self.csv is not None
exam = self.csv.iloc[idx]
filename = self.directory / "images" / exam["Image Index"]
image = self.open_image(filename)
metadata = self.retrieve_metadata(idx, filename, exam)
# example: exam['Finding Labels'] = 'Pneumonia|Cardiomegaly'
# goal here is to see if label is a substring of
# 'Pneumonia|Cardiomegaly' for each label in self.label_list
labels = [
1 if label in exam["Finding Labels"] else 0 for label in self.label_list
]
labels = np.array(labels).astype(np.float)
sample = {"image": image, "labels": labels, "metadata": metadata}
if self.transform is not None:
sample = self.transform(sample)
return sample