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

[data.classify.nih_cxr14] Fix file access

parent e8930219
No related branches found
No related tags found
No related merge requests found
Pipeline #91326 passed
......@@ -222,7 +222,7 @@ class RawDataLoader(BaseDataLoader):
# N.B.: some NIH CXR-14 images are encoded as color PNGs with an alpha
# channel. Most, are grayscale PNGs
image = PIL.Image.open(self.datadir / sample[0]).convert(mode="L")
image = PIL.Image.open(self.datadir / file_path).convert(mode="L")
image = to_dtype(to_image(image), torch.float32, scale=True)
image = tv_tensors.Image(image)
......
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