def __getitem__()

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