Commit ef5b3dad authored by Raphaël Courivaud's avatar Raphaël Courivaud Committed by Francisco Massa
Browse files

add webp format available in torch datasets folder (#736)

parent 1f8f7ea7
......@@ -151,7 +151,7 @@ class DatasetFolder(data.Dataset):
return fmt_str
IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff']
IMG_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff', 'webp']
def pil_loader(path):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment