Add support for .TIFF files in ImageFolder (#724)

1f8f7ea7 · lhenault · Francisco Massa · 98ca260b · 1f8f7ea7
Commit 1f8f7ea7 authored Jan 25, 2019 by lhenault Committed by Francisco Massa Jan 25, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/folder.py torchvision/datasets/folder.py +1 -1

No files found.
--- a/torchvision/datasets/folder.py
+++ b/torchvision/datasets/folder.py
@@ -151,7 +151,7 @@ class DatasetFolder(data.Dataset):
        return fmt_str
-IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif']
+IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff']
 def pil_loader(path):