Dataloader error with DICOM files

I’m trying to build a basic pneumonia classifier based on this dataset with DICOM files:

I tried copying the proces from this fastai2 documentation notebook:

Unfortunately I get the following error when I try to create the dataloader:

UnboundLocalError Traceback (most recent call last)
in ()
----> 1 dls = pneumonia.dataloaders(df.values, bs=bs)

11 frames
/usr/local/lib/python3.6/dist-packages/fastai2/medical/ in create(cls, fn, mode)
43 if isinstance(fn,bytes): im = Image.fromarray(pydicom.dcmread(pydicom.filebase.DicomBytesIO(fn)).pixel_array)
44 if isinstance(fn,Path): im = Image.fromarray(dcmread(fn).pixel_array)
—> 45 im.load()
46 im = im._new(
47 return cls(im.convert(mode) if mode else im)

UnboundLocalError: local variable ‘im’ referenced before assignment

My datablock looks like this:

pneumonia = DataBlock(blocks=(ImageBlock(cls=PILDicom), CategoryBlock),

               splitter=RandomSplitter(valid_pct=0.2, seed=42),

               get_x=ColReader(0, pref=pneumonia_img, suff='.dcm'),



And my dataloader like this:

dls = pneumonia.dataloaders(df.values, bs=bs)

I did get the tutorial notebook for medical imaging working without problems, which uses a very similar dateset structure. Did anyone run into a similar problem with this dataset?

It seems like a bug.

class PILDicom(PILBase):
    _open_args,_tensor_cls,_show_args = {},TensorDicom,TensorDicom._show_args
    def create(cls, fn:(Path,str,bytes), mode=None)->None:
        "Open a `DICOM file` from path `fn` or bytes `fn` and load it as a `PIL Image`"
        if isinstance(fn,bytes): im = Image.fromarray(pydicom.dcmread(pydicom.filebase.DicomBytesIO(fn)).pixel_array)
        if isinstance(fn,Path): im = Image.fromarray(dcmread(fn).pixel_array)
        im.load()  # <- Here, if your fn isn't a Path or byte object, it'll crash
        im = im._new(
        return cls(im.convert(mode) if mode else im)

Try to pass the path to the image and it should work. The path should be an instance of Pathclass

Edit: I’ll report it in github

1 Like


Thanks for the quick reply. This ‘pref=pneumonia_img’ is the Path to my image folder.
Which I defined as:

pneumonia_source = Path(root_dir + ‘Datasets/rsna-pneumonia-detection-challenge/’)
pneumonia_img = pneumonia_source/f"stage_2_train_images/"
Path(’/content/gdrive/My Drive/Datasets/rsna-pneumonia-detection-challenge/stage_2_train_images’)

Or should I pass the path somewhere else?

In this case, I don’t know why don’t work. In you are using Jupyter Notebook / Colab, you could debug the issue post mortem. You have two options:

  1. Set %pdb on before running the code so when there is an error, pdb kicks in. It’s a global setting so you could run at the beginning of the notebook. Set %pdb offo disable.
  2. Run %debug afterwards the error. It’s like `%pdb on but you control manually which error you want to debug.

Personally, I prefer running %debugafter I get an error because I control in which error I want to debug.

Please, post your findings in here. It may be another bug.


I couldn’t get more information from the %debug method. But I managed to get it to work! I don’t understand why the ColReader method didn’t work… This how my DataBlock and dataloader looks that did work.

pneumonia = DataBlock(blocks=(ImageBlock(cls=PILDicom), CategoryBlock),
get_x=lambda x:pneumonia_img/f"{x[0]}.dcm",
get_y=lambda x:x[5],

dls = pneumonia.dataloaders(df)

Thanks for the help!

1 Like

Glad you got his working. I started a blog taking a deeper look at medical imaging with fastai a couple of months ago. You can view the blog site (which uses fastpages :)) here: Medical Imaging, one of the blog pages is particularly looking at notebook_60

I also have a number of functions that I had to update such as with datasets that have more than 1 frame per dicom, changing the photometric representation, cmaps not implemented with show_images, implementing features such as DicomSplit which takes takes into account if the same patient occurs both in the test and validation sets etc. It was originally posted on this thread.

What I do find that works with ColReader is something like this:

blocks = (ImageBlock(cls=PILDicom),

getters = [
           ColReader('FileID', pref=source/'train/'),

and then specify the DataBlock

test = DataBlock(blocks=blocks, 
                   getters = getters,
                   item_tfms = Resize(256),

Hope that helps

1 Like

Hi @amritv,

That is an interesting blog, i’m definitely going to read it.

1 Like

Was struggling with loading DICOM as well and only worked with using lambda. However, learn.export() doesn’t support lambda functions so it is not possible to pickle the model.
@Martin2 Did you run into the same problem if you tried to bring the model into production?