How to add new format of image(.nii) to datasets.ImageFolder?

Madi · May 28, 2019, 7:18pm

RuntimeError: Found 0 files in subfolders of: D:/Python/python_stuff/datasets/MRI Normal/dev/train
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif

Geeks_Sid · May 28, 2019, 8:13pm

Hi,

It makes more sense in this scenario to write a custom dataloader since nifti is not supported as an extension yet.

Check out PyTorch Custom DataLoader.

I personally use CSV for DataLoading instead.

Madi · May 28, 2019, 8:26pm

Hi
Thanks.
Can we add any format to folder.py at the torchvision/datasets?

Geeks_Sid · May 28, 2019, 8:40pm

Have you checked out pytorch custom dataloader

Go ahead and check this
StackOverflow and forums taught me this.

Madi · May 28, 2019, 9:29pm

Not yet. I had a meeting.
I will check and let you know. Thanks a lot.
Also, I have added the nii format to folder.py, I got another error. it does not work.