From b376216485df45c6ddc7539aa0399f1020173960 Mon Sep 17 00:00:00 2001 From: Calvin Lee Date: Wed, 4 Apr 2018 15:45:49 -0400 Subject: [PATCH] Detect .tif files in ImageFolder --- torchvision/datasets/folder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/datasets/folder.py b/torchvision/datasets/folder.py index a6bb5742bc5..1df4bcbf44d 100644 --- a/torchvision/datasets/folder.py +++ b/torchvision/datasets/folder.py @@ -120,7 +120,7 @@ def __repr__(self): return fmt_str -IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm'] +IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif'] def pil_loader(path):