Fix ResourceWarning due to Image.open(path) (#176)

See https://github.com/python-pillow/Pillow/issues/835

Fix ResourceWarning due to Image.open(path) (#176)
See https://github.com/python-pillow/Pillow/issues/835
cd8c0c3c · Sam Gross · Soumith Chintala · f4c4d6cb · cd8c0c3c
Commit cd8c0c3c authored May 26, 2017 by Sam Gross Committed by Soumith Chintala May 26, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

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

No files found.
--- a/torchvision/datasets/folder.py
+++ b/torchvision/datasets/folder.py
@@ -39,7 +39,10 @@ def make_dataset(dir, class_to_idx):


 def pil_loader(path):
-    return Image.open(path).convert('RGB')
+    # open path as file to avoid ResourceWarning (https://github.com/python-pillow/Pillow/issues/835)
+    with open(path, 'rb') as f:
+        with Image.open(f) as img:
+            return img.convert('RGB')


 def accimage_loader(path):