Load Tiff images to dataset


(Hrishikesh Menon) #1

Hi,

I have a dataset consisting of tiff images (.tiff extension). How do I load it to a pytorch dataset (seeing as torchvision datasets don’t seem to accept this extension)?

Thank you,
richukuttan


(Hrishikesh Menon) #2

As a stop-gap method at least, I have converted all the images to .bmp format using imagemagick (in Ubuntu). I do not know whether this will affect performance, or how much of an effect it will have, but at least the dataset is now being loaded to pytorch.


(Marcin Elantkowski) #3

Looking at torchvision.datasets.ImageFolder code here it seems like
tiff is not supported. This seems like a small bug, since PIL.Image loads tiff without problems?

You can always copy the source I just linked and add tiff extension to IMG_EXTENSIONS, it should then work without issues?

As a hacky workaround:

ls /tmp/test/images/
foo.png  foo.tiff
from torchvision.datasets import ImageFolder
from torchvision.datasets.folder import IMG_EXTENSIONS

len(ImageFolder("/tmp/test/"))  # prints 1

IMG_EXTENSIONS.append('tiff')

len(ImageFolder("/tmp/test/"))  # prints 2